mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
Paweł Redzyński
ab08db5065
* plots: support images initial * plots: data: unify PlotsData * plots: data conversion refactor * plots: renderers: abstract functionality * plots: uncommenting tests * plots: self implement grouping * plots: rename group * plots: extract rendering tests * plots: move rendering tests to unit * plots: add image rendering tests * plots: dont use dpath to find vega plot * plots: test find_vega * plots: test render function * plots: move rendering out of repo package * plots: refactoring * plots: self review refactor
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from dvc.dvcfile import PIPELINE_LOCK
|
|
from dvc.repo.plots import PropsNotFoundError
|
|
from dvc.repo.plots.template import TemplateNotFoundError
|
|
from dvc.utils import relpath
|
|
from tests.func.plots.utils import _write_json
|
|
|
|
|
|
def test_plots_modify_existing_template(
|
|
tmp_dir, dvc, run_copy_metrics, custom_template
|
|
):
|
|
metric = [{"a": 1, "b": 2}, {"a": 2, "b": 3}]
|
|
_write_json(tmp_dir, metric, "metric_t.json")
|
|
stage = run_copy_metrics(
|
|
"metric_t.json",
|
|
"metric.json",
|
|
plots_no_cache=["metric.json"],
|
|
name="copy-metrics",
|
|
single_stage=False,
|
|
)
|
|
dvc.plots.modify(
|
|
"metric.json", props={"template": relpath(custom_template)}
|
|
)
|
|
stage = stage.reload()
|
|
assert stage.outs[0].plot == {"template": relpath(custom_template)}
|
|
|
|
|
|
def test_plots_modify_should_not_change_lockfile(
|
|
tmp_dir, dvc, run_copy_metrics, custom_template
|
|
):
|
|
_write_json(tmp_dir, [{"a": 1, "b": 2}], "metric_t.json")
|
|
run_copy_metrics(
|
|
"metric_t.json",
|
|
"metric.json",
|
|
plots_no_cache=["metric.json"],
|
|
name="copy-metrics",
|
|
single_stage=False,
|
|
)
|
|
|
|
(tmp_dir / PIPELINE_LOCK).unlink()
|
|
dvc.plots.modify(
|
|
"metric.json", props={"template": relpath(custom_template)}
|
|
)
|
|
assert not (tmp_dir / PIPELINE_LOCK).exists()
|
|
|
|
|
|
def test_plots_modify_not_existing_template(dvc):
|
|
with pytest.raises(TemplateNotFoundError):
|
|
dvc.plots.modify(
|
|
"metric.json", props={"template": "not-existing-template.json"}
|
|
)
|
|
|
|
|
|
def test_unset_nonexistent(tmp_dir, dvc, run_copy_metrics, custom_template):
|
|
metric = [{"a": 1, "b": 2}, {"a": 2, "b": 3}]
|
|
_write_json(tmp_dir, metric, "metric_t.json")
|
|
run_copy_metrics(
|
|
"metric_t.json",
|
|
"metric.json",
|
|
plots_no_cache=["metric.json"],
|
|
name="copy-metrics",
|
|
single_stage=False,
|
|
)
|
|
|
|
with pytest.raises(PropsNotFoundError):
|
|
dvc.plots.modify("metric.json", unset=["nonexistent"])
|
|
|
|
|
|
def test_dir_plots(tmp_dir, dvc, run_copy_metrics):
|
|
subdir = tmp_dir / "subdir"
|
|
subdir.mkdir()
|
|
|
|
metric = [{"first_val": 100, "val": 2}, {"first_val": 200, "val": 3}]
|
|
|
|
fname = "file.json"
|
|
_write_json(tmp_dir, metric, fname)
|
|
|
|
p1 = os.path.join("subdir", "p1.json")
|
|
p2 = os.path.join("subdir", "p2.json")
|
|
tmp_dir.dvc.run(
|
|
cmd=(
|
|
f"mkdir subdir && python copy.py {fname} {p1} && "
|
|
f"python copy.py {fname} {p2}"
|
|
),
|
|
deps=[fname],
|
|
single_stage=False,
|
|
plots=["subdir"],
|
|
name="copy_double",
|
|
)
|
|
dvc.plots.modify("subdir", {"title": "TITLE"})
|
|
|
|
result = dvc.plots.show()
|
|
assert result["workspace"]["data"][p1]["props"]["title"] == "TITLE"
|
|
assert result["workspace"]["data"][p2]["props"]["title"] == "TITLE"
|