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