diff --git a/emiproc/tests_utils/__init__.py b/emiproc/tests_utils/__init__.py index 4e81c31..0586893 100644 --- a/emiproc/tests_utils/__init__.py +++ b/emiproc/tests_utils/__init__.py @@ -2,14 +2,16 @@ import emiproc from pathlib import Path +TEST_DIR = Path(*emiproc.__path__) / ".." / "tests" +TEST_DIR.mkdir(exist_ok=True) -WEIGHTS_DIR = Path(*emiproc.__path__) / ".." / "tests" / ".weights" +WEIGHTS_DIR = TEST_DIR / ".weights" WEIGHTS_DIR.mkdir(exist_ok=True) -TEST_OUTPUTS_DIR = Path(*emiproc.__path__) / ".." / "tests" / ".outputs" +TEST_OUTPUTS_DIR = TEST_DIR / ".outputs" TEST_OUTPUTS_DIR.mkdir(exist_ok=True) if __name__ == "__main__": print(WEIGHTS_DIR) - print(WEIGHTS_DIR.is_dir()) \ No newline at end of file + print(WEIGHTS_DIR.is_dir())