From 6a34bebbcdc1c534c233a4cd8dee3d4d415cb94a Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Thu, 28 Nov 2019 14:42:12 +0100
Subject: [PATCH] more print statements
---
src/data_preparation.py | 3 +++
test/test_data_generator.py | 3 ++-
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/src/data_preparation.py b/src/data_preparation.py
index 258a2475..64fd7f52 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 9755881c..d7b58d32 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)
--
GitLab