Skip to content

Commit fa42c24

Browse files
authored
FIX: larger inputs for sorting tests - to test sorts stability (#390)
Reviewed at #390
1 parent 0cbcedb commit fa42c24

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

array_api_tests/test_sorting_functions.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,9 @@ def assert_scalar_in_set(
3333
@given(
3434
x=hh.arrays(
3535
dtype=hh.real_dtypes,
36-
shape=hh.shapes(min_dims=1, min_side=1),
36+
shape=hh.shapes(min_dims=1, min_side=1, max_side=50),
37+
# argsort needs to be tested on big enough arrays, hence we set `max_side=50`
38+
# see https://github.com/data-apis/array-api-tests/issues/389
3739
elements={"allow_nan": False},
3840
),
3941
data=st.data(),
@@ -94,7 +96,9 @@ def test_argsort(x, data):
9496
@given(
9597
x=hh.arrays(
9698
dtype=hh.real_dtypes,
97-
shape=hh.shapes(min_dims=1, min_side=1),
99+
shape=hh.shapes(min_dims=1, min_side=1, max_side=50),
100+
# sort needs to be tested on big enough arrays, hence we set `max_side=50`
101+
# see https://github.com/data-apis/array-api-tests/issues/389
98102
elements={"allow_nan": False},
99103
),
100104
data=st.data(),

0 commit comments

Comments
 (0)