test: Benchmark to be run on codspeed

benchmarks/test_benchmark_coo.py

@@ -1,15 +1,32 @@
+import operator
+
 import sparse
 
+import pytest
+
 import numpy as np
 
+DENSITY = 0.01
+SEED = 42
+
+
+@pytest.fixture(scope="module", side=[100, 500, 1000], rank=[1, 2, 3, 4])
+def elemwise_args(request):
+    side, rank = request.side, request.rank
+    if side**rank >= 2**26:
+        pytest.skip()
+    rng = np.random.default_rng(seed=SEED)
+    shape = (side,) * rank
+    x = sparse.random(shape, density=DENSITY, random_state=rng)
+    y = sparse.random(shape, density=DENSITY, random_state=rng)
+    return x, y
 
-def test_matmul(benchmark):
-    rng = np.random.default_rng(seed=42)
-    x = sparse.random((40, 40), density=0.01, random_state=rng)
-    y = sparse.random((40, 40), density=0.01, random_state=rng)
 
-    x @ y  # Numba compilation
+@pytest.mark.parametrize("f", [operator.add, operator.mul])
+def test_elemwise(benchmark, f, elemwise_args):
+    x, y = elemwise_args
+    f(x, y)
 
     @benchmark
-    def test_matmul():
-        x @ y
+    def bench():
+        f(x, y)