Submit
Path:
~
/
/
opt
/
alt
/
python37
/
lib64
/
python3.7
/
site-packages
/
numpy
/
core
/
tests
/
__pycache__
/
File Content:
test_einsum.cpython-37.pyc
B <�Fd�� � @ s� d dl mZmZmZ d dlZd dlmZmZm Z m Z m Z m Z m Z mZ dZe�dddddddd dddg �Zi Zx eee�D ]\ZZeee< qxW G d d � d e�ZG d d � d e�Zedkr�e� dS )� )�division�absolute_import�print_functionN)�TestCase�run_module_suite�assert_� assert_equal�assert_array_equal�assert_almost_equal� assert_raises�suppress_warningsZ abcdefghij� � � � � c @ s e Zd Zdd� Zdd� ZdEdd�Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDS )F� TestEinSumc C s �x dD �]}t ttj|d� t ttjd|d� t ttjdd|d� t ttjddd|d� t ttjddd|d� t ttjddd |d � t ttjddd |d � t ttjddd|d � t ttjfd�d|i� t ttjddd|d� t ttjdddgdg|d� t ttjddg|d� t ttjdd|d� t ttjdddg|d� t ttjdd|d� t ttjdddg|d� t ttjdd|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjdddg|d� t ttjd ddg|d� t ttjd ddg|d� t ttjd ddg|d� t ttjd ddgddgg|d� t ttjd t�d!��d"d#�|d� t ttjd$t�d!��d"d#�|d� t ttjdt�d!��d"d#�|d� t ttjd%dd&gdd&ggt�d'��d"d"�|d� qW d S )(N)TF)�optimize� r Ztest)�outr �W)�orderr Zblah)�castingr Z bad_data_type)�dtyper )Zbad_argr )?NNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNNr �,�i�ijz...izi...jzi...zij...zi..z.i...zj->..jzj->.j...zi%...z...j$zi->&zi->ijzij->jij�iir r r zii->izi->i� r )r � ValueError�np�einsum� TypeError�arange�reshape)�self�do_opt� r' �O/opt/alt/python37/lib64/python3.7/site-packages/numpy/core/tests/test_einsum.py�test_einsum_errors sV z TestEinSum.test_einsum_errorsc s� �x�dD �]v}t �d�� d� _t jd� |d�}t|j� k� t j� tg|d�}t|j� k� t jd� |d�}t|j� k� t|� � t j� ddg|d�}t|j� k� t|� � t jd� |d�}t|jd � d � jd < t jd� |d�}t|jd � t �d�� d� _t jd � |d�}t|j� k� t|� j � t j� ddg|d�}t|j� k� t|� j � t �d �� d � _t jd� |d�}t|j� k� t|� fdd�t d�D �� t j� ddgdg|d�}t|j� k� t|� fdd�t d�D �� t �d�� d� _t jd� |d�}t|j� k� t|dd� � D �� t j� tddgtdg|d�}t|j� k� t|dd� � D �� t jd� |d�}t|j� k� t|dd� � � ddd�D �� t j� ddtgtdg|d�}t|j� k� t|dd� � � ddd�D �� t jd � |d�}t|j� k� t|� fd d�t d�D �� t j� tddgdtg|d�}t|j� k� t|� fd d�t d�D �� t jd � |d�}t|j� k� t|� fd d�t d�D �� t j� dddgddg|d�}t|j� k� t|� fd!d�t d�D �� t jd"� |d�}t|j� k� t|� fd#d�t d�D �� t j� ddtgdtg|d�}t|j� k� t|� fd$d�t d�D �� t jd%� |d�}t|j� k� t|� fd&d�t d�D �� t j� dtdgdtg|d�}t|j� k� t|� fd'd�t d�D �� t jd(� |d�}t|j� k� t|d)d� � � ddd�D �� t j� dtdgtdg|d�}t|j� k� t|d*d� � � ddd�D �� t �d�� d� _t jd+� |d�}t|j� k� t|� fd,d�t d�D �� t j� dddgdg|d�}t|j� k� t|� fd-d�t d�D �� t �d.�� d/� _t jd0� |d�}t|j� k� t|� � dd�� t j� dddgdddg|d�}t|j� k� t|� � dd�� qW d S )1N)TFr )r r z...)r r r r Z WRITEABLEFZji� )r r zii->ic s g | ]}� ||f �qS r' r' )�.0r )�ar' r( � <listcomp>� s z0TestEinSum.test_einsum_views.<locals>.<listcomp>r c s g | ]}� ||f �qS r' r' )r+ r )r, r' r( r- � s � )r r r z ...ii->...ic s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )�xr' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )�range)r+ r' )r/ r( r- � s c s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )r/ r' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )r0 )r+ r' )r/ r( r- � s z ii...->...ic s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )r/ r' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )r0 )r+ r' )r/ r( r- � s r c s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )r/ r' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )r0 )r+ r' )r/ r( r- � s z ...ii->i...c s g | ]}� d d �||f �qS )Nr' )r+ r )r, r' r( r- � s c s g | ]}� d d �||f �qS )Nr' )r+ r )r, r' r( r- � s zjii->ijc s g | ]}� d d �||f �qS )Nr' )r+ r )r, r' r( r- � s c s g | ]}� d d �||f �qS )Nr' )r+ r )r, r' r( r- � s z ii...->i...c s( g | ] }� � d dd�dd�||f �qS )r r r N)� transpose)r+ r )r, r' r( r- � s c s( g | ] }� � d dd�dd�||f �qS )r r r N)r1 )r+ r )r, r' r( r- � s z i...i->i...c s( g | ] }� � d dd�dd�||f �qS )r r r N)r1 )r+ r )r, r' r( r- � s c s( g | ] }� � d dd�dd�||f �qS )r r r N)r1 )r+ r )r, r' r( r- � s z i...i->...ic s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )r/ r' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )r0 )r+ r' )r/ r( r- � s c s" g | ]� � fd d�t d�D ��qS )c s g | ]}� ||f �qS r' r' )r+ r )r/ r' r( r- � s z;TestEinSum.test_einsum_views.<locals>.<listcomp>.<listcomp>r )r0 )r+ r' )r/ r( r- � s ziii->ic s g | ]}� |||f �qS r' r' )r+ r )r, r' r( r- � s c s g | ]}� |||f �qS r' r' )r+ r )r, r' r( r- � s � )r r r zijk->jik) r r# �shaper! r �base�Ellipsisr �flags�Tr0 r1 Zswapaxes)r% r& �br' )r, r( �test_einsum_views^ s� z TestEinSum.test_einsum_viewsFc C s. xpt dd�D ]b}tj||d�}ttjd||d�tj|dd��|�� ttj|dgg |d�tj|dd��|�� q W x�t dd�D ]t}tjd | |d��d d |�}ttjd ||d�tj|dd��|�� ttj|tdgtg|d�tj|dd��|�� q~W x�t dd�D ]t}tjd | |d��d |�}ttjd ||d�tj|dd��|�� ttj|dtgtg|d�tj|dd��|�� �qW x�t dd�D ]v}tjd | |d��d d |�}ttjd ||d�tj|dd��|�� ttj|dtgtg|d�tj|dd��|�� �q�W xvt dd�D ]h}tj|| |d��||�}ttjd||d�t� |��|�� ttj|ddg|d�t� |��|�� �q W tt�dd d�d� x�t dd�D ]�}tjd | |d��d |�}tjd | |d��d d |�}ttjd|||d�t� ||�� ttj|tg|tg|d�t� ||�� �q�W x�t dd�D ]z}tjd | |d��d d |�}tj||d�}ttjd|||d�t� ||�� ttj|tdg|tdg|d�t� ||�� �q*W x�t dd�D ]�}tj|d d |d��|d d �}tj||d�}ttjd|||d�t� |j |j �j � ttj|dtg|dtg|d�t� |j |j �j � �q�W x~t dd�D ]p}tjd |d�d }tj||d�d }ttjd|||d�t� ||�� ttj|dg|dg|d�t� ||�� �qNW t� ���}|�tj� �x t dd�D �] }tjd| |d��d|�}tj||d�}ttjd|||d�t�||�� ttj|ddg|dg|d�t�||�� tjd|d�}tjd|||dd|d� t|t�|�d�|�d���|�� d|d< tj|ddg|dg|dd|d� t|t�|�d�|�d���|�� �q�W �x<t dd�D �],}tjd| |d��d|�}tj||d�}ttjd |j |j |d�t�|j |j �� ttj|j ddg|j dg|d�t�|j |j �� tjd|d�}tjd |j |j |dd|d� t|t�|j �d�|j �d���|�� d|d< tj|j ddg|j dg|dd|d� t|t�|j �d�|j �d���|�� �qW x�t dd�D ]�}|d k �s\|d k�rDtjd| |d��d|�}tj|d |d��|d �}ttjd |||d�t�||�� ttj|ddg|dd g|d�t�||�� �qDW x�t dd�D ]�}tjd| |d��d|�}tj|d |d��|d �}tjd |d��dd �}tjd |||dd|d� t|t�|�d�|�d���|�� d|d< tj|ddg|dd g|dd|d� t|t�|�d�|�d���|�� �q�W tjd|d��d d�}tjd!|d��dd"�}tjd#|d��d"d �}|d k� rrttjd$||||d�|�|��|�� ttj|ddg|dd g|d d g|d�|�|��|�� tjd%|d��d d �}tjd$||||dd|d� |�d��|�d��} | �|�d���|�} t|| � d|d< tj|ddg|dd g|d d g|dd|d� |�d��|�d��} | �|�d���|�} t|| � t�|�t�d �k� r�tjd&|d��d dd"�}tjd |d��dd d �}tt�d'||�tj||ddgddgfd(�� tt�|ddd g|ddd gd d g�tj||ddgddgfd(�� tjd)|d��d"d �}tjd*|||dd|d� t|tj|�d�|�d�ddgddgfd(��|�� d|d< tj|ddd g|ddd gd d g|dd|d� t|tj|�d�|�d�ddgddgfd(��|�� W d Q R X tjdd d+ddd,ddg|d�}tjdd-d.d+ddd dg|d�}t�d/d/d0d/d/d0d/d/g�}ttjd1|||d2d|d3�t�t�|dk|dk�|dk�� ttj|dg|dg|dgdgd2dd4� t�t�|dk|dk�|dk�� tjd5|d�}tt�d6d |�d t�|� � tt�d g |dgg �d t�|� � tt�d7|d �d t�|� � tt�|dgd g g �d t�|� � �x.t dd8�D �] }tj||d�}t�|�jdk� r<ttjd9|||d�t� ||�� ttjd:|||d�t�||�� ttjd;|d |d�d | � ttjd<d ||d�d | � ttjd7|d |d�d t�|� � ttjd6d ||d�d t�|� � ttjd9|dd � |d d� |d�t� |dd � |d d� �� ttjd:|dd � |d d� |d�t�|dd � |d d� �� ttjd;|dd � d |d�d |dd � � ttjd<d |dd � |d�d |dd � � ttjd7|dd � d |d�d t�|dd � � � ttjd6d |dd � |d�d t�|dd � � � � q<W tjd5td�}tjd||dd4�}t|t�|�� t|jt�|�� tj|dgg |dd4�}t|t�|�� t|jt�|�� t�d �d } t�d��d d �d } t�d��d d �d= } tt�d>| | | �d?� d S )@Nr � )r zi->)r ���)Zaxisr r r r z ...i->...z i...->...r z..., ...r � z ...i, ...i� z i..., i...zi,jzij, jzij,j�f8Zunsafe)r r r r .zji,j� �f2zij,jkr2 � r � zij,jk,kl� �< zijk, jil -> kl)�axes� z ijk,jil->kl���� g @g TFzi,i,i->i�?)r r r )r r r* z,i->zi,->� z...,...zi,izi,->iz,i->i� z z,mz,zm->� )r0 r r# r r! �sum�astyper$ r5 ZtraceZmultiply�innerr7 Zouterr �filterZComplexWarning�dotr � tensordot�arrayZ logical_and�itemsize�object) r% r r& �nr, r8 Zsup�c�d�tgt�p�q�rr' r' r( �check_einsum_sums� s� " & " " $ $ """ " ,, z TestEinSum.check_einsum_sumsc C s | � d� d S )NZi1)r] )r% r' r' r( �test_einsum_sums_int8� s z TestEinSum.test_einsum_sums_int8c C s | � d� d S )NZu1)r] )r% r' r' r( �test_einsum_sums_uint8� s z!TestEinSum.test_einsum_sums_uint8c C s | � d� d S )NZi2)r] )r% r' r' r( �test_einsum_sums_int16� s z!TestEinSum.test_einsum_sums_int16c C s | � d� d S )NZu2)r] )r% r' r' r( �test_einsum_sums_uint16� s z"TestEinSum.test_einsum_sums_uint16c C s | � d� | � dd� d S )NZi4T)r] )r% r' r' r( �test_einsum_sums_int32� s z!TestEinSum.test_einsum_sums_int32c C s | � d� | � dd� d S )NZu4T)r] )r% r' r' r( �test_einsum_sums_uint32� s z"TestEinSum.test_einsum_sums_uint32c C s | � d� d S )NZi8)r] )r% r' r' r( �test_einsum_sums_int64� s z!TestEinSum.test_einsum_sums_int64c C s | � d� d S )N�u8)r] )r% r' r' r( �test_einsum_sums_uint64� s z"TestEinSum.test_einsum_sums_uint64c C s | � d� d S )Nr@ )r] )r% r' r' r( �test_einsum_sums_float16� s z#TestEinSum.test_einsum_sums_float16c C s | � d� d S )NZf4)r] )r% r' r' r( �test_einsum_sums_float32 s z#TestEinSum.test_einsum_sums_float32c C s | � d� | � dd� d S )Nr> T)r] )r% r' r' r( �test_einsum_sums_float64 s z#TestEinSum.test_einsum_sums_float64c C s | � tj� d S )N)r] r Z longdouble)r% r' r' r( �test_einsum_sums_longdouble s z&TestEinSum.test_einsum_sums_longdoublec C s | � d� | � dd� d S )NZc8T)r] )r% r' r' r( �test_einsum_sums_cfloat64 s z$TestEinSum.test_einsum_sums_cfloat64c C s | � d� d S )NZc16)r] )r% r' r' r( �test_einsum_sums_cfloat128 s z%TestEinSum.test_einsum_sums_cfloat128c C s | � tj� d S )N)r] r Z clongdouble)r% r' r' r( � test_einsum_sums_clongdouble s z'TestEinSum.test_einsum_sums_clongdoublec C s� t �d�}t �d�}tt �d||�dgdggg� tt jd||dd�dgdggg� t �dt j�}t �dt j�}tt �d |||�t �d ||�� tt jd |||dd�t jd ||dd�� t �d d �}t �d d ��dd�}t �d d��dd�}tt �d|||�d d gd dgd dgddggddgddgddgddggg� tt jd|||dd�d d gd dgd dgddggddgddgddgddggg� d S )N)r r )r r r zij...,j...->i...r T)r )r r< r r r )r r< r= zijklm,ijn,ijn->z ijklm,ijn->r r r r* r z x,yx,zx->xzy� rK � r? � � r2 �0 �@ )r �onesr r! Zint64r# r$ )r% r, r8 rW r' r' r( �test_einsum_misc s& " zTestEinSum.test_einsum_miscc C s� t �d��ddd�}t �d�}t �d||�}tt �d||�|� tt �d||�|� tt �d||�|� tt jd||d d �|� tt jd||d d �|� tt jd||d d �|� t �d ��d �}t �d ��d�}t �d||�}tt �d||�|� tt �d||�|� tt �d||�|� tt �d||�|� tt jd||d d �|� tt jd||d d �|� tt jd||d d �|� tt jd||d d �|� ddddg}t �t �|���|�}t �|d �}t �d||�}tt �d||�|� tt �d||�|� tt �d||�|� tt jd||d d �|� tt jd||d d �|� tt jd||d d �|� d\}}} t �|| | ��ddd||| �}t �|| | d ��||| d�}t �d||�}tt �d ||�|� tt jd ||d d �|� d S ) Nr2 r r r z ijk,j->ijkzij...,j...->ij...zij...,...j->ij...zij...,j->ij...T)r r< )r r r )r r z ik,kj->ijzik...,k...->i...zik...,...kj->i...jz...k,kjz ik,k...->i...r z ijkl,k->ijlzijkl,kz...kl,kz ...kl,k...)� rv �x r z...lmn,...lmno->...oz...lmn,lmno->...o)r r# r$ r! r Zprod) r% �A�B�ref�dimsr, �v�J�K�Mr' r' r( �test_einsum_broadcast1 sJ "" z TestEinSum.test_einsum_broadcastc C s� t �d��dd��t j�}t �d��ddd��t j�}t �d||�}t j||dd�}t||� t �d ��dd��t j �}t �d ��ddd d ��t j�}t �d ||�}t j||dd�}t||� d S ) Nr r r i@ i� z cl, cpx->lpx)r r )rE r* i � rs zcl, cpxy->lpxy) r r# r$ rN �float32Zint16r! rR r Zfloat64)r% rx ry Zes�tpr' r' r( �test_einsum_fixedstridebugf s z%TestEinSum.test_einsum_fixedstridebugc C sj t j�ddd�}t �d�}t jd||d� t �d�}||d d �d f |d d �d f ||f }t||� d S )Nr r )r r r r )r r zaabb->ab)r r )r �randomZnormal�zerosr! r# r )r% r/ Zy1�idxZy2r' r' r( � test_einsum_fixed_collapsingbug� s (z*TestEinSum.test_einsum_fixed_collapsingbugc C s� t �d�}t �d�d d d� }t �d�}d|d d d�< t jd||||d� t|j|� t �d�d d d� }t jd||||d� t|j|� t �d�d }t �d�}d|d < t �d t j�}t jd |||d� t|j|� d S ) N)r r rF r r z mi,mi,mi->m)r z im,im,im->m)r r r ).r )r r z ij,jk->ik)r rt r! r r4 r� )r% r/ r Z correct_baser' r' r( �(test_einsum_all_contig_non_contig_output� s z3TestEinSum.test_einsum_all_contig_non_contig_outputc C sh t jdt jd�d d� }d|d< t jdt jd�d d� }t jdt jd�}t jd|||d�}t||� d S ) N)rp r r )r r T.)r r r z...ij,...jk->...ik)r )r r� Zbool_rt r! r )r% r, r rY Zresr' r' r( �test_small_boolean_arrays� s z$TestEinSum.test_small_boolean_arraysc C s� |g}|� d�d � d�}x,|D ]$}dd� |D �}|�tjj|� � q W tj|ddi�}tj|ddi�}t||� tj|dd i�}t||� d S ) Nz->r r c S s g | ] }t | �qS r' )�global_size_dict)r+ r/ r' r' r( r- � s z/TestEinSum.optimize_compare.<locals>.<listcomp>r F�greedy�optimal)�split�appendr r� �randr! r )r% �string�operands�terms�termr{ �noopt�optr' r' r( �optimize_compare� s zTestEinSum.optimize_comparec C s | � d� | � d� d S )Nz a,ab,abc->abcz a,b,ab->ab)r� )r% r' r' r( �test_hadamard_like_products� s z&TestEinSum.test_hadamard_like_productsc C s" | � d� | � d� | � d� d S )Nzea,fb,gc,hd,abcd->efghzea,fb,abcd,gc,hd->efghzabcd,ea,fb,gc,hd->efgh)r� )r% r' r' r( �test_index_transformations� s z%TestEinSum.test_index_transformationsc C sT | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� d S )Nz$acdf,jbje,gihb,hfac,gfac,gifabc,hfacz cd,bdhe,aidb,hgca,gc,hgibcd,hgaczabhe,hidj,jgba,hiab,gabz bde,cdh,agdb,hica,ibd,hgicd,hiacz chd,bde,agbc,hiad,hgc,hgi,hiadz chd,bde,agbc,hiad,bdi,cgh,agdbzbdhe,acad,hiab,agac,hibd)r� )r% r' r' r( � test_complex� s zTestEinSum.test_complexc C s@ | � d� | � d� | � d� | � d� | � d� | � d� d S )Nz ab,ab,c->z ab,ab,c->cz ab,ab,cd,cd->zab,ab,cd,cd->aczab,ab,cd,cd->cdzab,ab,cd,cd,ef,ef->)r� )r% r' r' r( � test_collapse� s zTestEinSum.test_collapsec C s@ | � d� | � d� | � d� | � d� | � d� | � d� d S )Nzab,cd,ef->abcdefzab,cd,ef->acdfzab,cd,de->abcdez ab,cd,de->bezab,bcd,cd->abcdzab,bcd,cd->abd)r� )r% r' r' r( � test_expand� s zTestEinSum.test_expandc C s| | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d � | � d � | � d � | � d � d S ) Nz eb,cb,fb->cefzdd,fb,be,cdb->cefzbca,cdb,dbf,afc->zdcc,fce,ea,dbf->abzfdf,cdd,ccd,afe->aezabcd,adzed,fcd,ff,bcf->bezbaa,dcf,af,cde->bezbd,db,eac->acezfff,fae,bef,def->abdzefc,dbc,acf,fd->abez ba,ac,da->bcd)r� )r% r' r' r( �test_edge_cases� s zTestEinSum.test_edge_casesc C s6 | � d� | � d� | � d� | � d� | � d� d S )Nzab,abzab,bazabc,abczabc,baczabc,cba)r� )r% r' r' r( �test_inner_product� s z TestEinSum.test_inner_productc C sr | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d � | � d � | � d � d S ) Nzaab,fa,df,ecc->bdezecb,fef,bad,ed->aczbcf,bbb,fbf,fc->z bb,ff,be->ezbcb,bb,fc,fff->zfbb,dfd,fc,fc->zafd,ba,cc,dc->bfzadb,bc,fa,cfc->dzbbd,bda,fc,db->acfzdba,ead,cad->bcezaef,fbc,dca->bde)r� )r% r' r' r( �test_random_cases� s z TestEinSum.test_random_casesN)F)$�__name__� __module__� __qualname__r) r9 r] r^ r_ r` ra rb rc rd rf rg rh ri rj rk rl rm ru r� r� r� r� r� r� r� r� r� r� r� r� r� r� r' r' r' r( r sF L | 5 r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestEinSumPathc C sL |g}|� d�d � d�}x,|D ]$}dd� |D �}|�tjj|� � q W |S )Nz->r r c S s g | ] }t | �qS r' )r� )r+ r/ r' r' r( r- s z1TestEinSumPath.build_operands.<locals>.<listcomp>)r� r� r r� r� )r% r� r� r� r� r{ r' r' r( �build_operands s z TestEinSumPath.build_operandsc C sp t |�t |�k}t|� xJtt |�d �D ]6}|t||d t�M }|||d ||d kM }q*W t|� d S )Nr )�lenr r0 � isinstance�tuple)r% �compZ benchmarkZret�posr' r' r( �assert_path_equal s z TestEinSumPath.assert_path_equalc C s� | � d�}tj|ddi�\}}| �|ddg� tj|ddi�\}}| �|ddg� | � d�}tj|ddi�\}}| �|ddg� tj|ddi�\}}| �|ddg� d S ) Nz a,b,c->abcr )r� r � einsum_path)r r r )r� r zacdf,jbje,gihb,hfac)r r r r )r� r r� r� )r% Z outer_test�path�path_strZ long_testr' r' r( �test_memory_contraints! s z%TestEinSumPath.test_memory_contraintsc C s� | � d�}tj|ddi�\}}| �|ddddddd g� tj|dd i�\}}| �|dd d dd dd g� | � d�}tj|ddi�\}}| �|dd dd ddd g� tj|dd i�\}}| �|dddd ddd g� d S )Nz$acdf,jbje,gihb,hfac,gfac,gifabc,hfacr r� r� )r r )r r )r r )r r )r r r� )r r )r r )r r )r r z chd,bde,agbc,hiad,bdi,cgh,agdb)r r )r r )r� r r� r� )r% Z long_test1r� r� Z long_test2r' r' r( �test_long_paths3 s z TestEinSumPath.test_long_pathsc C sh | � d�}tj|ddi�\}}| �|dddg� tj|ddi�\}}| �|dddg� | � d�}tj|ddi�\}}| �|dd ddg� tj|ddi�\}}| �|dd ddg� | � d �}tj|ddi�\}}| �|dd ddg� tj|ddi�\}}| �|dd ddg� | � d �}tj|ddi�\}}| �|dd ddg� tj|ddi�\}}| �|dd ddg� d S ) Nz eb,cb,fb->cefr r� r� )r r )r r r� zdd,fb,be,cdb->cef)r r zbca,cdb,dbf,afc->)r r zdcc,fce,ea,dbf->ab)r� r r� r� )r% Z edge_test1r� r� Z edge_test2Z edge_test3Z edge_test4r' r' r( �test_edge_pathsJ s( z TestEinSumPath.test_edge_pathsc C s� | � d�}tj|ddi�\}}| �|ddg� tj|ddi�\}}| �|dddd g� dddd g}tj|d|i�\}}| �||� tj|ddi�}tj|d|i�}t||� d S ) Nzdcc,fce,ea,dbf->abr Fr� )r r r r T)r r )r r )r r )r� r r� r� r! r )r% Z path_testr� r� Zexp_pathr� r� r' r' r( �test_path_type_inputm s z#TestEinSumPath.test_path_type_inputN) r� r� r� r� r� r� r� r� r� r' r' r' r( r� s #r� �__main__)Z __future__r r r Znumpyr Z numpy.testingr r r r r r r r �charsrS Zsizesr� �zip�size�charr r� r� r' r' r' r( �<module> s" ( u
Edit
Rename
Chmod
Delete
FILE
FOLDER
Name
Size
Permission
Action
test_abc.cpython-37.opt-1.pyc
2107 bytes
0644
test_abc.cpython-37.pyc
2107 bytes
0644
test_api.cpython-37.opt-1.pyc
11505 bytes
0644
test_api.cpython-37.pyc
11505 bytes
0644
test_arrayprint.cpython-37.opt-1.pyc
10986 bytes
0644
test_arrayprint.cpython-37.pyc
10986 bytes
0644
test_datetime.cpython-37.opt-1.pyc
49163 bytes
0644
test_datetime.cpython-37.pyc
49163 bytes
0644
test_defchararray.cpython-37.opt-1.pyc
29206 bytes
0644
test_defchararray.cpython-37.pyc
29206 bytes
0644
test_deprecations.cpython-37.opt-1.pyc
20438 bytes
0644
test_deprecations.cpython-37.pyc
20438 bytes
0644
test_dtype.cpython-37.opt-1.pyc
21522 bytes
0644
test_dtype.cpython-37.pyc
21647 bytes
0644
test_einsum.cpython-37.opt-1.pyc
26207 bytes
0644
test_einsum.cpython-37.pyc
26207 bytes
0644
test_errstate.cpython-37.opt-1.pyc
1914 bytes
0644
test_errstate.cpython-37.pyc
1914 bytes
0644
test_extint128.cpython-37.opt-1.pyc
6702 bytes
0644
test_extint128.cpython-37.pyc
6702 bytes
0644
test_function_base.cpython-37.opt-1.pyc
12187 bytes
0644
test_function_base.cpython-37.pyc
12337 bytes
0644
test_getlimits.cpython-37.opt-1.pyc
5695 bytes
0644
test_getlimits.cpython-37.pyc
5695 bytes
0644
test_half.cpython-37.opt-1.pyc
13558 bytes
0644
test_half.cpython-37.pyc
13558 bytes
0644
test_indexerrors.cpython-37.opt-1.pyc
7545 bytes
0644
test_indexerrors.cpython-37.pyc
7545 bytes
0644
test_indexing.cpython-37.opt-1.pyc
41938 bytes
0644
test_indexing.cpython-37.pyc
41938 bytes
0644
test_item_selection.cpython-37.opt-1.pyc
3473 bytes
0644
test_item_selection.cpython-37.pyc
3473 bytes
0644
test_longdouble.cpython-37.opt-1.pyc
7461 bytes
0644
test_longdouble.cpython-37.pyc
7461 bytes
0644
test_machar.cpython-37.opt-1.pyc
1389 bytes
0644
test_machar.cpython-37.pyc
1389 bytes
0644
test_mem_overlap.cpython-37.opt-1.pyc
26795 bytes
0644
test_mem_overlap.cpython-37.pyc
26833 bytes
0644
test_memmap.cpython-37.opt-1.pyc
7512 bytes
0644
test_memmap.cpython-37.pyc
7589 bytes
0644
test_multiarray.cpython-37.opt-1.pyc
227544 bytes
0644
test_multiarray.cpython-37.pyc
227633 bytes
0644
test_nditer.cpython-37.opt-1.pyc
69947 bytes
0644
test_nditer.cpython-37.pyc
69947 bytes
0644
test_numeric.cpython-37.opt-1.pyc
91401 bytes
0644
test_numeric.cpython-37.pyc
91401 bytes
0644
test_numerictypes.cpython-37.opt-1.pyc
13866 bytes
0644
test_numerictypes.cpython-37.pyc
13866 bytes
0644
test_print.cpython-37.opt-1.pyc
7470 bytes
0644
test_print.cpython-37.pyc
7470 bytes
0644
test_records.cpython-37.opt-1.pyc
15572 bytes
0644
test_records.cpython-37.pyc
15572 bytes
0644
test_regression.cpython-37.opt-1.pyc
87286 bytes
0644
test_regression.cpython-37.pyc
87286 bytes
0644
test_scalarinherit.cpython-37.opt-1.pyc
1902 bytes
0644
test_scalarinherit.cpython-37.pyc
1902 bytes
0644
test_scalarmath.cpython-37.opt-1.pyc
21079 bytes
0644
test_scalarmath.cpython-37.pyc
21079 bytes
0644
test_scalarprint.cpython-37.opt-1.pyc
1257 bytes
0644
test_scalarprint.cpython-37.pyc
1257 bytes
0644
test_shape_base.cpython-37.opt-1.pyc
17242 bytes
0644
test_shape_base.cpython-37.pyc
17242 bytes
0644
test_ufunc.cpython-37.opt-1.pyc
45351 bytes
0644
test_ufunc.cpython-37.pyc
45351 bytes
0644
test_umath.cpython-37.opt-1.pyc
94853 bytes
0644
test_umath.cpython-37.pyc
94853 bytes
0644
test_umath_complex.cpython-37.opt-1.pyc
14807 bytes
0644
test_umath_complex.cpython-37.pyc
14807 bytes
0644
test_unicode.cpython-37.opt-1.pyc
13276 bytes
0644
test_unicode.cpython-37.pyc
13276 bytes
0644
N4ST4R_ID | Naxtarrr
