mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Peter Rowlands (변기호)
70ddb7c931
* experiments: include experiment commit times in show * --no-timestamp can be used to hide timestamps * update tests * clean up kwargs
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from datetime import datetime
|
|
|
|
from dvc.dvcfile import PIPELINE_FILE
|
|
from tests.func.test_repro_multistage import COPY_SCRIPT
|
|
|
|
|
|
def test_show_simple(tmp_dir, scm, dvc):
|
|
tmp_dir.gen("copy.py", COPY_SCRIPT)
|
|
tmp_dir.gen("params.yaml", "foo: 1")
|
|
dvc.run(
|
|
cmd="python copy.py params.yaml metrics.yaml",
|
|
metrics_no_cache=["metrics.yaml"],
|
|
params=["foo"],
|
|
single_stage=True,
|
|
)
|
|
|
|
assert dvc.experiments.show()["workspace"] == {
|
|
"baseline": {
|
|
"metrics": {"metrics.yaml": {"foo": 1}},
|
|
"params": {"params.yaml": {"foo": 1}},
|
|
"queued": False,
|
|
"timestamp": None,
|
|
}
|
|
}
|
|
|
|
|
|
def test_show_experiment(tmp_dir, scm, dvc):
|
|
tmp_dir.gen("copy.py", COPY_SCRIPT)
|
|
tmp_dir.gen("params.yaml", "foo: 1")
|
|
dvc.run(
|
|
cmd="python copy.py params.yaml metrics.yaml",
|
|
metrics_no_cache=["metrics.yaml"],
|
|
params=["foo"],
|
|
name="foo",
|
|
)
|
|
scm.add(["copy.py", "params.yaml", "metrics.yaml", "dvc.yaml", "dvc.lock"])
|
|
scm.commit("baseline")
|
|
baseline_rev = scm.get_rev()
|
|
timestamp = datetime.fromtimestamp(
|
|
scm.repo.rev_parse(baseline_rev).committed_date
|
|
)
|
|
|
|
dvc.reproduce(PIPELINE_FILE, experiment=True, params=["foo=2"])
|
|
results = dvc.experiments.show()
|
|
|
|
expected_baseline = {
|
|
"metrics": {"metrics.yaml": {"foo": 1}},
|
|
"params": {"params.yaml": {"foo": 1}},
|
|
"queued": False,
|
|
"timestamp": timestamp,
|
|
}
|
|
expected_params = {"foo": 2}
|
|
|
|
assert set(results.keys()) == {"workspace", baseline_rev}
|
|
experiments = results[baseline_rev]
|
|
assert len(experiments) == 2
|
|
for rev, exp in experiments.items():
|
|
if rev == "baseline":
|
|
assert exp == expected_baseline
|
|
else:
|
|
assert exp["metrics"]["metrics.yaml"] == expected_params
|
|
assert exp["params"]["params.yaml"] == expected_params
|
|
assert exp["timestamp"] > timestamp
|