diff --git a/tests/unit/test_data_prep.py b/tests/unit/test_data_prep.py index 24b0445..591f8fa 100644 --- a/tests/unit/test_data_prep.py +++ b/tests/unit/test_data_prep.py @@ -15,7 +15,6 @@ logging_basic_config, smart_unique, ) - from util import load_df diff --git a/tests/unit/test_spare_scores.py b/tests/unit/test_spare_scores.py index f31f2b3..0ea7058 100644 --- a/tests/unit/test_spare_scores.py +++ b/tests/unit/test_spare_scores.py @@ -1,12 +1,14 @@ +import sys import unittest from pathlib import Path -import sys + import pandas as pd -sys.path.append("../spare_scores") -from spare_scores import spare_test, spare_train +sys.path.append("../../spare_scores") from util import load_df, load_model +from spare_scores import spare_test, spare_train + class CheckSpareScores(unittest.TestCase): @@ -76,7 +78,7 @@ def test_spare_train(self): ) status, result_data = result["status"], result["data"] - + metadata = result_data[1] self.assertTrue(status == "OK") self.assertTrue(metadata["mdl_type"] == self.model_fixture[1]["mdl_type"]) diff --git a/tests/unit/test_util.py b/tests/unit/test_util.py index e2f73d9..2265f36 100644 --- a/tests/unit/test_util.py +++ b/tests/unit/test_util.py @@ -7,7 +7,7 @@ import numpy as np import pandas as pd -sys.path.append("../spare_scores") +sys.path.append("../../spare_scores") from util import ( add_file_extension, check_file_exists,