Skip to content

Commit 4465128

Browse files
committed
BUG: work around jax/hypothesis combination being unreasonable
1 parent c79116c commit 4465128

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

array_api_tests/test_utility_functions.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -65,10 +65,12 @@ def test_any(x, data):
6565
out=result, expected=expected, kw=kw)
6666

6767

68+
# NB: hh.int_dtypes instead of hh.numeric_dtypes because of
69+
# https://github.com/data-apis/array-api-tests/issues/368
6870
@pytest.mark.unvectorized
6971
@pytest.mark.min_version("2024.12")
7072
@given(
71-
x=hh.arrays(hh.numeric_dtypes, hh.shapes(min_dims=1, min_side=1)),
73+
x=hh.arrays_no_scalars(hh.int_dtypes, hh.shapes(min_dims=1, min_side=1)),
7274
data=st.data(),
7375
)
7476
def test_diff(x, data):
@@ -103,7 +105,7 @@ def test_diff(x, data):
103105
@pytest.mark.min_version("2024.12")
104106
@pytest.mark.unvectorized
105107
@given(
106-
x=hh.arrays(hh.numeric_dtypes, hh.shapes(min_dims=1, min_side=1)),
108+
x=hh.arrays(hh.int_dtypes, hh.shapes(min_dims=1, min_side=1)),
107109
data=st.data(),
108110
)
109111
def test_diff_append_prepend(x, data):

0 commit comments

Comments
 (0)