fixed a test

rwijtvliet · Nov 15, 2024 · 99b9d43 · 99b9d43
1 parent af91bd5
commit 99b9d43
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/tests/tools/test_frames.py b/tests/tools/test_frames.py
@@ -6,10 +6,25 @@
 from portfolyo import dev, testing, tools
 
 
+@pytest.fixture(
+    params=[
+        [1, 2, 3, 4, 25, 7, 8],
+        [1, 2, 3, 4, nan, 7, 8],
+        [1, 2, 3, 4, nan, 7, 8],
+        [1, 2, 3, 4, nan, 7, 8],
+        [3, 2, 1, nan, nan, 7, 8],
+        [3, 2, 1, nan, nan, 7, 8],
+        [3, 2, 1, nan, nan, 7, 8],
+    ]
+)
+def values(request):
+    return request.param
+
+
 @pytest.mark.parametrize(
     ("values", "maxgap", "gapvalues"),
     [
-        ([1, 2, 3, 4, 25, 7, 8], 1, []),
+        ([1, 2, 3, 4, 25, 7, 8], 1, None),
         ([1, 2, 3, 4, nan, 7, 8], 1, [5.5]),
         ([1, 2, 3, 4, nan, 7, 8], 2, [5.5]),
         ([1, 2, 3, 4, nan, 7, 8], 3, [5.5]),