Skip to content
Snippets Groups Projects
Commit 699228d9 authored by Felix Kleinert's avatar Felix Kleinert
Browse files

#87 update multiply extremes and tests

parent 00acb7bd
No related branches found
No related tags found
3 merge requests!90WIP: new release update,!89Resolve "release branch / CI on gpu",!82Felix issue087 bug multiply extreme none check
...@@ -447,6 +447,12 @@ class DataPrep(object): ...@@ -447,6 +447,12 @@ class DataPrep(object):
if True only extract values larger than extreme_values if True only extract values larger than extreme_values
:param timedelta: used as arguments for np.timedelta in order to mark extreme values on datetime :param timedelta: used as arguments for np.timedelta in order to mark extreme values on datetime
""" """
# check if labels or history is None
if (self.label is None) or (self.history is None):
logging.debug(f"{self.station} has `None' labels, skip multiply extremes")
return
# check type if inputs # check type if inputs
extreme_values = helpers.to_list(extreme_values) extreme_values = helpers.to_list(extreme_values)
for i in extreme_values: for i in extreme_values:
......
...@@ -445,6 +445,20 @@ class TestDataPrep: ...@@ -445,6 +445,20 @@ class TestDataPrep:
assert upsampled.shape[2] == sum([f(orig, gt, 1), f(orig, gt, 2)]) assert upsampled.shape[2] == sum([f(orig, gt, 1), f(orig, gt, 2)])
assert f(upsampled, lt, -1) == 0 assert f(upsampled, lt, -1) == 0
def test_multiply_extremes_none_label(self, data):
data.transform("datetime")
data.make_history_window("variables", 3, "datetime")
# data.make_labels("variables", "o3", "datetime", 2)
data.label = None
assert data.multiply_extremes([1], extremes_on_right_tail_only=False) is None
def test_multiply_extremes_none_history(self,data ):
data.transform("datetime")
data.make_history_window("variables", 3, "datetime")
# data.make_labels("variables", "o3", "datetime", 2)
data.label = None
assert data.multiply_extremes([1], extremes_on_right_tail_only=False) is None
def test_get_extremes_history(self, data): def test_get_extremes_history(self, data):
data.transform("datetime") data.transform("datetime")
data.make_history_window("variables", 3, "datetime") data.make_history_window("variables", 3, "datetime")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment