Skip to content

Commit 6388c11

Browse files
committed
Increase benchmark size
1 parent 6af5d9a commit 6388c11

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_stats.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -371,8 +371,8 @@ def test_stats_benchmark(
371371
axis: Literal[0, 1] | None,
372372
dtype: type[np.float32 | np.float64],
373373
) -> None:
374-
shape = (10_000, 10_000) if "sparse" in array_type.mod else (1000, 1000)
375-
arr = array_type.random(shape, dtype=dtype)
374+
n = 100_000 if "sparse" in array_type.mod else 10_000
375+
arr = array_type.random((n, n), dtype=dtype)
376376

377377
func(arr, axis=axis) # warmup: numba compile
378378
benchmark(func, arr, axis=axis)

0 commit comments

Comments
 (0)