Skip to content

Commit a905183

Browse files
committed
Fix matrix variable shapes in matmul test
Corrects the shapes of matrix variables in test_matrix_matmul_return_type to ensure proper 2D matrix multiplication and type assertion.
1 parent ad51377 commit a905183

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_matrix_variable.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -406,6 +406,6 @@ def test_matrix_matmul_return_type():
406406
assert type(x[:, None] @ x[None, :]) is MatrixExpr
407407

408408
# test 2D @ 2D → 2D
409-
y = m.addMatrixVar((3, 4))
410-
z = m.addMatrixVar((2, 3))
409+
y = m.addMatrixVar((2, 3))
410+
z = m.addMatrixVar((3, 4))
411411
assert type(y @ z) is MatrixExpr

0 commit comments

Comments
 (0)