Skip to content

Commit 31e3813

Browse files
committed
corrected rng.integers & changed func name
1 parent edf1f86 commit 31e3813

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

benchmarks/test_benchmark_coo.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,12 +29,12 @@ def bench():
2929
x @ y
3030

3131

32-
def id_of_test(param):
32+
def get_test_id(param):
3333
side, rank = param
3434
return f"{side=}-{rank=}"
3535

3636

37-
@pytest.fixture(params=itertools.product([100, 500, 1000], [1, 2, 3, 4]), ids=id_of_test)
37+
@pytest.fixture(params=itertools.product([100, 500, 1000], [1, 2, 3, 4]), ids=get_test_id)
3838
def elemwise_args(request, seed, max_size):
3939
side, rank = request.param
4040
if side**rank >= max_size:
@@ -77,7 +77,7 @@ def bench():
7777
f(x, y)
7878

7979

80-
@pytest.fixture(params=itertools.product([100, 500, 1000], [1, 2, 3]), ids=id_of_test)
80+
@pytest.fixture(params=itertools.product([100, 500, 1000], [1, 2, 3]), ids=get_test_id)
8181
def indexing_args(request, seed, max_size):
8282
side, rank = request.param
8383
if side**rank >= max_size:
@@ -117,7 +117,7 @@ def test_index_fancy(benchmark, indexing_args, seed):
117117
side = x.shape[0]
118118
rank = x.ndim
119119
rng = np.random.default_rng(seed=seed)
120-
index = rng.integers((side // 2,) * rank)
120+
index = rng.integers(0, side, size=(side // 2,) * rank)
121121

122122
x[index] # Numba compilation
123123

0 commit comments

Comments
 (0)