From 9facd09abb0c6fcb1f90d8fb912370519ea488ab Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Mon, 29 Nov 2021 16:07:56 +0100
Subject: [PATCH] updated a test
---
.../data_handler/data_handler_mixed_sampling.py | 7 -------
.../test_data_handler_mixed_sampling.py | 17 ++++++++++++-----
2 files changed, 12 insertions(+), 12 deletions(-)
diff --git a/mlair/data_handler/data_handler_mixed_sampling.py b/mlair/data_handler/data_handler_mixed_sampling.py
index 3e625ec7..2a3385e5 100644
--- a/mlair/data_handler/data_handler_mixed_sampling.py
+++ b/mlair/data_handler/data_handler_mixed_sampling.py
@@ -357,13 +357,6 @@ class DataHandlerSeparationOfScalesSingleStation(DataHandlerMixedSamplingWithKzF
self.history = self.stride(data, dim_name_of_shift, window, offset=self.window_history_offset)
def stride(self, data: xr.DataArray, dim: str, window: int, offset: int = 0) -> xr.DataArray:
-
- # this is just a code snippet to check the results of the kz filter
- # import matplotlib
- # matplotlib.use("TkAgg")
- # import matplotlib.pyplot as plt
- # xr.concat(res, dim="filter").sel({"variables":"temp", "Stations":"DEBW107", "datetime":"2010-01-01T00:00:00"}).plot.line(hue="filter")
-
time_deltas = np.round(self.time_delta(self.cutoff_period)).astype(int)
start, end = window, 1
res = []
diff --git a/test/test_data_handler/test_data_handler_mixed_sampling.py b/test/test_data_handler/test_data_handler_mixed_sampling.py
index 5fedeebb..77b5d280 100644
--- a/test/test_data_handler/test_data_handler_mixed_sampling.py
+++ b/test/test_data_handler/test_data_handler_mixed_sampling.py
@@ -213,11 +213,18 @@ class TestDataHandlerSeparationOfScales:
assert obj.data_handler_transformation.__qualname__ == DataHandlerSeparationOfScalesSingleStation.__qualname__
def test_requirements(self):
- obj = object.__new__(DataHandlerMixedSamplingWithKzFilter)
- req1 = object.__new__(DataHandlerMixedSamplingWithFilterSingleStation)
- req2 = object.__new__(DataHandlerKzFilterSingleStation)
- req = list(set(req1.requirements() + req2.requirements()))
- assert sorted(obj._requirements) == sorted(remove_items(req, "station"))
+ reqs = get_all_args(DefaultDataHandler)
+ obj = object.__new__(DataHandlerSeparationOfScales)
+ assert sorted(obj.own_args()) == reqs
+
+ reqs = get_all_args(DataHandlerSeparationOfScalesSingleStation, DataHandlerKzFilterSingleStation,
+ DataHandlerMixedSamplingWithKzFilterSingleStation,DataHandlerFilterSingleStation,
+ DataHandlerSingleStation, remove=["self", "id_class"])
+ assert sorted(obj._requirements) == reqs
+ reqs = get_all_args(DataHandlerSeparationOfScalesSingleStation, DataHandlerKzFilterSingleStation,
+ DataHandlerMixedSamplingWithKzFilterSingleStation,DataHandlerFilterSingleStation,
+ DataHandlerSingleStation, DefaultDataHandler, remove=["self", "id_class"])
+ assert sorted(obj.requirements()) == reqs
class TestDataHandlerSeparationOfScalesSingleStation:
--
GitLab