| Current File : //usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyc |
�
E�`Qc @ s~ d d l Z d d l m Z d d l Z d d l Td d l m Z d d l Z d e f d � � YZ
e d k rz e � n d S( i����N( t Decimal( t *( t WarningManagert
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 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 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_argt ,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 ( t
assert_raisest
ValueErrort npt einsumt TypeErrort aranget reshape( t self( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_errors
s> " (+++'c C s
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 � } 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�^ qo� t j | t d d g t d g � } t | j | k � t | g | D]/ } g t d � D] } | | | f ^ q�^ q�� t j d
| � } t | j | k � t | g | j
d d d � D]/ } g t d � D] } | | | f ^ qs^ 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 ^ q�^ q�� t j d | � } t | j | k � t | g t d � D] } | d d � | | f ^ qW� 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 ^ q%� 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 ^ q�� t j | d d t g d t g � } t | j | k � t | g t d � D]. } | j
d d d � d d � | | f ^ qr� t j d | � } t | j | k � t | g t d � D]. } | j
d d d � d d � | | f ^ q�� 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 ^ qp^ q]� t j d � } d | _ t j d | � } t | j | k � t | g t d � D] } | | | | f ^ q�� t j | d d d g d g � } t | j | k � t | g t d � D] } | | | | f ^ qJ � 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 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 t assert_t baset Ellipsist assert_equalt Tt ranget transposet swapaxes( R t at bR t x( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_viewsH 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�Wx� 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
| | � � qOWx� 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 | | � � qWx� 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 � z� 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 | � � q_Wx�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 �t | | j d � j | j d � � j | j d � � j | � � d | d <t j | d d g | d d g | d d g d | d d d d �t | | j d � j | j d � � j | j d � � j | � � 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 | j � Xt 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 r(t 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 � � q(q(Wt 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 ..., ...s
...i, ...ii s
i..., i...s i,jt ignorei s ij, js ij,jR t f8R t unsafe.s ji,ji t f2i s ij,jki i 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 outerR t __enter__t warningst simplefiltert ComplexWarningt dotR t tensordott __exit__t arrayt Truet Falset logical_andt itemsizet object( R R t nR% R& t ctxt ct dt pt qt r( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_uint64� s c C s | j d � d S( NR- ( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_sums_float32� s c C s | j d � d S( NR+ ( RJ ( R ( ( sB /usr/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( RJ R t
longdouble( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ ( R ( ( sB /usr/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( RJ R t clongdouble( R ( ( sB /usr/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& RE ( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt test_einsum_misc� s + '( t __name__t
__module__R R( RJ RL RN RP RR RT RV RX RZ R[ R] R^ R` Rb Rd Rf Ri ( ( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyR s& > � � t __main__(
t syst decimalR t numpyR t
numpy.testingt numpy.testing.utilsR R7 t TestCaseR Rj t run_module_suite( ( ( sB /usr/lib64/python2.7/site-packages/numpy/core/tests/test_einsum.pyt <module> s
� �