U
    ZÊ¹j  ã                   @   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t‚tt ttƒt ƒs&t‚t t ttƒtƒ} | tt t ksJt‚t t ttƒtƒ} | t tttƒksnt‚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sÚt‚d S )Nr   é   é   é   r   éÿÿÿÿé   )r   ÚAssertionErrorÚ
isinstancer   r   r   r   r   r   r   Úeyer   ©Úexpr© r   ú`/var/www/html/bwave/python/venv/lib/python3.8/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t‚|  ¡ dks,t‚t tddgƒtttgƒdd�} | jd	ksXt‚|  ¡ dksht‚t | | dd�} | jd
ks„t‚|  ¡ dks”t‚t t d¡tddgddggƒdd�} | jd
ksÆt‚|  ¡ dksÖt‚d S )Nr   r   F)Úevaluater   r   r   r   )r   r   )r   r   r   r   r   )	r   Úshaper   Z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s6t‚d S )N)r   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   