Merge pull request #240 from asmeurer/test_eye-fix

rgommers · web-flow · commit 8fb5e01ee797 · 2024-03-07T09:37:04.000+01:00
Avoid using .size in test_eye
diff --git a/array_api_tests/test_creation_functions.py b/array_api_tests/test_creation_functions.py
@@ -359,7 +359,7 @@ def test_eye(n_rows, n_cols, kw):
         [[1 if j - i == k else 0 for j in range(_n_cols)] for i in range(n_rows)],
         dtype=out.dtype  # Note: dtype already checked above.
     )
-    if expected.size == 0:
+    if 0 in expected.shape:
         expected = xp.reshape(expected, (n_rows, _n_cols))
     ph.assert_array_elements("eye", out=out, expected=expected, kw=kw)
 

Original file line number	Diff line number	Diff line change
`@@ -359,7 +359,7 @@ def test_eye(n_rows, n_cols, kw):`
`359`	`359`	`[[1 if j - i == k else 0 for j in range(_n_cols)] for i in range(n_rows)],`
`360`	`360`	`dtype=out.dtype # Note: dtype already checked above.`
`361`	`361`	`)`
`362`		`- if expected.size == 0:`
	`362`	`+ if 0 in expected.shape:`
`363`	`363`	`expected = xp.reshape(expected, (n_rows, _n_cols))`
`364`	`364`	`ph.assert_array_elements("eye", out=out, expected=expected, kw=kw)`
`365`	`365`