a kh @sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZe edd d Zee efedd d Ze ed ddZe ddddZe ddddZdS)Eq)Expr)Integer)BooleanAnd) MatrixExpr) ShapeError)Union)argsreturncGshtdd|D\}}tgddt|dd|ddDddt|dd|ddDRS)aReturn the symbolic condition how ``MatAdd``, ``HadamardProduct`` makes sense. Parameters ========== args The list of arguments of matrices to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_matadd_valid >>> m, n, p, q = symbols('m n p q') >>> A = MatrixSymbol('A', m, n) >>> B = MatrixSymbol('B', p, q) >>> is_matadd_valid(A, B) Eq(m, p) & Eq(n, q) css|] }|jVqdSNshape.0argrO/var/www/auris/lib/python3.9/site-packages/sympy/matrices/expressions/_shape.py z"is_matadd_valid..css|]\}}t||VqdSr rrijrrrr"rNcss|]\}}t||VqdSr rrrrrr#rziprr rowscolsrrris_matadd_valid s ""r!cGs>tdd|D\}}tddt|dd|ddDS)aReturn the symbolic condition how ``MatMul`` makes sense Parameters ========== args The list of arguments of matrices and scalar expressions to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_matmul_valid >>> m, n, p, q = symbols('m n p q') >>> A = MatrixSymbol('A', m, n) >>> B = MatrixSymbol('B', p, q) >>> is_matmul_valid(A, B) Eq(n, p) css|]}t|tr|jVqdSr ) isinstancerrrrrrr=rz"is_matmul_valid..css|]\}}t||VqdSr rrrrrr>rNrrrrrrris_matmul_valid'sr#)rr cCst|j|jS)aReturn the symbolic condition how the matrix is assumed to be square Parameters ========== arg The matrix to be tested for. Examples ======== >>> from sympy import MatrixSymbol, symbols >>> from sympy.matrices.expressions._shape import is_square >>> m, n = symbols('m n') >>> A = MatrixSymbol('A', m, n) >>> is_square(A) Eq(m, n) )rrr )rrrr is_squareAsr$NcGsjtdd|D\}}tttdd|dkr>td|tttdd|dkrftd|dS) z:Validate matrix shape for addition only for integer valuescss|] }|jVqdSr r)rxrrrrZrz*validate_matadd_integer..cSst|ttfSr r"intrr%rrr[rz)validate_matadd_integer..rz!Matrices have mismatching shape: cSst|ttfSr r&r(rrrr)]rN)rlensetfilterr rrrrvalidate_matadd_integerXs r-cGsft|dd|ddD]F\}}|j|j}}t|ttfrt|ttfr||krtd||qdS)z@Validate matrix shape for multiplication only for integer valuesNrrzMatrices are not aligned)rr rr"r'rr )r ABrrrrrvalidate_matmul_integeras"$r0)Zsympy.core.relationalrZsympy.core.exprrZsympy.core.numbersrZsympy.logic.boolalgrrZ"sympy.matrices.expressions.matexprrZsympy.matrices.exceptionsr typingr r!r#r$r-r0rrrrs