Modified tests to run faster. (#5)

sun-data · May 26, 2024 · 23dd8b3 · 23dd8b3
1 parent c1b4522
commit 23dd8b3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ndfilters/tests/__init__.py → ndfilters/_tests/__init__.py b/ndfilters/tests/__init__.py → ndfilters/_tests/__init__.py
diff --git a/ndfilters/tests/test_trimmed_mean.py → ndfilters/_tests/test_trimmed_mean.py b/ndfilters/tests/test_trimmed_mean.py → ndfilters/_tests/test_trimmed_mean.py
@@ -8,14 +8,14 @@
 @pytest.mark.parametrize(
     argnames="array",
     argvalues=[
-        np.random.random(16),
-        np.random.random((16, 17)),
-        np.random.random((16, 17, 18)),
+        np.random.random(5),
+        np.random.random((5, 6)),
+        np.random.random((5, 6, 7)),
     ],
 )
 @pytest.mark.parametrize(
     argnames="kernel_shape",
-    argvalues=[5, (5,), (5, 6), (5, 6, 7)],
+    argvalues=[2, (3,), (3, 4), (3, 4, 5)],
 )
 @pytest.mark.parametrize(
     argnames="axis",