Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyo
�
� 9Zc @` s� d d l m Z m Z m Z d d l Z d d l Z d d l m Z m Z m
Z
m Z m Z m
Z
d e f d � � YZ e d k r� e � n d S( i ( t divisiont absolute_importt print_functionN( t TestCaset run_module_suitet assert_t assert_equalt assert_array_equalt
assert_raisest
TestEinSumc B` s� e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z
d � Z d
� Z d � Z
d � Z d
� Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s� t t t j � t t t j d � t t t j d d � t t t j d d d d �t t t j d d d d �t t t j d d d d �t t t j d d d d
�t t t j d d d d �t t t j d% d � t t t j d d d � t t t j d
d d g d g � t t t j d
d g � t t t j d d � t t t j d d d g � t t t j d d � t t t j d d d g � t t t j d d � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g � t t t j d d d g d d g g � t t t j d t j d � j d d � � t t t j d! t j d � j d d � � t t t j d t j d � j d d � � t t t j d" d d# g d d# g g d t j d$ � j d d � �d S(& Nt i t outt testt ordert Wt castingt blaht dtypet
bad_data_typet bad_argi? t ,t it ijs ...is i...js i...s ij...s i..s .i...s j->..js j->.j...s i%...s ...j$s i->&s i->ijs ij->jijt iii i i s ii->is i->ii i ( N( R t
ValueErrort npt einsumt TypeErrort Nonet aranget reshape( t self( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_errors s@ " (+++'c C` sk
t j d � } d | _ t j d | � } t | j | k � t j | t g � } t | j | k � t j d | � } t | j | k � t | | � t j | d d g � } t | j | k � t | | � t j d | � } t | j d � t | j d <t j d | � } t | j d � t j d � } d | _ t j d | � } t | j | k � t | | j
� t j | d d g � } t | j | k � t | | j
� t j d
� } d | _ t j d | � } t | j | k � t | g t d � D] } | | | f ^ q�� t j | d d g d g � } t | j | k � t | g t d � D] } | | | f ^ q\� t j d � } d | _ t j d
| � } t | j | k � t | g | D]/ } g t d � D] } | | | f ^ q�^ q�� t j | t d d g t d g � } t | j | k � t | g | D]/ } g t d � D] } | | | f ^ qP^ q=� t j d | � } t | j | k � t | g | j d d d � D]/ } g t d � D] } | | | f ^ q�^ q�� t j | d d t g t d g � } t | j | k � t | g | j d d d � D]/ } g t d � D] } | | | f ^ qP^ q=� t j d | � } t | j | k � t | g t d � D] } | d d � | | f ^ q�� t j | t d d g d t g � } t | j | k � t | g t d � D] } | d d � | | f ^ q� t j d | � } t | j | k � t | g t d � D] } | d d � | | f ^ qy� t j | d d d g d d g � } t | j | k � t | g t d � D] } | d d � | | f ^ q�� t j d | � } t | j | k � t | g t d � D]. } | j d d d � d d � | | f ^ qG� t j | d d t g d t g � } t | j | k � t | g t d � D]. } | j d d d � d d � | | f ^ q�� t j d | � } t | j | k � t | g t d � D]. } | j d d d � d d � | | f ^ q3� t j | d t d g d t g � } t | j | k � t | g t d � D]. } | j d d d � d d � | | f ^ q�� t j d | � } t | j | k � t | g | j d d d � D]/ } g t d � D] } | | | f ^ q;^ q(� t j | d t d g t d g � } t | j | k � t | g | j d d d � D]/ } g t d � D] } | | | f ^ q�^ q�� t j d � } d | _ t j d | � } t | j | k � t | g t d � D] } | | | | f ^ q7 � t j | d d d g d g � } t | j | k � t | g t d � D] } | | | | f ^ q� � t j d � } d | _ t j d | � } t | j | k � t | | j
d d � � t j | d d d g d d d g � } t | j | k � t | | j
d d � � d S( Ni i i s ...R i i t WRITEABLEt jii s ii->ii s ...ii->...is ii...->...is ...ii->i...s jii->ijs ii...->i...s i...i->i...s i...i->...is iii->ii i s ijk->jik( i i ( i i ( i i ( i i i ( i i i ( i i i ( R R t shapeR R t baset EllipsisR t flagst Falset Tt ranget transposet swapaxes( R t at bR t x( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_viewsM s�
0 0 C$C I$ I9$99$9H$HH$H I$ I 3!3 'c C` s� x� t d d � D]� } t j | d | �} t t j d | � t j | d d �j | � � t t j | d g g � t j | d d �j | � � q Wx� t d d � D]� } t j d= | d | �j d d | � } t t j d
| � t j | d d �j | � � t t j | t d g t g � t j | d d �j | � � q� Wx� t d d � D]� } t j d | d | �j d | � } t t j d | � t j | d d �j | � � t t j | d t g t g � t j | d d �j | � � qWWx� t d d � D]� } t j d> | d | �j d d | � } t t j d | � t j | d d �j | � � t t j | d t g t g � t j | d d �j | � � qWx� t d d � D]� } t j | | d | �j | | � } t t j d | � t j | � j | � � t t j | d d g � t j | � j | � � q�Wt t j d
d d � d � x� t d d � D]� } t j d | d | �j d | � } t j d? | d | �j d d | � } t t j d
| | � t j
| | � � t t j | t g | t g � t j
| | � � qkWx� t d d � D]� } t j d@ | d | �j d d | � } t j | d | �} t t j d | | � t j | | � � t t j | t d g | t d g � t j | | � � q+Wx� t d d � D]� } t j | d d d | �j | d d � } t j | d | �} t t j d | | � t j | j | j � j � t t j | d t g | d t g � t j | j | j � j � q�Wx� t d d � D]� } t j d d | �d } t j | d | �d } t t j d | | � t j
| | � � t t j | d g | d g � t j
| | � � q�Wt j � �� t j d t j � x�t d d � D]y} t j d | d | �j d | � } t j | d | �} t t j d | | � t j | | � � t t j | d d g | d g � t j | | � � t j d d | �} t j d | | d | d d d d �t | t j | j d � | j d � � j | � � d | d <t j | d d g | d g d | d d d d �t | t j | j d � | j d � � j | � � qrWx�t d d � D]�} t j d | d | �j d | � } t j | d | �} t t j d | j | j � t j | j | j � � t t j | j d d g | j d g � t j | j | j � � t j d d | �} t j d | j | j d | d d d d �t | t j | j j d � | j j d � � j | � � d | d <t j | j d d g | j d g d | d d d d �t | t j | j j d � | j j d � � j | � � q�Wx� t d d � D]� } | d k s� | d k r� t j d | d | �j d | � } t j | d d | �j | d � } t t j d | | � t j | | � � t t j | d d g | d d g � t j | | � � q� q� WxMt d d � D]<} t j d | d | �j d | � } t j | d d | �j | d � } t j d! d | �j d d � } t j d | | d | d d d d �t | t j | j d � | j d � � j | � � d | d <t j | d d g | d d g d | d d d d �t | t j | j d � | j d � � j | � � q�
Wt j d d | �j d d � } t j d" d | �j d d# � } t j d$ d | �j d# d � } | d k r� t t j d% | | | � | j | � j | � � t t j | d d g | d d g | d d g � | j | � j | � � n t j d& d | �j d d � } t j d% | | | d | d d d d �| j d � j | j d � � } | j | j d � � j | � } t | | � d | d <t j | d d g | d d g | d d g d | d d d d �| j d � j | j d � � } | j | j d � � j | � } t | | � t j | � t j d � k r)t j d' d | �j d d d# � } t j d! d | �j d d d � } t t j d( | | � t j | | d) d d g d d g f �� t t j | d d d g | d d d g d d g � t j | | d) d d g d d g f �� t j d* d | �j d# d � } t j d+ | | d | d d d d �t | t j | j d � | j d � d) d d g d d g f �j | � � d | d <t j | d d d g | d d d g d d g d | d d d d �t | t j | j d � | j d � d) d d g d d g f �j | � � n Wd QXt j d d d, d d d- d d g d | �} t j d d. d/ d, d d d d g d | �} t j t t t t t t t t g � } t t j d0 | | | d d1 d d �t j t j | d k | d k � | d k � � t t j | d g | d g | d g d g d d1 d d �t j t j | d k | d k � | d k � � t j d2 d | �} t t j d3 d | � d t j | � � t t j d g | d g g � d t j | � � t t j d4 | d � d t j | � � t t j | d g d g g � d t j | � � xKt d d5 � D]:} t j | d | �} t j | � j d k rJt t j d6 | | � t j
| | � � t t j d7 | | � t j | | � � t t j d8 | d � d | � t t j d9 d | � d | � t t j d4 | d � d t j | � � t t j d3 d | � d t j | � � t t j d6 | d | d � t j
| d | d � � t t j d7 | d | d � t j | d | d � � t t j d8 | d d � d | d � t t j d9 d | d � d | d � t t j d4 | d d � d t j | d � � t t j d3 d | d � d t j | d � � qJqJWt j d2 d t �} t j d | d | d d �} t | t j | � � t | j t j | � � t j | d g g d | d d �} t | t j | � � t | j t j | � � t j d � d } t j d � j d d � d } t j d � j d d � d: }
t t j d; | | |
� d< � d S(A Ni i R s i->t axisi����i i i s ...i->...s i...->...R s ..., ...i i s
...i, ...ii s
i..., i...s i,jt ignores ij, js ij,jR t f8R t unsafe.s ji,ji t f2i s ij,jki i i i s ij,jk,kli i< s ijk, jil -> klt axesi
s ijk,jil->kli����i
g @g s i,i,i->it ?i s ,i->s i,->i s ...,...s i,is i,->is ,i->ii s z,mz,zm->i� i i i i ( R) R R R R t sumt astypeR R% t tracet multiplyt innerR( t outert warningst catch_warningst simplefiltert ComplexWarningt dotR t tensordott arrayt TrueR' t logical_andt itemsizet object( R R t nR, R- t ct dt tgtt pt qt r( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt check_einsum_sums� sr 1#( #% #( #%+5 %(( (($,1$ (5
%(4
$
%4@+
*
#%%(>%%!%
'
!!! 0 !
!!
3
!!
$$+3+!
'%
6
'.--' 1- 1)2)2(( )) ((18 $ c C` s | j d � d S( Nt i1( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_int8� s c C` s | j d � d S( Nt u1( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_uint8� s c C` s | j d � d S( Nt i2( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_int16� s c C` s | j d � d S( Nt u2( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_uint16� s c C` s | j d � d S( Nt i4( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_int32� s c C` s | j d � d S( Nt u4( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_uint32� s c C` s | j d � d S( Nt i8( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_int64� s c C` s | j d � d S( Nt u8( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_uint64� s c C` s | j d � d S( NR4 ( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_float16� s c C` s | j d � d S( Nt f4( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_float32� s c C` s | j d � d S( NR2 ( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_float64� s c C` s | j t j � d S( N( RO R t
longdouble( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_longdouble� s c C` s | j d � d S( Nt c8( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_cfloat64� s c C` s | j d � d S( Nt c16( RO ( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_cfloat128� s c C` s | j t j � d S( N( RO R t clongdouble( R ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_clongdouble� s c C` s` t j d � } t j d � } t t j d | | � d g d g g g � t j d t j � } t j d t j � } t t j d | | | � t j d
| | � � t j d d � } t j d d � j d d � } t j d d � j d d � } t t j d | | | � d d g d d g d d
g d d g g d d g d d g d d g d d g g g � d S( Ni i s ij...,j...->i...i i i i i s ijklm,ijn,ijn->s ijklm,ijn->i s x,yx,zx->xzyi i i i i i i i0 i@ ( i i ( i i i ( i i i i i ( i i i ( R t onesR R t int64R R ( R R, R- RI ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_misc� s + 'c
C` s� t j d � j d d d � } t j d � } t j d | | � } t t j d | | � | � t t j d | | � | � t t j d | | � | � t j d � j d � } t j d � j d � } t j d
| | � } t t j d | | � | � t t j d | | � | � t t j d
| | � | � t t j d | | � | � d d d d g } t j t j | � � j | � } t j | d � } t j d | | � } t t j d | | � | � t t j d | | � | � t t j d | | � | � d \ } } } t j | | | � j d d d | | | � } t j | | | d � j | | | d � } t j d | | � } t t j d | | � | � d S( Ni i i s
ijk,j->ijks ij...,j...->ij...s ij...,...j->ij...s ij...,j->ij...i i s ik,kj->ijs ik...,k...->i...s ik...,...kj->i...js ...k,kjs
ik,k...->i...i s ijkl,k->ijls ijkl,ks ...kl,ks
...kl,k...i� ix i s ...lmn,...lmno->...os ...lmn,lmno->...oi i ( i i ( i i ( i� i� ix ( R R R R R t prod(
R t At Bt reft dimsR, t vt Jt Kt M( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_broadcast s2 ! /-c C` s t j d � j d d � j t j � } t j d � j d d d � j t j � } t j d | | � } t j | | d d �} t | | � t j d
� j d d � j t j � } t j d � j d d d d � j t j � } t j d | | � } t j | | d d �} t | | � d S( Ni i i�
s cl,cpx->lpxR5 i i@ s
cl,cpxy->lpxyi i i@ ( i i i i i@ i � ( i i (
R R R R8 t float32t int16R RB R t float64( R Rp Rq t est tp( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_fixedstridebug( s '*
'-c C` s� t j j d d d � } t j d � } t j d | d | �t j d � } | | d d � d f | d d � d f | | f } t | | � d S( Ni i i s aabb->abR ( i i i i ( i i ( R t randomt normalt zerosR R R R ( R R. t y1t idxt y2( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_fixed_collapsingbugF s 6c C` s/ t j d
� } t j d � d d d � } t j d � } d | d d d � <t j d | | | d | �t | j | � t j d � d d d � } t j d | | | d | �t | j | � t j d � d } t j d
� } d | d <t j d t j � } t j d | | d | �t | j | � d S( Ni i
i s mi,mi,mi->mR s im,im,im->m.i s ij,jk->ik( i i ( i i i ( .i ( i i i ( .i ( i i ( R Rl R R R$ Ry ( R R. R t correct_base( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt( test_einsum_all_contig_non_contig_outputP s
c C` s� t j d d t j �d } t | d <t j d d t j �d } t j d
d t j �} t j d | | d | �} t | | � d S( Ni i R i .s ...ij,...jk->...ikR ( i i i ( i i i ( i i i ( R R� t bool_RD Rl R R ( R R, R RK t res( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_small_boolean_arraysg s
( t __name__t
__module__R R/ RO RQ RS RU RW RY R[ R] R_ R` Rb Rc Re Rg Ri Rk Rn Rx R~ R� R� R� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyR s0 A � � %
t __main__( t
__future__R R R R= t numpyR t
numpy.testingR R R R R R R R� ( ( ( sO /opt/alt/python27/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt <module> s .� � i
Mini Shell