| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677 |
- """
- We have a few different kind of Matrices
- Matrix, ImmutableMatrix, MatrixExpr
- Here we test the extent to which they cooperate
- """
- from sympy.core.symbol import symbols
- from sympy.matrices import (Matrix, MatrixSymbol, eye, Identity,
- ImmutableMatrix)
- from sympy.matrices.expressions import MatrixExpr, MatAdd
- from sympy.matrices.common import classof
- from sympy.testing.pytest import raises
- SM = MatrixSymbol('X', 3, 3)
- SV = MatrixSymbol('v', 3, 1)
- MM = Matrix([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
- IM = ImmutableMatrix([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
- meye = eye(3)
- imeye = ImmutableMatrix(eye(3))
- ideye = Identity(3)
- a, b, c = symbols('a,b,c')
- def test_IM_MM():
- assert isinstance(MM + IM, ImmutableMatrix)
- assert isinstance(IM + MM, ImmutableMatrix)
- assert isinstance(2*IM + MM, ImmutableMatrix)
- assert MM.equals(IM)
- def test_ME_MM():
- assert isinstance(Identity(3) + MM, MatrixExpr)
- assert isinstance(SM + MM, MatAdd)
- assert isinstance(MM + SM, MatAdd)
- assert (Identity(3) + MM)[1, 1] == 6
- def test_equality():
- a, b, c = Identity(3), eye(3), ImmutableMatrix(eye(3))
- for x in [a, b, c]:
- for y in [a, b, c]:
- assert x.equals(y)
- def test_matrix_symbol_MM():
- X = MatrixSymbol('X', 3, 3)
- Y = eye(3) + X
- assert Y[1, 1] == 1 + X[1, 1]
- def test_matrix_symbol_vector_matrix_multiplication():
- A = MM * SV
- B = IM * SV
- assert A == B
- C = (SV.T * MM.T).T
- assert B == C
- D = (SV.T * IM.T).T
- assert C == D
- def test_indexing_interactions():
- assert (a * IM)[1, 1] == 5*a
- assert (SM + IM)[1, 1] == SM[1, 1] + IM[1, 1]
- assert (SM * IM)[1, 1] == SM[1, 0]*IM[0, 1] + SM[1, 1]*IM[1, 1] + \
- SM[1, 2]*IM[2, 1]
- def test_classof():
- A = Matrix(3, 3, range(9))
- B = ImmutableMatrix(3, 3, range(9))
- C = MatrixSymbol('C', 3, 3)
- assert classof(A, A) == Matrix
- assert classof(B, B) == ImmutableMatrix
- assert classof(A, B) == ImmutableMatrix
- assert classof(B, A) == ImmutableMatrix
- raises(TypeError, lambda: classof(A, C))
|
