From fe30a3d827ecf886b56823022be1f31c31476017 Mon Sep 17 00:00:00 2001
From: Bert Palm <bert.palm@ufz.de>
Date: Mon, 22 Mar 2021 13:24:48 +0100
Subject: [PATCH] fixed wrapper-test

---
 tests/funcs/test_harm_funcs.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/tests/funcs/test_harm_funcs.py b/tests/funcs/test_harm_funcs.py
index a83368090..3b1ec42c6 100644
--- a/tests/funcs/test_harm_funcs.py
+++ b/tests/funcs/test_harm_funcs.py
@@ -150,13 +150,19 @@ def test_gridInterpolation(data, method):
         interpolate(data, field, flagger, freq, order=10, method=method, downcast_interpolation=True)
 
 
-def test_wrapper(data):
+@pytest.mark.parametrize('func, kws', [
+    ('linear', dict(to_drop=None)),
+    ('shift', dict(method="nshift", to_drop=None)),
+    ('interpolate', dict(method="spline")),
+    ('aggregate', dict(value_func=np.nansum, method="nagg", to_drop=None)),
+])
+def test_wrapper(data, func, kws):
     # we are only testing, whether the wrappers do pass processing:
     field = 'data'
     freq = "15min"
     flagger = initFlagsLike(data)
 
-    linear(data, field, flagger, freq, to_drop=None)
-    aggregate(data, field, flagger, freq, value_func=np.nansum, method="nagg", to_drop=None)
-    shift(data, field, flagger, freq, method="nshift", to_drop=None)
-    interpolate(data, field, flagger, freq, method="spline")
+    import saqc
+    func = getattr(saqc.funcs, func)
+    func(data, field, flagger, freq, **kws)
+
-- 
GitLab