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

more print statements

parent 19f0d017
No related branches found
No related tags found
2 merge requests!17update to v0.4.0,!16handle station type
Pipeline #26597 passed
...@@ -77,16 +77,19 @@ class DataPrep(object): ...@@ -77,16 +77,19 @@ class DataPrep(object):
file_name = self._set_file_name() file_name = self._set_file_name()
meta_file = self._set_meta_file_name() meta_file = self._set_meta_file_name()
try: try:
logging.debug(f"try to load local data from: {file_name}")
data = self._slice_prep(xr.open_dataarray(file_name)) data = self._slice_prep(xr.open_dataarray(file_name))
self.data = self.check_for_negative_concentrations(data) self.data = self.check_for_negative_concentrations(data)
self.meta = pd.read_csv(meta_file, index_col=0) self.meta = pd.read_csv(meta_file, index_col=0)
if self.station_type is not None: if self.station_type is not None:
self.check_station_type() self.check_station_type()
logging.debug("loading finished")
except FileNotFoundError as e: except FileNotFoundError as e:
logging.warning(e) logging.warning(e)
data, self.meta = self.download_data_from_join(file_name, meta_file) data, self.meta = self.download_data_from_join(file_name, meta_file)
data = self._slice_prep(data) data = self._slice_prep(data)
self.data = self.check_for_negative_concentrations(data) self.data = self.check_for_negative_concentrations(data)
logging.debug("loaded new data from JOIN")
def check_station_type(self): def check_station_type(self):
""" """
......
...@@ -54,7 +54,8 @@ class TestDataGenerator: ...@@ -54,7 +54,8 @@ class TestDataGenerator:
for i, d in enumerate(gen, start=1): for i, d in enumerate(gen, start=1):
assert i == gen._iterator assert i == gen._iterator
def test_getitem(self, gen): def test_getitem(self, caplog, gen):
caplog.set_level(logging.DEBUG)
print(gen) print(gen)
gen.kwargs = {'statistics_per_var': {'o3': 'dma8eu', 'temp': 'maximum'}} gen.kwargs = {'statistics_per_var': {'o3': 'dma8eu', 'temp': 'maximum'}}
print(gen) print(gen)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment