From 326d94fc726009bae01773c9694ffb7e8212033b Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Thu, 21 Jul 2022 15:19:29 +0200
Subject: [PATCH] updated tests
---
test/test_data_handler/test_iterator.py | 1 -
test/test_run_modules/test_model_setup.py | 6 ++++++
2 files changed, 6 insertions(+), 1 deletion(-)
diff --git a/test/test_data_handler/test_iterator.py b/test/test_data_handler/test_iterator.py
index fe740c09..b5fb30a9 100644
--- a/test/test_data_handler/test_iterator.py
+++ b/test/test_data_handler/test_iterator.py
@@ -331,7 +331,6 @@ class TestPermuteData:
class TestFProc:
-
@pytest.fixture
def collection(self):
coll = []
diff --git a/test/test_run_modules/test_model_setup.py b/test/test_run_modules/test_model_setup.py
index 6e8d3ea9..83fe409f 100644
--- a/test/test_run_modules/test_model_setup.py
+++ b/test/test_run_modules/test_model_setup.py
@@ -140,6 +140,12 @@ class DummyData:
def __init__(self, number_of_samples=np.random.randint(100, 150)):
self.number_of_samples = number_of_samples
+ def __len__(self, upsampling=False):
+ if upsampling is False:
+ return self._len
+ else:
+ return self._len_upsampling
+
def get_X(self, upsampling=False, as_numpy=True):
X1 = np.random.randint(0, 10, size=(self.number_of_samples, 14, 1, 5)) # samples, window, variables
X2 = np.random.randint(21, 30, size=(self.number_of_samples, 10, 1, 2)) # samples, window, variables
--
GitLab