def test_only_squares(): A = MatrixSymbol('A', n, m) C = MatrixSymbol('C', n, n) D = MatrixSymbol('D', n, n) assert only_squares(C) == [C] assert only_squares(C, D) == [C, D] assert only_squares(C, A, A.T, D) == [C, A * A.T, D]
def test_only_squares(): A = MatrixSymbol('A', n, m) C = MatrixSymbol('C', n, n) D = MatrixSymbol('D', n, n) assert only_squares(C) == [C] assert only_squares(C, D) == [C, D] assert only_squares(C, A, A.T, D) == [C, A*A.T, D]
def test_only_squares(): assert only_squares(C) == [C] assert only_squares(C, D) == [C, D] assert only_squares(C, A, A.T, D) == [C, A * A.T, D]
def test_only_squares(): assert only_squares(C) == [C] assert only_squares(C, D) == [C, D] assert only_squares(C, A, A.T, D) == [C, A*A.T, D]