From 7ca52286c867677b6067bf52552116ccf2470230 Mon Sep 17 00:00:00 2001 From: ctrlaltaf Date: Fri, 7 Jun 2024 09:51:52 -0700 Subject: [PATCH] final please --- tests/test_pytest.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/test_pytest.py b/tests/test_pytest.py index 20acc85..38c3dd9 100644 --- a/tests/test_pytest.py +++ b/tests/test_pytest.py @@ -14,6 +14,8 @@ import_graph_from_pickle, ) from tools.workflow import run_workflow +import os + def test_algorithm_attributes(): @@ -47,6 +49,15 @@ def test_algorithm_inherits_class(): def test_algorithm_workflow(): + if not os.path.exists("output"): + os.makedirs("output") + if not os.path.exists("output/dataset"): + os.makedirs("output/dataset") + if not os.path.exists("output/data"): + os.makedirs("output/data") + if not os.path.exists("output/images"): + os.makedirs("output/images") + output_data_path = Path("./output/data/") output_image_path = Path("./output/images/") input_directory_path = Path("./tests/testing-dataset/")