Skip to content

Commit 72d1d0c

Browse files
authored
Merge pull request #424 from ev-br/test_repeat_no_unsigned
MAINT: only draw signed ints for `repeats` in test_repeat
2 parents 705089d + 0f6a31a commit 72d1d0c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

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)