From 84ea90480a5c106d7281e63d769cde23bcab3477 Mon Sep 17 00:00:00 2001 From: Yaron Haviv Date: Sun, 12 Jun 2022 19:02:37 +0300 Subject: [PATCH] [Artifacts] Fix artifacts output test (#2040) --- tests/system/base.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/system/base.py b/tests/system/base.py index 55f85b0ee76..d4fe2708ae1 100644 --- 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: