Skip to content

Commit 0f6a31a

Browse files
committed
MAINT: only draw signed ints for repeats in test_repeat
Strictly speaking, the spec requires all integers. However uints make us skip the test on _all_ -compat-wrapped libraries.
1 parent 705089d commit 0f6a31a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

array_api_tests/test_manipulation_functions.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -328,8 +328,10 @@ def test_repeat(x, kw, data):
328328
axis = kw.get("axis", None)
329329
size = math.prod(shape) if axis is None else shape[axis]
330330
repeat_strat = st.integers(1, 10)
331+
# TODO: draw `repeats` from hh.all_int_types to include unsigned ints
332+
# That however breaks down on torch (unit8) and numpy (uint64)
331333
repeats = data.draw(repeat_strat
332-
| hh.arrays(dtype=hh.all_int_dtypes, elements=repeat_strat,
334+
| hh.arrays(dtype=hh.int_dtypes, elements=repeat_strat,
333335
shape=st.sampled_from([(1,), (size,)])),
334336
label="repeats")
335337
if isinstance(repeats, int):

0 commit comments

Comments
 (0)