🐛 fix unit-test

- wasnt' run before apparently
RasmussenLab · Jul 2, 2024 · 9d6858b · 9d6858b
1 parent 266ad94
commit 9d6858b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/models/test_collect_dumps.py b/tests/models/test_collect_dumps.py
@@ -5,5 +5,8 @@ def test_select_content():
     test_cases = ['model_metrics_HL_1024_512_256_dae',
                   'model_metrics_HL_1024_512_vae',
                   'model_metrics_collab']
-    for test_case in test_cases:
-        assert select_content(test_case, first_split='metrics_') == test_case.split('metrics_')[1]
+    expected = ['HL_1024_512_256',
+                'HL_1024_512',
+                'collab']
+    for test_case, v in zip(test_cases, expected):
+        assert select_content(test_case, first_split='metrics_') == v