a
    lh                     @   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sJJ t t ttt} | t tttksnJ 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sJ d S )Nr            r      )r   
isinstancer   r   r   r   r   r   r   eyer   expr r   O/var/www/auris/lib/python3.9/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sXJ |  dkshJ t | | dd} | jd
ksJ |  dksJ t tdtddgddggdd} | jd
ksJ |  dksJ 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s6J d S )N)r   r   r   r	   r
   r   r   r   r   r   r   test_TensorProduct_getitem7   s    
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   