test: Benchmark to be run on codspeed (#740)

Author: DeaMariaLeon

benchmarks/test_benchmark_coo.py

@@ -1,15 +1,38 @@
+import itertools
+import operator
+
 import sparse
 
+import pytest
+
 import numpy as np
 
+DENSITY = 0.01
+SEED = 42
+
+
+def elemwise_test_name(param):
+    side, rank = param
+    return f"{side=}-{rank=}"
+
+
+@pytest.fixture(scope="module", params=itertools.product([100, 500, 1000], [1, 2, 3, 4]), ids=elemwise_test_name)
+def elemwise_args(request):
+    side, rank = request.param
+    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)