mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +01:00
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
import pytest
|
|
|
|
from dvc.render.html import HTML, PAGE_HTML, MissingPlaceholderError
|
|
|
|
CUSTOM_PAGE_HTML = """<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<title>TITLE</title>
|
|
<script type="text/javascript" src="vega"></script>
|
|
<script type="text/javascript" src="vega-lite"></script>
|
|
<script type="text/javascript" src="vega-embed"></script>
|
|
</head>
|
|
<body>
|
|
{plot_divs}
|
|
</body>
|
|
</html>"""
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"template,page_elements,expected_page",
|
|
[
|
|
(
|
|
None,
|
|
["content"],
|
|
PAGE_HTML.format(plot_divs="content", refresh_tag="", scripts=""),
|
|
),
|
|
(
|
|
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 = "<head></head><body></body>"
|
|
|
|
with pytest.raises(MissingPlaceholderError):
|
|
HTML(template)
|