Skip to content
Snippets Groups Projects
Commit bb7fa953 authored by lukas leufen's avatar lukas leufen
Browse files

error found, removed print statements

parent f8ded3c6
No related branches found
No related tags found
2 merge requests!17update to v0.4.0,!16handle station type
Pipeline #26606 passed
......@@ -98,7 +98,6 @@ class DataGenerator(keras.utils.Sequence):
station = self.get_station_key(key)
data = DataPrep(self.data_path, self.network, station, self.variables, station_type=self.station_type,
**self.kwargs)
print("data: ", data)
data.interpolate(self.interpolate_dim, method=self.interpolate_method, limit=self.limit_nan_fill)
data.transform("datetime", method=self.transform_method)
data.make_history_window(self.interpolate_dim, self.window_history)
......
......@@ -43,13 +43,9 @@ class TestDataGenerator:
gen.stations = ['station1', 'station2', 'station3']
assert len(gen) == 3
def test_getitem(self, caplog, gen):
caplog.set_level(logging.DEBUG)
print(gen)
def test_getitem(self, gen):
gen.kwargs = {'statistics_per_var': {'o3': 'dma8eu', 'temp': 'maximum'}}
print(gen)
station = gen["DEBW107"]
print(station)
assert len(station) == 2
assert station[0].Stations.data == "DEBW107"
assert station[0].data.shape[1:] == (8, 1, 2)
......
......@@ -43,7 +43,8 @@ class TestPreProcessing:
PreProcessing()
assert caplog.record_tuples[0] == ('root', 20, 'PreProcessing started')
assert caplog.record_tuples[1] == ('root', 20, 'check valid stations started')
assert caplog.record_tuples[-2] == ('root', 20, PyTestRegex(r'run for \d+\.\d+s to check 5 station\(s\)'))
assert caplog.record_tuples[-2] == ('root', 20, PyTestRegex(r'run for \d+\.\d+s to check 5 station\(s\). Found '
r'5/5 valid stations.'))
RunEnvironment().__del__()
def test_run(self, obj_with_exp_setup):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment