Skip to content

Commit b2a663b

Browse files
committed
WIP/squash: also test reshape(..., copy=None)
1 parent abdd97f commit b2a663b

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

array_api_tests/test_manipulation_functions.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -428,15 +428,16 @@ def test_repeat(x, kw, data):
428428
@given(
429429
x=hh.arrays(dtype=hh.all_dtypes, shape=reshape_shape),
430430
shape=hh.reshape_shapes(reshape_shape),
431-
kw=hh.kwargs(copy=st.booleans())
431+
kw=hh.kwargs(copy=st.booleans() | st.none())
432432
)
433433
def test_reshape(x, shape, kw):
434434
repro_snippet = ph.format_snippet(f"xp.reshape({x!r},{shape!r}, **kw) with {kw =}")
435-
try:
435+
try:
436436
out = xp.reshape(x, shape, **kw)
437437

438438
ph.assert_dtype("reshape", in_dtype=x.dtype, out_dtype=out.dtype)
439-
# TODO: test copy
439+
440+
print(x.shape, shape, kw) # XXX remove
440441

441442
_shape = list(shape)
442443
if any(side == -1 for side in shape):

0 commit comments

Comments
 (0)