[TEST] New test for special cases

Author: ricardosp4

tests/ndarray/test_matmul.py

@@ -6,22 +6,22 @@
 
 @pytest.mark.parametrize(
     ("ashape", "achunks", "ablocks"),
-    [
+    {
         ((12, 10), (7, 5), (3, 3)),
         ((10,), (9,), (7,)),
-    ],
+    },
 )
 @pytest.mark.parametrize(
     ("bshape", "bchunks", "bblocks"),
-    [
+    {
         ((10,), (4,), (2,)),
         ((10, 5), (3, 4), (1, 3)),
         ((10, 12), (2, 4), (1, 2)),
-    ],
+    },
 )
 @pytest.mark.parametrize(
     "dtype",
-    [np.float32, np.float64, np.complex64, np.complex128],
+    {np.float32, np.float64, np.complex64, np.complex128},
 )
 def test_matmul(ashape, achunks, ablocks, bshape, bchunks, bblocks, dtype):
     a = blosc2.linspace(0, 10, dtype=dtype, shape=ashape, chunks=achunks, blocks=ablocks)
@@ -37,19 +37,19 @@ def test_matmul(ashape, achunks, ablocks, bshape, bchunks, bblocks, dtype):
 
 @pytest.mark.parametrize(
     ("ashape", "achunks", "ablocks"),
-    [
+    {
         ((12, 11), (7, 5), (3, 1)),
         ((0, 0), (0, 0), (0, 0)),
         ((10,), (4,), (2,)),
-    ],
+    },
 )
 @pytest.mark.parametrize(
     ("bshape", "bchunks", "bblocks"),
-    [
+    {
         ((1, 5), (1, 4), (1, 3)),
         ((4, 6), (2, 4), (1, 3)),
         ((5,), (4,), (2,)),
-    ],
+    },
 )
 def test_matmul_shapes(ashape, achunks, ablocks, bshape, bchunks, bblocks):
     a = blosc2.linspace(0, 10, shape=ashape, chunks=achunks, blocks=ablocks)
@@ -64,7 +64,7 @@ def test_matmul_shapes(ashape, achunks, ablocks, bshape, bchunks, bblocks):
 
 @pytest.mark.parametrize(
     "scalar",
-    [
+    {
         5,  # int
         5.3,  # float
         1 + 2j,  # complex
@@ -74,7 +74,7 @@ def test_matmul_shapes(ashape, achunks, ablocks, bshape, bchunks, bblocks):
         np.float64(5.3),  # NumPy float64
         np.complex64(1 + 2j),  # NumPy complex64
         np.complex128(1 + 2j),  # NumPy complex128
-    ],
+    },
 )
 def test_matmul_scalars(scalar):
     vector = blosc2.asarray(np.array([1, 2, 3]))
@@ -113,3 +113,30 @@ def test_matmul_dims(ashape, bshape):
 
     with pytest.raises(ValueError):
         blosc2.matmul(b, a)
+
+
+@pytest.mark.parametrize(
+    ("ashape", "achunks", "ablocks", "adtype"),
+    {
+        ((7, 10), (7, 5), (3, 5), np.float32),
+        ((10,), (9,), (7,), np.complex64),
+    },
+)
+@pytest.mark.parametrize(
+    ("bshape", "bchunks", "bblocks", "bdtype"),
+    {
+        ((10,), (4,), (2,), np.float64),
+        ((10, 6), (9, 4), (2, 3), np.complex128),
+        ((10, 12), (2, 4), (1, 2), np.complex128),
+    },
+)
+def test_matmul_especial_cases(ashape, achunks, ablocks, adtype, bshape, bchunks, bblocks, bdtype):
+    a = blosc2.linspace(0, 10, dtype=adtype, shape=ashape, chunks=achunks, blocks=ablocks)
+    b = blosc2.linspace(0, 10, dtype=bdtype, shape=bshape, chunks=bchunks, blocks=bblocks)
+    blosc2_res = blosc2.matmul(a, b)
+
+    na = a[:]
+    nb = b[:]
+    np_res = np.matmul(na, nb)
+
+    np.testing.assert_allclose(blosc2_res, np_res, rtol=1e-6)