diff --git a/test/test_modules.py b/test/test_modules.py
index b2add462a973e01d90e4bccb3252e2d42233791a..ce519e40984d20cb16ed5a4f970f8dd721cb0945 100644
--- a/test/test_modules.py
+++ b/test/test_modules.py
@@ -69,21 +69,21 @@ class TestPreProcessing:
assert dummy_list[val] == list(range(10, 13))
assert dummy_list[test] == list(range(13, 15))
- @mock.patch("DataGenerator", return_value=object.__new__(DataGenerator))
- @mock.patch("DataGenerator[station]", return_value=(np.ones(10), np.zeros(10)))
- def test_create_set_split(self):
- stations = ['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087']
- pre = object.__new__(PreProcessing)
- pre.setup = ExperimentSetup({}, stations=stations, var_all_dict={'o3': 'dma8eu', 'temp': 'maximum'},
- train_kwargs={"start": "2000-01-01", "end": "2007-12-31"})
- kwargs = {'start': '1997-01-01', 'end': '2017-12-31', 'statistics_per_var': pre.setup.var_all_dict, }
- train = pre.create_set_split(stations, pre.setup.__dict__, kwargs, slice(0, 3), "train")
- # stopped here. It is a mess with all the different kwargs, args etc. Restructure the idea of how to implement
- # the data sets. Because there are multiple kwargs declarations and which counts in the end. And there are
- # multiple declarations of the DataGenerator class. Why this? Is it somehow possible, to select elements from
- # this iterator class. Furthermore the names of the DataPrep class is not distinct, because there is no time
- # range provided in file's name. Given the case, that first to total DataGen is called with a short period for
- # data loading. But then, for the data split (I don't know why this could happen, but it is very likely because
- # osf the current multiple declarations of kwargs arguments) the desired time range exceeds the previou
- # mentioned and short time range. But nevertheless, the file with the short period is loaded and used (because
- # during DataPrep loading, the available range is checked).
+ # @mock.patch("DataGenerator", return_value=object.__new__(DataGenerator))
+ # @mock.patch("DataGenerator[station]", return_value=(np.ones(10), np.zeros(10)))
+ # def test_create_set_split(self):
+ # stations = ['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087']
+ # pre = object.__new__(PreProcessing)
+ # pre.setup = ExperimentSetup({}, stations=stations, var_all_dict={'o3': 'dma8eu', 'temp': 'maximum'},
+ # train_kwargs={"start": "2000-01-01", "end": "2007-12-31"})
+ # kwargs = {'start': '1997-01-01', 'end': '2017-12-31', 'statistics_per_var': pre.setup.var_all_dict, }
+ # train = pre.create_set_split(stations, pre.setup.__dict__, kwargs, slice(0, 3), "train")
+ # # stopped here. It is a mess with all the different kwargs, args etc. Restructure the idea of how to implement
+ # # the data sets. Because there are multiple kwargs declarations and which counts in the end. And there are
+ # # multiple declarations of the DataGenerator class. Why this? Is it somehow possible, to select elements from
+ # # this iterator class. Furthermore the names of the DataPrep class is not distinct, because there is no time
+ # # range provided in file's name. Given the case, that first to total DataGen is called with a short period for
+ # # data loading. But then, for the data split (I don't know why this could happen, but it is very likely because
+ # # osf the current multiple declarations of kwargs arguments) the desired time range exceeds the previou
+ # # mentioned and short time range. But nevertheless, the file with the short period is loaded and used (because
+ # # during DataPrep loading, the available range is checked).
diff --git a/test/test_modules/__init__.py b/test/test_modules/__init__.py
deleted file mode 100644
index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000
diff --git a/test/test_modules/test_experiment_setup.py b/test/test_modules/test_experiment_setup.py
index cf634b2c541db73320d1ecc461f30f672a479c93..162ea469f47a52fafa2a5f0f9c4bf796f0adc0a2 100644
--- a/test/test_modules/test_experiment_setup.py
+++ b/test/test_modules/test_experiment_setup.py
@@ -12,9 +12,9 @@ class TestExperimentSetup:
@pytest.fixture
def empty_obj(self, caplog):
- caplog.set_level(logging.DEBUG)
obj = object.__new__(ExperimentSetup)
obj.time = TimeTracking()
+ caplog.set_level(logging.DEBUG)
return obj
def test_set_param_by_value(self, caplog, empty_obj):
@@ -36,15 +36,10 @@ class TestExperimentSetup:
res = empty_obj._get_parser_args({'test2': 2, 'test10str': "10"})
assert res == {'test2': 2, 'test10str': "10"}
- def test_get_parser_args_from_argparser(self, empty_obj):
- parser = argparse.ArgumentParser()
- parser.add_argument('--experiment_date', type=str, nargs=1, default="TODAY")
- assert empty_obj._get_parser_args(parser) == {"experiment_date": "TODAY"}
-
def test_get_parser_args_from_parse_args(self, empty_obj):
parser = argparse.ArgumentParser()
- parser.add_argument('--experiment_date', type=str, nargs=1, default="TOMORROW")
- parser_args = parser.parse_args()
+ parser.add_argument('--experiment_date', type=str)
+ parser_args = parser.parse_args(["--experiment_date", "TOMORROW"])
assert empty_obj._get_parser_args(parser_args) == {"experiment_date": "TOMORROW"}
def test_init_default(self):