diff --git a/src/data_preparation.py b/src/data_preparation.py index 258a24753e7a7744b75b68f3ab2b68a77fe90061..64fd7f52b2309a79217d6d24d65757b35904a2e9 100644 --- a/src/data_preparation.py +++ b/src/data_preparation.py @@ -77,16 +77,19 @@ class DataPrep(object): file_name = self._set_file_name() meta_file = self._set_meta_file_name() try: + logging.debug(f"try to load local data from: {file_name}") data = self._slice_prep(xr.open_dataarray(file_name)) self.data = self.check_for_negative_concentrations(data) self.meta = pd.read_csv(meta_file, index_col=0) if self.station_type is not None: self.check_station_type() + logging.debug("loading finished") except FileNotFoundError as e: logging.warning(e) data, self.meta = self.download_data_from_join(file_name, meta_file) data = self._slice_prep(data) self.data = self.check_for_negative_concentrations(data) + logging.debug("loaded new data from JOIN") def check_station_type(self): """ diff --git a/test/test_data_generator.py b/test/test_data_generator.py index 9755881c46943a294d744e589fd202e93d3f6f32..d7b58d32b37df71a7236cd6b61db55b371f7f7ef 100644 --- a/test/test_data_generator.py +++ b/test/test_data_generator.py @@ -54,7 +54,8 @@ class TestDataGenerator: for i, d in enumerate(gen, start=1): assert i == gen._iterator - def test_getitem(self, gen): + def test_getitem(self, caplog, gen): + caplog.set_level(logging.DEBUG) print(gen) gen.kwargs = {'statistics_per_var': {'o3': 'dma8eu', 'temp': 'maximum'}} print(gen)