import pytest
from dvc.utils.html import HTML, PAGE_HTML, MissingPlaceholderError
CUSTOM_PAGE_HTML = """
TITLE
{plot_divs}
"""
@pytest.mark.parametrize(
"template,page_elements,expected_page",
[
(None, ["content"], PAGE_HTML.format(plot_divs="content")),
(
CUSTOM_PAGE_HTML,
["content"],
CUSTOM_PAGE_HTML.format(plot_divs="content"),
),
],
)
def test_html(tmp_dir, template, page_elements, expected_page):
page = HTML(template)
page.elements = page_elements
result = page.embed()
assert result == expected_page
def test_no_placeholder(tmp_dir):
template = ""
with pytest.raises(MissingPlaceholderError):
HTML(template)