update unit tests

pipelines/.coverage-gitlab-ci.yaml

@@ -14,6 +14,8 @@
   - unittest_tf_tools
   - unittest_train_tools
   - unittest_metrics
+  - unittest_classification_tools
+  - unittest_data_tools
 
 test_coverage:
   stage: coverage_test_stage

umami/tests/unit/preprocessing/test_preprocess_plotting.py

@@ -25,7 +25,7 @@ class PreprocessPlotting_TestCase(unittest.TestCase):
 
         # Load test preprocessing config file from current directory
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
         self.config = upt.Configuration(self.config_file)
 

umami/tests/unit/preprocessing/test_preprocessing_tools.py

@@ -20,7 +20,7 @@ class ConfigurationTestCase(unittest.TestCase):
         Set a example config file.
         """
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
 
     def test_missing_key_error(self):
@@ -88,7 +88,9 @@ class PrepareSamplesTestCase(unittest.TestCase):
     class c_args:
         def __init__(self) -> None:
             self.sample = "ttbar"
-            self.config_file = "test_preprocess_config.yaml"
+            self.config_file = os.path.join(
+                os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
+            )
             self.shuffle_array = True
 
     def setUp(self):

umami/tests/unit/preprocessing/test_preprocessing_tools_Resampling.py

@@ -107,7 +107,7 @@ class UnderSamplingTestCase(unittest.TestCase):
         Create a default dataset for testing.
         """
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
         self.config = Configuration(self.config_file)
         self.sampling_config = self.config.sampling
@@ -234,7 +234,7 @@ class PDFResamplingTestCase(unittest.TestCase):
         Create a default dataset for testing.
         """
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
         self.config = Configuration(self.config_file)
         sampling_config = self.config.sampling
@@ -348,7 +348,7 @@ class ProbabilityRatioUnderSamplingTestCase(unittest.TestCase):
         Create a default dataset for testing.
         """
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
         self.config = Configuration(self.config_file)
         sampling_config = self.config.sampling

umami/tests/unit/preprocessing/test_preprocessing_tools_Scaling.py

@@ -46,7 +46,7 @@ class ScalingTestCase(unittest.TestCase):
 
     def setUp(self):
         self.config_file = os.path.join(
-            os.path.dirname(__file__), "test_preprocess_config.yaml"
+            os.path.dirname(__file__), "fixtures", "test_preprocess_config.yaml"
         )
         self.config = Configuration(self.config_file)
         self.config.var_file = os.path.join(