o
    GZh                     @   s   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZ d dlmZmZmZmZ edddZed	ddZed
ddZdd Zdd Zdd ZdS )    )TensorProduct)Matrix)MatrixSymbol)Array)xyz)ijklA   BCc                  C   s   t dddks	J tt ttt sJ t t ttt} | tt t ks%J t t ttt} | t tttks7J t t	dt
ddgddgg} | t
ddgddggddgddgggddgddggddgddggggksmJ d S )Nr            r      )r   
isinstancer   r   r   r   r   r   r   eyer   expr r   P/var/www/auris/lib/python3.10/site-packages/sympy/tensor/tests/test_functions.pytest_TensorProduct_construction   s    r   c                  C   s   t dddd} | jdksJ |  dksJ t tddgtttgdd} | jd	ks,J |  dks4J t | | dd} | jd
ksBJ |  dksJJ t tdtddgddggdd} | jd
kscJ |  dkskJ d S )Nr   r   F)evaluater   r   r   r   )r   r   )r   r   r   r   r   )r   shapeZrankr   r   r   r   r   r   r   r   r   test_TensorProduct_shape%   s   $r   c                  C   s:   t tt} | ttttf tttf tttf  ksJ d S )N)r   r   r   r	   r
   r   r   r   r   r   r   test_TensorProduct_getitem7   s   
0r    N)Zsympy.tensor.functionsr   Zsympy.matrices.denser   Z"sympy.matrices.expressions.matexprr   Zsympy.tensor.arrayr   Z	sympy.abcr   r   r   r	   r
   r   r   r   r   r   r   r   r    r   r   r   r   <module>   s    