Skip to content

Commit d86a0a1

Browse files
committed
Add (commented out) stacking tests for solve()
1 parent 75ca73a commit d86a0a1

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

array_api_tests/test_linalg.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,15 @@ def _x2_shapes(draw):
539539
@pytest.mark.xp_extension('linalg')
540540
@given(*solve_args())
541541
def test_solve(x1, x2):
542-
linalg.solve(x1, x2)
542+
res = linalg.solve(x1, x2)
543+
544+
# TODO: This requires an upstream fix to ndindex
545+
# (https://github.com/Quansight-Labs/ndindex/pull/131)
546+
547+
# if x2.ndim == 1:
548+
# _test_stacks(linalg.solve, x1, x2, res=res, dims=1)
549+
# else:
550+
# _test_stacks(linalg.solve, x1, x2, res=res, dims=2)
543551

544552
@pytest.mark.xp_extension('linalg')
545553
@given(

0 commit comments

Comments
 (0)