[Artifacts] Fix artifacts output test (mlrun#2040)

guy1992l · Jun 12, 2022 · 84ea904 · 84ea904
1 parent edbcee0
commit 84ea904
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/tests/system/base.py b/tests/system/base.py
@@ -244,10 +244,9 @@ def _verify_run_outputs(
         iteration_results: bool = False,
     ):
         self._logger.debug("Verifying run outputs", spec=run_outputs)
-        iterpath = str(best_iteration) if best_iteration else ""
-        assert run_outputs["model"] == str(output_path / iterpath / "model.txt")
-        assert run_outputs["html_result"] == str(output_path / iterpath / "result.html")
-        assert run_outputs["chart"] == str(output_path / iterpath / "chart.html")
+        assert run_outputs["model"].startswith(str(output_path))
+        assert run_outputs["html_result"].startswith(str(output_path))
+        assert run_outputs["chart"].startswith(str(output_path))
         assert run_outputs["mydf"] == f"store://artifacts/{project}/{name}_mydf:{uid}"
         if accuracy:
             assert run_outputs["accuracy"] == accuracy
@@ -256,9 +255,7 @@ def _verify_run_outputs(
         if best_iteration:
             assert run_outputs["best_iteration"] == best_iteration
         if iteration_results:
-            assert run_outputs["iteration_results"] == str(
-                output_path / "iteration_results.csv"
-            )
+            assert run_outputs["iteration_results"].startswith(str(output_path))
 
     @staticmethod
     def _has_marker(test: typing.Callable, marker_name: str) -> bool: