Skip to content

Commit 31a3461

Browse files
daavooskshetry
authored andcommitted
live: Update tests
1 parent e6d52ac commit 31a3461

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/func/test_live.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
metrics_logger.log("loss", 1-i/r)
2121
metrics_logger.log("accuracy", i/r)
2222
image = Image.new("RGB", (10,10), (i,i,i))
23-
metrics_logger.log("image.jpg", image)
23+
metrics_logger.log_image("image.jpg", image)
2424
metrics_logger.next_step()"""
2525
)
2626

@@ -131,10 +131,10 @@ def test_live_provides_metrics(tmp_dir, dvc, live_stage):
131131
assert (tmp_dir / "logs").is_dir()
132132
plots_data = dvc.plots.show()
133133
files = get_files(plots_data)
134-
assert os.path.join("logs", "accuracy.tsv") in files
135-
assert os.path.join("logs", "loss.tsv") in files
136-
assert os.path.join("logs", "0", "image.jpg") in files
137-
assert os.path.join("logs", "1", "image.jpg") in files
134+
assert os.path.join("logs", "scalars", "accuracy.tsv") in files
135+
assert os.path.join("logs", "scalars", "loss.tsv") in files
136+
assert os.path.join("logs", "images", "0", "image.jpg") in files
137+
assert os.path.join("logs", "images", "1", "image.jpg") in files
138138

139139

140140
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):
146146
assert (tmp_dir / "logs").is_dir()
147147
plots_data = dvc.plots.show()
148148
files = get_files(plots_data)
149-
assert os.path.join("logs", "accuracy.tsv") in files
150-
assert os.path.join("logs", "loss.tsv") in files
149+
assert os.path.join("logs", "scalars", "accuracy.tsv") in files
150+
assert os.path.join("logs", "scalars", "loss.tsv") in files
151151

152152

153153
@pytest.mark.parametrize("typ", ("live", "live_no_cache"))

0 commit comments

Comments
 (0)