| Current File : //usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyo |
�
E�`Qc @ sI d d l Td d l Td d l Td d l m Z m Z d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d
� � YZ
d e f d � � YZ d
e f d � � YZ d e f d � � YZ
d e f d � � YZ d e f d � � YZ d e f d � � YZ d e f d � � YZ d � Z e d k rEe � n d S( i����( t *( t matrixt asmatrixt TestApplyAlongAxisc B s& e Z d � Z d d � Z d � Z RS( c C sF t d d � } t t t d | � t | � t t | � d � � d S( Ni i
t di i ( i i
( t onest assert_array_equalt apply_along_axist lent shape( t selft a( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_simple s i c C sF t d d � } t t t d | � t | � t t | � d � � d S( Ni
ie R i i ( i
ie ( R R R R R ( R
t levelR ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_simple101 s c C sV t d � j d � } t t t d | � d d d g d d d g d d
d g g � d S(
Ni i i i i! i$ i' i* i- i0 i3 ( i i i ( t aranget reshapeR R t sum( R
R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_3d s ( t __name__t
__module__R R R ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s t TestApplyOverAxesc B s e Z d � Z RS( c C s_ t d � j d d d � } t t | d d g � } t | t d g d g d g g g � � d S( Ni i i i i i< i\ i| ( R R t apply_over_axesR R t array( R
R t aoa_a( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s ( R R R ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s t TestArraySplitc B sP e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C sA t d � } y t | d � } t d � Wn t k
r< n Xd S( Ni
i ( R t array_splitt assert_t
ValueError( R
R t res( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_0_split s
c C sz t d � } t | d � } t d � g } t | | � t | d � } t d � t d d � g } t | | � t | d � } t d � t d d � t d d � g } t | | � t | d � } t d � t d d � t d d � t d d � g } t | | � t | d � } t d � t d d � t d d � t d d � t d d � g } t | | � t | d � } t d � t d d � t d d � t d d � t d d
� t d
d � g } t | | � t | d � } t d � t d d � t d d � t d d � t d d � t d d
� t d
d � g } t | | � t | d � } t d � t d d � t d d � t d d � t d d � t d d � t d d
� t d
d � g } t | | � t | d
� } t d � t d d � t d d � t d d � t d d � t d d � t d d � t d d
� t d
d � g } t | | � t | d � } t d � t d d � t d d � t d d � t d d � t d d � t d d � t d d � t d d
� t d
d � g
} t | | � t | d � } t d � t d d � t d d � t d d � t d d � t d d � t d d � t d d � t d d
� t d
d � t g � g } t | | � d S( Ni
i i i i i i i i i i ( R R t compare_resultsR ( R
R R t desired( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split% sT
'
3
?
9
9
9*
96
-0
-0'c C sw t t d � t d � g � } t | d d d �} t t d � g � t t d � g � t g � g } t | | � d S( Ni
i t axisi ( R R R R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_rows[ s 3c C s� t t d � t d � g � } t | d d d �} t t d � t d � g � t t d d � t d d � g � t t d d � t d d � g � g } t | | � d S( Ni
i R" i����i i ( R R R R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_colsa s !'c C sq t t d � t d � g � } t | d � } t t d � g � t t d � g � t g � g } t | | � d S( s2 This will fail if we change default axis
i
i N( R R R R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_defaulti s 3c C sw t d � } d d d g } t | | d d �} t d d � t d d � t d d � t d d � g } t | | � d S( Ni
i i i R" i����i ( R R R ( R
R t indicesR R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_simpler s
6c C st t d � } d d d g } t | | d d �} t g � t d d � t d d � t d d � g } t | | � d S( Ni
i i i R" i����( R R R R ( R
R R& R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_low_boundy s
3c C s� t d � } d d d d d g } t | | d d �} t g � t d d � t d d � t d d � t g � t g � g } t | | � d S( Ni
i i i i R" i����( R R R R ( R
R R& R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_high_bound� s -(
R R R R! R# R$ R% R' R( R) ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s 6 t TestSplitc B s e Z d Z d � Z d � Z RS( s� * This function is essentially the same as array_split,
except that it test if splitting will result in an
equal split. Only test for this case.
*c C sG t d � } t | d � } t d � t d d � g } t | | � d S( Ni
i i ( R t splitR ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_equal_split� s c C sA t d � } y t | d � } t d � Wn t k
r< n Xd S( Ni
i i ( R R+ R R ( R
R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_unequal_split� s
( R R t __doc__R, R- ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR* � s t
TestDstackc B s, e Z d � Z d � Z d � Z d � Z RS( c C sS t d � } t d � } t | | g � } t d d g g g � } t | | � d S( Ni i ( R t dstackR ( R
R t bR R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_0D_array� s
c C sY t d g � } t d g � } t | | g � } t d d g g g � } t | | � d S( Ni i ( R R0 R ( R
R R1 R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_1D_array� s
c C sw t d g d g g � } t d g d g g � } t | | g � } t d d g g d d g g g � } t | | � d S( Ni i ( R R0 R ( R
R R1 R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_2D_array� s
$c C sh t d d g � } t d d g � } t | | g � } t d d g d d g g g � } t | | � d S( Ni i ( R R0 R ( R
R R1 R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_2D_array2� s
!( R R R2 R3 R4 R5 ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR/ � s t
TestHsplitc B s) e Z d Z d � Z d � Z d � Z RS( s& only testing for integer splits.
c C s? t d � } y t | d � t d � Wn t k
r: n Xd S( Ni i i ( R t hsplitR R ( R
R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR2 � s
c C s\ t d d d d g � } t | d � } t d d g � t d d g � g } t | | � d S( Ni i i i ( R R7 R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR3 � s $c C s� t d d d d g d d d d g g � } t | d � } t d d g d d g g � t d d g d d g g � g } t | | � d S( Ni i i i ( R R7 R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR4 � s
<( R R R. R2 R3 R4 ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR6 � s t
TestVsplitc B s e Z d Z d � Z d � Z RS( s& only testing for integer splits.
c C sK t d d d d g � } y t | d � t d � Wn t k
rF n Xd S( Ni i i i i ( R t vsplitR R ( R
R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR3 � s
c C s� t d d d d g d d d d g g � } t | d � } t d d d d g g � t d d d d g g � g } t | | � d S( Ni i i i ( R R9 R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR4 � s
6( R R R. R3 R4 ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR8 � s t
TestDsplitc B s e Z d Z d � Z d � Z RS( s& only testing for integer splits.
c C s] t d d d d g d d d d g g � } y t | d � t d � Wn t k
rX n Xd S( Ni i i i i ( R t dsplitR R ( R
R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR4 � s
c C s� t d d d d g d d d d g g d d d d g d d d d g g g � } t | d � } t d d g d d g g d d g d d g g g � t d d g d d g g d d g d d g g g � g } t | | � d S( Ni i i i ( R R; R ( R
R R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_3D_array� s 39( R R R. R4 R< ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR: � s t TestSqueezec B s e Z d � Z RS( c C s� t d d d d d � } t d d d d d � } t d d d d � } t t | � t | d � � t t | � t | d � � t t | � t | d � � d g g g } t | � } t | d � t | j d � t t | � t � d S( Ni i
i g �?i ( i i
i
( i i
i ( i i
( t randR t squeezeR t assert_equalt ndimt typet ndarray( R
R R1 t cR ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_basic� s
( R R RE ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR= � s t TestKronc B s e Z d � Z RS( c C s3 t d d g � } t | � } t t t | | � � t � t t t | | � � t � t t t | | � � t � t t t | | � � t � d t f d � � Y} | | j | j | j � } t t t | | � � t � t t t | | � � | � t t t | | � � t � t t t | | � � | � d S( Ni t myarrayc B s e Z d Z RS( g ( R R t __array_priority__( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRG s (
R R R@ RB t kronRC R R t dtypet data( R
R t mRG t ma( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_return_type s ( R R RN ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRF
s t TestTilec B s# e Z d � Z d � Z d � Z RS( c C s� t d d d g � } d d g d d g g } t t | d � d d d d d d g � t t | d � d d d d d d g d d d d d d g g � t t | d � d d d d d d g g � t t | d � d d d d g d d d d g g � t t | d � d d g d d g d d g d d g g � t t | d � d d d d g d d d d g d d d d g d d d d g g � d S(
Ni i i i i ( i i ( i i ( i i ( i i ( R R@ t tile( R
R R1 ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRE s (@+4:-c C s5 t g g g � } t | d � j } t | d � d S( Ni i i i ( i i i ( i i i ( R RP R R@ ( R
R R ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_empty* s c
C s� d d l j } d d
d d d
d g } d d d d d d g } xt | D]l } | j d d d | �} xK | D]C } t | | j � } t | | � } t | | � } t | | � qk WqF Wd S( Ni����i i i i i i
t size( i ( i i ( i i ( i i ( i i i ( i i ( i ( i i ( i i i ( i i i ( i i i i ( i i ( t numpy.randomt randomt randintR RJ RP RI R@ (
R
t nrt repsR t sR1 t rR t larget klarge( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_kroncompare/ s
( R R RE RQ R\ ( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRO s c C s6 x/ t t | � � D] } t | | | | � q Wd S( N( t rangeR R ( R R t i( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR = s t __main__N( t
numpy.testingt numpy.libt
numpy.coret numpyR R t TestCaseR R R R* R/ R6 R8 R: R= RF RO R R t run_module_suite( ( ( sE /usr/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt <module> s"
m