Reduce test time

Author: lshaw8317

tests/ndarray/test_lazyexpr.py

@@ -16,16 +16,16 @@
 from blosc2.lazyexpr import ne_evaluate
 from blosc2.ndarray import get_chunks_idx, npvecdot
 
-NITEMS_SMALL = 1_000
-NITEMS = 10_000
+NITEMS_SMALL = 100
+NITEMS = 1000
 
 
 @pytest.fixture(params=[np.float32, np.float64])
 def dtype_fixture(request):
     return request.param
 
 
-@pytest.fixture(params=[(NITEMS_SMALL,), (NITEMS,), (NITEMS // 100, 100)])
+@pytest.fixture(params=[(NITEMS_SMALL,), (NITEMS,), (NITEMS // 10, 100)])
 def shape_fixture(request):
     return request.param
 
@@ -36,7 +36,7 @@ def shape_fixture(request):
         (True, True),
         (True, False),
         pytest.param((False, True), marks=pytest.mark.heavy),
-        (False, False),
+        pytest.param((False, False), marks=pytest.mark.heavy),
     ]
 )
 def chunks_blocks_fixture(request):
@@ -1811,9 +1811,9 @@ def test_simpleproxy(xp, dtype):
         npa = blosc_matrix[()]
         res = (npb & npa) | np.logical_not(npb)
     else:
-        N = 10
+        N = 5
         shape_a = (N, N, N)
-        blosc_matrix = blosc2.full(shape=shape_a, fill_value=3, dtype=np.dtype(dtype), chunks=(N // 3,) * 3)
+        blosc_matrix = blosc2.full(shape=shape_a, fill_value=3, dtype=np.dtype(dtype), chunks=(N // 2,) * 3)
         foreign_matrix = xp.ones(shape_a, dtype=dtype_)
         if dtype == "complex128":
             foreign_matrix += 0.5j

tests/ndarray/test_lazyexpr_fields.py

@@ -12,8 +12,8 @@
 import blosc2
 from blosc2.lazyexpr import ne_evaluate
 
-NITEMS_SMALL = 1_000
-NITEMS = 10_000
+NITEMS_SMALL = 100
+NITEMS = 1000
 
 
 @pytest.fixture(
@@ -41,7 +41,9 @@ def dtype_fixture(request):
     return request.param
 
 
-@pytest.fixture(params=[(NITEMS_SMALL,), (NITEMS,), (NITEMS // 100, 100)])
+@pytest.fixture(
+    params=[(NITEMS_SMALL,), (NITEMS,), pytest.param((NITEMS // 10, 100), marks=pytest.mark.heavy)]
+)
 def shape_fixture(request):
     return request.param
 
@@ -52,7 +54,7 @@ def shape_fixture(request):
         (True, True),
         (True, False),
         pytest.param((False, True), marks=pytest.mark.heavy),
-        (False, False),
+        pytest.param((False, False), marks=pytest.mark.heavy),
     ]
 )
 def chunks_blocks_fixture(request):

tests/ndarray/test_linalg.py

@@ -16,7 +16,7 @@
         ((12, 10), (7, 5), (3, 3)),
         ((10,), (9,), (7,)),
         ((0,), (0,), (0,)),
-        ((40, 10, 10), (2, 3, 4), (1, 2, 2)),
+        ((4, 10, 10), (2, 3, 4), (1, 2, 2)),
     },
 )
 @pytest.mark.parametrize(
@@ -25,9 +25,9 @@
         ((10,), (4,), (2,)),
         ((10, 5), (3, 4), (1, 3)),
         ((10, 12), (2, 4), (1, 2)),
-        ((200, 10, 22), (23, 2, 4), (4, 1, 2)),
+        ((3, 10, 3), (2, 2, 4), (1, 1, 2)),
         ((0,), (0,), (0,)),
-        ((20, 40, 10, 10), (5, 2, 3, 4), (2, 1, 2, 2)),
+        ((6, 3, 10, 10), (5, 2, 3, 4), (2, 1, 2, 2)),
     },
 )
 @pytest.mark.parametrize(
@@ -302,13 +302,13 @@ def test_matmul_disk():
         ),
         # 5Dx5D->no reduce
         (
-            (3, 4, 5, 6, 7),
-            (2, 3, 4, 5, 6),
-            (1, 2, 2, 3, 3),
-            (5, 6, 7, 4, 8),
-            (4, 5, 6, 3, 7),
-            (2, 3, 3, 2, 4),
-            (2, 3, 4, 5, 6, 2, 3, 3, 2, 4),  # output dims = 10
+            (1, 2, 1, 5, 3),
+            (1, 1, 1, 2, 2),
+            (1, 1, 1, 1, 1),
+            (2, 3, 2, 1, 5),
+            (1, 2, 1, 1, 3),
+            (1, 2, 1, 1, 1),
+            (1, 2, 1, 2, 2, 2, 1, 2, 1, 3),  # output dims = 10
             ([], []),
         ),
     ],

tests/ndarray/test_ndarray.py

@@ -294,13 +294,13 @@ def test_oindex():
     np.testing.assert_allclose(arr[:], nparr)
 
 
-@pytest.mark.parametrize("c", [None, 10])
+@pytest.mark.parametrize("c", [None, 3])
 def test_fancy_index(c):
     # Test 1d
     ndim = 1
     chunks = (c,) * ndim if c is not None else None
     dtype = np.dtype("float")
-    d = 1 + int(1000 / dtype.itemsize) if c is None else 50
+    d = 1 + int(1000 / dtype.itemsize) if c is None else 10
     shape = (d,) * ndim
     arr = blosc2.linspace(0, 100, num=np.prod(shape), shape=shape, dtype=dtype, chunks=chunks)
     rng = np.random.default_rng()
@@ -319,7 +319,7 @@ def test_fancy_index(c):
     chunks = (c,) * ndim if c is not None else None
     arr = blosc2.linspace(0, 100, num=np.prod(shape), shape=shape, dtype=dtype, chunks=chunks)
     rng = np.random.default_rng()
-    idx = rng.integers(low=-d, high=d, size=(100,))  # mix of +ve and -ve indices
+    idx = rng.integers(low=-d, high=d, size=(30,))  # mix of +ve and -ve indices
 
     row = idx
     col = rng.permutation(idx)