Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyc
�
� 9Zc @` s� d Z d d l m Z m Z m Z d d l Z d d l 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 e d k r� e � n d S( s- Test functions for 1D array set operations.
i ( t divisiont absolute_importt print_functionN( t run_module_suitet TestCaset assert_array_equalt assert_equal( t ediff1dt intersect1dt setxor1dt union1dt setdiff1dt uniquet in1dt
TestSetOpsc B` st 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
RS( c C` s� d � } d d d d d d d g d } d d d d g } d d d d g } d d d d d d d g d } t j d d d d g d � } g } | j t j d � | j t j d
� | j d � | j d � xK | D]C } t j | | � } t j | | � }
| | |
| | | | � q� Wd
} t j t | � | � } | | (t j t | � | � }
| |
(| | |
| | | | � d d g } t j t t | | � � | � } t j t t | | � � | � }
| | |
| | | | � d d d g } t
t j | � d d g � d d d g } d d d g } d d d d d d g } t | � }
t
|
| � t | d t �\ } } t
| | � t
| | � t j
d � } d | d <t j | d t �\ } } t
| t j d � � d S( Nc
S` s� d } | j d | � } t | � } t | | | � | j d | � } t | d d d � \ } } t | | | � t | | | � | j d | � } t | d d d � \ } } t | | | � t | | | � | j d | � } t | d d d � \ } } t | | | � t | | | � | j d | � } t | d d d � \ } }
} t | | | � t |
| | � t | | | � | j d | � } t | d d d � \ } }
} t | | | � t |
| | � t | | | � | j d
| � } t | d d d � \ } }
} t | | | � t |
| | � t | | | � | j d | � } t | d d d � \ } }
} } t | | | � t |
| | � t | | | � t | | | � d S( Ns check {0} failed for type {1}t valuest return_indexi i t return_inverset
return_countss return_index and return_inverses return_index and return_countss return_inverse and return_countss. return_index, return_inverse and return_counts( t formatR R (
t at bt i1t i2t ct dtt base_msgt msgt vt jt j1t j2t j3( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt check_all sL !i i i i i
i i t
AllIntegert AllFloats
datetime64[D]s timedelta64[D]t Ot t ig �?y y �?R .( R% R& ( R% R& y �? y �? �y �? �y �? ( i i ( i i ( i i ( t npt multiplyt extendt typecodest appendt arrayt emptyt lent listt zipR R t Truet chararrayt zeros( t selfR! R R R R R t typesR t aat bbt unqt invt a1t a2t a2_inv( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_unique sN 0
!!
c C` s t j d d d d g � } t j d d d d d g � } t j d d d g � } t | | d t �} t | | � t j d d d d d g � } t j d d d d d d d g � } t j d d d g � } t | | � } t | | � t g t g g � � d S( Ni i i i i i t
assume_unique( R' R, R R1 R ( R4 R R t ecR t ed( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_intersect1dv s
$
c C` sg t j d d d d g � } t j d d d d d g � } t j d d d g � } t | | � } t | | � t j d d d g � } t j d d d g � } t j d d d d d d g � } t | | � } t | | � t j d d d d g � } t j d d d d g � } t j d d d d d d g � } t | | � } t | | � t g t g g � � d S( Ni i i i i i i i ( R' R, R R ( R4 R R R? R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt
test_setxor1d� s
!
!
c C` s� t j g � } t j d g � } t j d d g � } t g t | � � t d g t | d d �� t d g t | d d �� t d d g t | d d d d �� t g t | � � t d g t | � � d S( Ni i i t to_begint to_endi����( R' R, R R ( R4 t zero_elemt one_elemt two_elem( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_ediff1d� s %c C` s� xYd D]Q} d d d d g } d d d d d g | } t j t t t t g � } t | | d t �} t | | � d | d
<t j t t t t g � } t | | d t �} t | | � d \ | d
<| d <t j t t t t g � } t | | d t �} t | | � t j d d d d d d d d d d d d d d g � } d d d g | } t t t t t t t t t t t t t t g } t | | � } t | | � | d d d g | } t t t t t t t t t t t t t t g } t | | � } t | | � t j d d d d g � } t j d d d d d g | � } t j t t t t g � } t | | � } t | | � t j d d d d d g � } t j d d d d d d g | � } t j t t t t t g � } t | | � } t | | � t j d d g � } t j d d g | � } t j t t g � } t | | � } t | | � q Wt j d g � } t j d g � } t j t g � } t | | � } t | | � t t g g � g � d S(
Ni i
i i i i i R> i i ( i i
( i i ( R' R, R1 t FalseR
R ( R4 t multR R R? R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_in1d� s\
9
"
%
c C` s� t j d d d d d d d d g � } t j d d g � } t j t t t t t t t t g � } t | | � } t | | � d S( NR R R t dt e( R' R, R1 RI R
R ( R4 R R R? R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_in1d_char_array� s
''c C` s� x� d D]� } t j d d d d d d d d d d d d d d g � } d d d g | } t t j t | | � � t | | d t �� q Wd S(
s Test in1d's invert parameteri i
i i i i t invertN( i i
( R' R, R RO R
R1 ( R4 RJ R R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_in1d_invert� s
9c C` s� t j d � j d d � } t j d d � j d d � } t j d d � j d d � } t j t t t t t t g � } t t | | d t �| � t t | | d t �| � t t | | d t �| � t t | | d t �| � d S( Ni i i i i? i R> ( R' t aranget reshapeR, RI R1 R R
( R4 R R t long_bR? ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_in1d_ravel� s ! c C` s� t j d d d d d g � } t j d d d d d d d g � } t j d d d d d d g � } t | | � } t | | � t g t g g � � d S( Ni i i i i i ( R' R, R
R ( R4 R R R? R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_union1d s $!
c C` s t j d d d d d d d d g � } t j d d d d d d d g � } t j d d g � } t | | � } t | | � t j d � } t j d � } t j d d
g � } t | | � } t | | � t g t g g � � t j d t j � } t t | g � j t j � d S( Ni i i i i i i i i i ( ( R' R, R R RQ t uint32R t dtype( R4 R R R? R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_setdiff1d s '$
c C` sV t j d d d g � } t j d d d g � } t t | | � t j d g � � d S( NR R R t s( R' R, R R ( R4 R R ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt test_setdiff1d_char_array$ s c C` s� t j d d d d d g � } t j d d d d d d d g � } t | | � } t | | � } t | | � } t | | � } t | | � d S( Ni i i i i i i i ( R' R, R R R
R R ( R4 R R t c1t aux1t aux2t c2( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt
test_manyways) s $( t __name__t
__module__R= RA RB RH RK RN RP RT RU RX RZ R_ ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyR s e <
t __main__( t __doc__t
__future__R R R t numpyR' t
numpy.testingR R R R t numpy.lib.arraysetopsR R R R
R R R
R R` ( ( ( sS /opt/alt/python27/lib64/python2.7/site-packages/numpy/lib/tests/test_arraysetops.pyt <module> s "4� &
Mini Shell