mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Saugat Pachhai
b9000ed926
* pre-commit: update black to 21.5b1 also updated others, but black is more noticeable * remove magic trailing comma
102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
import json
|
|
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()
|
|
p1_content = json.loads(result[p1])
|
|
p2_content = json.loads(result[p2])
|
|
|
|
assert p1_content["title"] == p2_content["title"] == "TITLE"
|
|
assert p1_content == p2_content
|