mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 23:09:44 +01:00
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
import os
|
|
|
|
import dpath
|
|
|
|
from dvc.render.utils import match_renderers, render
|
|
from dvc.render.vega import VegaRenderer
|
|
|
|
|
|
def assert_website_has_image(page_path, revision, filename, image_content):
|
|
index_path = page_path / "index.html"
|
|
assert index_path.is_file()
|
|
index_content = index_path.read_text()
|
|
|
|
resources_filename = f"{revision}_{filename.replace(os.sep, '_')}"
|
|
image_path = page_path / "static" / resources_filename
|
|
assert image_path.is_file()
|
|
|
|
img_html = f'<img src="{os.path.join("static", resources_filename)}">'
|
|
assert img_html in index_content
|
|
with open(image_path, "rb") as fobj:
|
|
assert fobj.read() == image_content
|
|
|
|
|
|
def test_render(tmp_dir, dvc):
|
|
data = {
|
|
"HEAD": {
|
|
"data": {
|
|
"file.json": {
|
|
"data": [{"y": 5}, {"y": 6}],
|
|
"props": {"fields": {"y"}},
|
|
},
|
|
os.path.join("sub", "other_file.jpg"): {"data": b"content"},
|
|
}
|
|
},
|
|
"v2": {
|
|
"data": {
|
|
"file.json": {
|
|
"data": [{"y": 3}, {"y": 5}],
|
|
"props": {"fields": {"y"}},
|
|
},
|
|
"other_file.jpg": {"data": b"content2"},
|
|
}
|
|
},
|
|
"v1": {
|
|
"data": {
|
|
"some.csv": {
|
|
"data": [{"y": 2}, {"y": 3}],
|
|
"props": {"fields": {"y"}},
|
|
},
|
|
"another.gif": {"data": b"content3"},
|
|
}
|
|
},
|
|
}
|
|
renderers = match_renderers(data, dvc.plots.templates)
|
|
|
|
render(dvc, renderers, path=os.path.join("results", "dir"))
|
|
page_path = tmp_dir / "results" / "dir"
|
|
index_path = page_path / "index.html"
|
|
|
|
assert index_path.is_file()
|
|
assert_website_has_image(
|
|
page_path, "HEAD", os.path.join("sub", "other_file.jpg"), b"content"
|
|
)
|
|
assert_website_has_image(page_path, "v2", "other_file.jpg", b"content2")
|
|
assert_website_has_image(page_path, "v1", "another.gif", b"content3")
|
|
|
|
def clean(txt: str) -> str:
|
|
return txt.replace("\n", "").replace("\r", "").replace(" ", "")
|
|
|
|
def get_vega_string(data, filename):
|
|
file_data = dpath.util.search(data, ["*", "*", filename])
|
|
return VegaRenderer(
|
|
file_data, dvc.plots.templates.load()
|
|
).partial_html()
|
|
|
|
index_content = index_path.read_text()
|
|
assert clean(get_vega_string(data, "file.json")) in clean(index_content)
|
|
assert clean(get_vega_string(data, "some.csv")) in clean(index_content)
|