Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyo
�
� 9Zc @` s� d d l m Z m Z m Z d d l Z d d l m Z m Z m Z m
Z
m Z m Z m
Z
m Z m Z m Z d d l m Z m Z m Z m Z m Z 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 e f d � � YZ$ d � Z% e& d k r�e � n d S( i ( t divisiont absolute_importt print_functionN(
t apply_along_axist apply_over_axest array_splitt splitt hsplitt dsplitt vsplitt dstackt kront tile( t run_module_suitet TestCaset assert_t assert_equalt assert_array_equalt
assert_raisest assert_warnst TestApplyAlongAxisc B` s& e Z d � Z d d � Z d � Z RS( c C` sI t j d d � } t t t d | � t | � t j | j d � � d S( Ni i
t di i ( i i
( t npt onesR R t lent shape( t selft a( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_simple s i c C` sI t j d d � } t t t d | � t | � t j | j d � � d S( Ni
ie R i i ( i
ie ( R R R R R R ( R t levelR ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_simple101 s c C` s\ t j d � j d � } t t t j 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 ( R t aranget reshapeR R t sum( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_3d s ( t __name__t
__module__R R R" ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s t TestApplyOverAxesc B` s e Z d � Z RS( c C` sh t j d � j d d d � } t t j | d d g � } t | t j d g d g d g g g � � d S( Ni i i i i i< i\ i| ( R R R R R! R t array( R R t aoa_a( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR s ( R# R$ R ( ( ( sR /opt/alt/python27/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` s& t j d � } t t t | d � d S( Ni
i ( R R R t
ValueErrorR ( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_0_split' s c C` sC t j d � } t | d � } t j d � g } t | | � t | d � } t j d � t j d d � g } t | | � t | d � } t j d � t j d d � t j d d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d
� t j d
d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d � t j d d
� t j d
d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d
� t j d
d � g } t | | � t | d
� } t j d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d
� t j d
d � g } t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d
� t j d
d � g
} t | | � t | d � } t j d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d � t j d d
� t j d
d � t j g � g } t | | � d S( Ni
i i i i i i i i i i ( R R R t compare_resultsR& ( R R t rest desired( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split+ sb
!
0
*
*$
*3
*-
*-$
*-3
*--
*--!c C` s< t j t j d � t j d � g � } t | d d d �} t j t j d � g � t j t j d � g � t j d � g } t | | � t | j j | d j j k � t | d d d g d d �} t j d � t j t j d � g � t j t j d � g � g } t | | � t | j j | d j j k � d S(
Ni
i t axisi i����i i ( i i
( i i
( R R& R R t zerosR+ R t dtypet type( R R R, t tgt( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_rowsh s '0
$
c C` s� t j t j d � t j d � g � } t | d d d �} t j t j d � t j d � g � t j t j d d � t j d d � g � t j t j d d � t j d d � g � g } t | | � d S( Ni
i R/ i����i i ( R R& R R R+ ( R R R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_colsw s '$*0c C` s� t j t j d � t j d � g � } t | d � } t j t j d � g � t j t j d � g � t j d � g } t | | � t | j j | d j j k � d S( s2 This will fail if we change default axis
i
i i i����N( i i
( R R& R R R0 R+ R R1 R2 ( R R R, R3 ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_integer_split_2D_default s '0
c C` s� t j d � } d d d g } t | | d d �} t j d d � t j d d � t j d d � t j d d � g } t | | � d S( Ni
i i i R/ i����i ( R R R R+ ( R R t indicesR, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_simple� s -c C` s� t j d � } d d d g } t | | d d �} t j g � t j d d � t j d d � t j d d � g } t | | � d S( Ni
i i i R/ i����( R R R R& R+ ( R R R7 R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_low_bound� s *c C` s� t j d � } d d d d d g } t | | d d �} t j g � t j d d � t j d d � t j d d � t j g � t j g � g } t | | � d S( Ni
i i i i R/ i����( R R R R& R+ ( R R R7 R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_index_split_high_bound� s *-(
R# R$ R* R. R4 R5 R6 R8 R9 R: ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR( & s = t TestSplitc B` s e Z d � Z d � Z RS( c C` sP t j d � } t | d � } t j d � t j d d � g } t | | � d S( Ni
i i ( R R R R+ ( R R R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_equal_split� s !c C` s& t j d � } t t t | d � d S( Ni
i ( R R R R) R ( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_unequal_split� s ( R# R$ R< R= ( ( ( sR /opt/alt/python27/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` s\ t j d � } t j d � } t | | g � } t j d d g g g � } t | | � d S( Ni i ( R R& R
R ( R R t bR, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_0D_array� s
c C` sb t j d g � } t j d g � } t | | g � } t j d d g g g � } t | | � d S( Ni i ( R R& R
R ( R R R? R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_1D_array� s
c C` s� t j d g d g g � } t j d g d g g � } t | | g � } t j d d g g d d g g g � } t | | � d S( Ni i ( R R& R
R ( R R R? R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_2D_array� s
'c C` sq t j d d g � } t j d d g � } t | | g � } t j d d g d d g g g � } t | | � d S( Ni i ( R R& R
R ( R R R? R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_2D_array2� s
$( R# R$ R@ RA RB RC ( ( ( sR /opt/alt/python27/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` sB t j d � } y t | d � t d � Wn t k
r= n Xd S( Ni i i ( R R& R R R) ( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR@ � s
c C` se t j d d d d g � } t | d � } t j d d g � t j d d g � g } t | | � d S( Ni i i i ( R R& R R+ ( R R R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRA � s *c C` s� t j d d d d g d d d d g g � } t | d � } t j d d g d d g g � t j d d g d d g g � g } t | | � d S( Ni i i i ( R R& R R+ ( R R R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRB � s
B( R# R$ t __doc__R@ RA RB ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRD � s t
TestVsplitc B` s e Z d Z d � Z d � Z RS( s& Only testing for integer splits.
c C` sN t j d d d d g � } y t | d � t d � Wn t k
rI n Xd S( Ni i i i i ( R R& R R R) ( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRA � s
c C` s� t j d d d d g d d d d g g � } t | d � } t j d d d d g g � t j d d d d g g � g } t | | � d S( Ni i i i ( R R& R R+ ( R R R, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRB � s
<( R# R$ RE RA RB ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRF � s t
TestDsplitc B` s e Z d � Z d � Z RS( c C` s` t j d d d d g d d d d g g � } y t | d � t d � Wn t k
r[ n Xd S( Ni i i i i ( R R& R R R) ( R R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRB s
c C` s� t j 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 j d d g d d g g d d g d d g g g � t j 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, R- ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_3D_array
s 6<( R# R$ RB RH ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRG s t TestSqueezec B` s e Z d � Z RS( c C` s d d l m } | d d d d d � } | d d d d d � } | d d d d � } t t j | � t j | d � � t t j | � t j | d � � t t j | � t j | d � � d g g g } t j | � } t | d � t | j d � t t | � t j � d S(
Ni ( t randi i
i g �?( i i
i
( i i
i ( i i
(
t numpy.randomRJ R R t squeezeR R t ndimR2 t ndarray( R RJ R R? t cR, ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_basic s """
( R# R$ RP ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRI s t TestKronc B` s e Z d � Z RS( c C` sN t j d d g � } t j | � } t t t | | � � t j � t t t | | � � t j � t t t | | � � t j � t t t | | � � t j � d t j f d � � Y} | | j | j | j
� } t t t | | � � t j � t t t | | � � | � t t t | | � � t j � t t t | | � � | � d S( Ni t myarrayc B` s e Z d Z RS( g ( R# R$ t __array_priority__( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRR 4 s ( R R t asmatrixR R2 R RN t matrixR R1 t data( R R t mRR t ma( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_return_type, s ( R# R$ RY ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRQ + s t TestTilec B` s, e Z d � Z d � Z d � Z d � Z RS( c C` s� t j 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& R R ( R R R? ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRP ? s (@+4:-c C` sB t j d � } t | d � } | d 7} t | t j d � � d S( Ni i i ( R R R R ( R R R? ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt( test_tile_one_repetition_on_array_gh4679J s
c C` sl t j g g g � } t j g g g � } t | d � j } t | d � j } t | d � t | d � d S( Ni i i i ( i i i ( i i ( i i i ( R R& R R R ( R R R? RO R ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt
test_emptyP s
c
C` s� d d l m } d d
d d d
d g } d d d d d d g } xt | D]l } | d d d | �} xN | D]F } t j | | j � } t | | � } t | | � } t | | � qi WqG Wd S( Ni ( t randinti 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 ( RK R] R R R1 R R R (
R R] t repsR t sR? t rR t larget klarge( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt test_kroncompareX s
( R# R$ RP R[ R\ Rd ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRZ > s t TestMayShareMemoryc B` s e Z d � Z RS( c C` si t j d � } t j d � } | j t j | | � � | j t j | | d d d � � � | j t j | | d d d � � � | j t j | | d d � d d d � f � � | j t j | d d d � | � � | j t j | d d d � | � � | j t j | d d � d d d � f | � � | j t j | d d � d d d � f | � � d S(
Ni2 i< i i i����i i ( i2 i< ( i i< i ( R R t
assertTruet may_share_memoryt assertFalse( R R t d2( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRP g s &&2&&2( R# R$ RP ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyRe f s c C` s6 x/ t t | � � D] } t | | | | � q Wd S( N( t rangeR R ( R, R- t i( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyR+ v s t __main__(' t
__future__R R R t numpyR t numpy.lib.shape_baseR R R R R R R R
R R t
numpy.testingR
R R R R R R R R% R( R; R> RD RF RG RI RQ RZ Re R+ R# ( ( ( sR /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_shape_base.pyt <module> s$ F4} (
Mini Shell