test(cli train): add assertion that checkpoints exist

CAREamics · Jun 5, 2024 · 4418423 · 4418423
1 parent 9c2dfdd
commit 4418423
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/cli/test_main.py b/tests/cli/test_main.py
@@ -39,6 +39,8 @@ def test_train(tmp_path: Path, minimum_configuration: dict):
             str(tmp_path),
         ],
     )
+    assert (tmp_path/"checkpoints").is_dir()
+    assert len(list((tmp_path/"checkpoints").glob("*"))) > 0
     assert result.exit_code == 0