Make super fast benchmarks bigger

Author: flying-sheep

tests/test_stats.py

@@ -207,8 +207,19 @@ def test_stats_benchmark(
     axis: Literal[0, 1, None],
     dtype: type[np.float32 | np.float64],
 ) -> None:
+    density = 0.1 if func is stats.is_constant and is_along_major_axis(array_type, axis) else 0.01
+
     shape = (10_000, 10_000) if "sparse" in array_type.mod else (1000, 1000)
-    arr = array_type.random(shape, dtype=dtype)
+    arr = array_type.random(shape, dtype=dtype, density=density)
 
     func(arr, axis=axis)  # warmup: numba compile
     benchmark(func, arr, axis=axis)
+
+
+def is_along_major_axis(array_type: ArrayType[Any], axis: Literal[0, 1, None]) -> bool:
+    if axis is None or not (array_type.flags & Flags.Sparse):
+        return False
+    cls = array_type.inner.cls if array_type.inner else array_type.cls
+    if not issubclass(cls, types.CSBase):
+        return False
+    return (axis == 1) is (cls.format == "csr")