Skip to content

Commit 8d1ded4

Browse files
committed
Fix the input strategies to test_matmul
1 parent 28c0242 commit 8d1ded4

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

array_api_tests/test_linalg.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,13 +238,12 @@ def test_inv(x):
238238
assert res.shape == x.shape, "inv() did not return the correct shape"
239239
assert res.dtype == x.dtype, "inv() did not return the correct dtype"
240240

241-
_test_stacks(_array_module.linalg.inv, x, {}, res)
241+
_test_stacks(_array_module.linalg.inv, x, res=res)
242242

243243
# TODO: Test that the result is actually the inverse
244244

245245
@given(
246-
x1=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),
247-
x2=xps.arrays(dtype=xps.floating_dtypes(), shape=shapes),
246+
*two_mutual_arrays(numeric_dtype_objects)
248247
)
249248
def test_matmul(x1, x2):
250249
if (x1.shape == () or x2.shape == ()

0 commit comments

Comments
 (0)