Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

live: Update tests

Browse Source
This commit is contained in:
David de la Iglesia Castro
2022-02-10 16:46:32 +05:45
committed by Saugat Pachhai
parent e6d52ac594
commit 31a3461465
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/func/test_live.py
+7 -7
View File
@@ -20,7 +20,7 @@ LIVE_SCRIPT = dedent(
metrics_logger.log("loss", 1-i/r)
metrics_logger.log("accuracy", i/r)
image = Image.new("RGB", (10,10), (i,i,i))
metrics_logger.log("image.jpg", image)
metrics_logger.log_image("image.jpg", image)
metrics_logger.next_step()"""
)
@@ -131,10 +131,10 @@ def test_live_provides_metrics(tmp_dir, dvc, live_stage):
assert (tmp_dir / "logs").is_dir()
plots_data = dvc.plots.show()
files = get_files(plots_data)
assert os.path.join("logs", "accuracy.tsv") in files
assert os.path.join("logs", "loss.tsv") in files
assert os.path.join("logs", "0", "image.jpg") in files
assert os.path.join("logs", "1", "image.jpg") in files
assert os.path.join("logs", "scalars", "accuracy.tsv") in files
assert os.path.join("logs", "scalars", "loss.tsv") in files
assert os.path.join("logs", "images", "0", "image.jpg") in files
assert os.path.join("logs", "images", "1", "image.jpg") in files
def test_live_provides_no_metrics(tmp_dir, dvc, live_stage):
@@ -146,8 +146,8 @@ def test_live_provides_no_metrics(tmp_dir, dvc, live_stage):
assert (tmp_dir / "logs").is_dir()
plots_data = dvc.plots.show()
files = get_files(plots_data)
assert os.path.join("logs", "accuracy.tsv") in files
assert os.path.join("logs", "loss.tsv") in files
assert os.path.join("logs", "scalars", "accuracy.tsv") in files
assert os.path.join("logs", "scalars", "loss.tsv") in files
@pytest.mark.parametrize("typ", ("live", "live_no_cache"))