Current File : //opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyc
�
� 9Zc @` s� d Z d d l m Z m Z m Z d d l Z d d l 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 d d l m Z 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& m' Z' m( Z( m) Z) m* Z* m+ Z+ m, Z, m- Z- m. Z. m/ Z/ m0 Z0 m1 Z1 m2 Z2 m3 Z3 m4 Z4 m5 Z5 m6 Z6 m7 Z7 m8 Z8 m9 Z9 m: Z: m; Z; m< Z< m= Z= d d l j> j? Z@ d e
f d � � YZA d
e
f d � � YZB d e
f d
� � YZC d e
f d � � YZD d e
f d � � YZE d e
f d � � YZF d e
f d � � YZG d e
f d � � YZH d e
f d � � YZI d e
f d � � YZJ d e
f d � � YZK d e
f d � � YZL d e
f d! � � YZM d" e
f d# � � YZN eO d$ k r�e � n d S(% s� Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant
:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu
:version: $Id: test_extras.py 3473 2007-10-29 15:18:13Z jarrod.millman $
i ( t divisiont absolute_importt print_functionN( t TestCaset run_module_suitet assert_warnst clear_and_catch_warnings( t assert_t assert_array_equalt assert_equalt assert_almost_equal( t arrayt aranget maskedt MaskedArrayt masked_arrayt getmaskarrayt shapet nomaskt onest zerost count( t
atleast_1dt
atleast_2dt
atleast_3dt mr_t dott polyfitt covt corrcoeft mediant averaget uniquet setxor1dt setdiff1dt union1dt intersect1dt in1dt ediff1dt apply_over_axest apply_along_axist compress_ndt compress_rowcolst mask_rowcolst clump_maskedt clump_unmaskedt flatnotmasked_contiguoust notmasked_contiguoust notmasked_edgest
masked_allt masked_all_liket diagflatt TestGenericc B` s> e Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s4 t d
d t �} t d d g d d d g d t �} t | | � t j i d d g d 6d d g d 6� } t d d | �} t d d g d d d g d | �} t | | � t d d | �} t d d g d d g g d d d g d d g g d | �} t | | � t j d d d d g f g � } t d d | �} t d d f d d! f g d d d" f d d# f g d | �} t | | � t d$ d | �} t d d% f d d& f g d d d' f d d( f g d | �} t | | � t d) d | �} t d d* f g g d d d+ f g g d | �} t | | � d S(, Ni t dtypei t maskt at bt namest ft formatsi t bat bb( 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 i ( R7 R: ( R< R: ( R= R: ( i ( i i ( i i ( i i ( i i ( i ( i i ( i i ( i i ( i i ( i i ( i i ( i i ( R1 t floatR R t npR5 ( t selft testt controlt dt( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_masked_all' s0 $
)$
!!
!
0c C` sY t d d g d t �} t | � } t d d g d d d g d t �} t | | � t j i d d g d 6d d g d 6� } t d d g d d d g d | �} t | � } t d d g d d d g d | �} t | | � t j d d d d g f g � } t d d f d d f g d d d f d d f g d | �} t | � } t | | � d S( Ni i R5 R6 R7 R8 R9 R: R; i i
R< R= ( i i ( i i ( i i ( i i ( i
i
( i
i
( i i ( i i ( R7 R: ( R< R: ( R= R: ( i i ( i i ( i i ( i i ( R R> R2 R R? R5 ( R@ t baseRA RB RC ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_masked_all_likeE s $
)$ $
!! c C` s xt d d � D]� } x� t d | � D]� } t j | d t �} t j | | d t �} t d | � } | d | @d k | _ d } x+ | | � D] } | | j | j � 7} q� W| t k r� t
| j � j � | � q' | j | _ t
| j � j � | � q' Wq Wd S( Ni i i R5 i ( t rangeR? R t intt fullR R6 t datat sumR- R t
compressed( R@ R: t it jt kt jaR7 t st sl( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt check_clumpY s
c C` s� t t j d � � } t | d d d d d d g <t | � } t d d � t d d � t d d � g } t | | � | j t � d S(
Ni
i i i i i i i i ( R R? R R
R, t sliceR RS ( R@ R7 RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_clump_maskedi s *
c C` sy t t j d � � } t | d d d d d d g <t | � } t d d � t d d � g } t | | � | j t � d S(
Ni
i i i i i i i i ( R R? R R
R- RT R RS ( R@ R7 RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_clump_unmaskedt s
c C` s� t d � } t | � } t | t d | j � � t | | d k | d k B| d k B<t | � } t | t d d � t d d � g � t | (t | � } t | d � d S( Ni
i i i i i i ( R R. R RT t sizeR
t None( R@ R7 RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_flatnotmasked_contiguous~ s $ % ( t __name__t
__module__RD RF RS RU RV RY ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR4 % s
t TestAveragec B` s5 e Z d � Z d � Z d � Z d � Z d � Z RS( c C` s� t d d d d g d t t t t g �} t d t | d d �� t d t | d d d d d g �� t | d d d d d g d d
�\ } } t d | � | j | d k � t | (t t | d d �j t g � t d d d d g d t t t t g �} | j d d � } t | d d � d
f <t t | d d �d d g � t t | d d
�j d t g � t d d g t | d d �� t | d d d d
�\ } } t | d d g � d S(
Ng g �?g @g @R6 t axisi t weightst returnedi g @i ( R t Truet FalseR R t
assertTrueR
R6 t reshape( R@ t ottt resultt wts( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_testAverage1� s *%*
* # c C` s� d d d d d d g } d d d d d d g d d d d d d g g } t d d t j �} t t | d d �d � t t | d d d | �d � t t d d t j �d t d � g � } t t | d � t j j t j d � � d d
� t t | d d �t j d � d d � t t | d d �t | d d �t | d d �d g � t t | d d | �d � t t | d d d | �d
d d d d d g � t t | d d �t | d d �t | d d �d g � t d � } d d d d d d g } d d d d d d g d d d d d d g g } t
d � } d d d d d d g } t t t | | � d d �d � t t t | | � d d �d � t t t | | � d d �j t
g � t t t | | � d d �d
� t t t t | | � d d �� d � t | | � }
t t |
d � d � t t |
d d �d
d d d d d g � t t |
d d �d d g � t t |
d d d | �d
d d d d d g � d S( Ni i i R5 R] g @R^ g @g @g (@g 4@g @g g �?g @g $@g �X@g @g @g������
@g������
@( R R? t float_R R R RX t addt reduceR R R R6 R` R ( R@ t w1t w2t xt yt m1t m2t m3t m4t m5t z( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_testAverage2� s> 0 +3*) ) 0 ""("(+ c C` s t d � } t d � d } t | | g | | g g d d d d �\ } } t t | � t | � � t | j | j � t t d � d d d d d g d d �\ } } t t | � t | � � t t d � d d �\ } } t t | � t | � � t t d � d t d � d d �\ } } t t | � t | � � t d d g d d g g t � } t | t t g t t g g � } t | d d �} t | d
d g � t | d d �}
t |
d d g � t | d d �}
t |
d � t | d d �}
t |
d d g � d S( Ni i R] i R_ i i R^ i g �?g @g �?g @g �?g @( i i i ( i i i ( i i i ( i i i g������@( R R R R R R R> R Ra R` RX ( R@ R7 R8 t r1Rk t r2Rl t a2dt a2dmt a2dat a2dma( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_testAverage3� s* 00 *!!
c C` sf t t d d g � � } t | d � t t d d d d g d t t t t g �� } t | d � d S( Ni i g �?i i R6 ( R R R Ra R` ( R@ R7 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_onintegers_with_mask� s
0c
C` s� t j d d d d d g d d d d d g g d t �} t d d d d d g d d d d d g g d | �} t | � } t j | j � � } t | j | j � t | j | j � t | d d �} t | j d d �t | j d d �d } t | j | j � t | j | j � t | d d �} t | j d d �t | j d d �d } t | j | j � t | j | j � t j d d d d d g d d d d d g g � } t | d | �}
t j | j � d | | �} t |
j | j � t |
j | j � t | d | d d �} t | j d | d d �t | j d | d d �d } t | j | j � t | j | j � t | d | d d �} t | j d | d d �t | j d | d d �d } t | j | j � t | j | j � d S( Ni i R5 y @i y @i y @i y @y "@y �?i y @i y @y @R6 R] g �?g �?g @R^ y �? @y @ @y @ @y @ @y �?y @ @y @ @y @ @( R? R t boolR R RL R
t realt imag(
R@ R6 R7 t avt expectedt av0t expected0t av1t expected1Rf t wavt wav0t wav1( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_complex� sB ! ,, ( RZ R[ Rg Ru R| R} R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR\ � s
" t TestConcatenatorc B` s e Z d � Z d � Z RS( c C` s� t t d t d d d d d d g � � t d � } d d d d d g } t | d | �} t | d d | f } | j t | t � � t | d d d d d d d d d d d d g � t | j t | d d | f � d S(
Ni i i i i i i R6 ( i i i i i i ( R R R R R Rb t
isinstanceR R6 ( R@ R8 t mt dt c( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_1d s ) 1c C` s� t j j d d � } t j j d d � } t j t j j d d � d � } t j t j j d d � d � } t | d | �} t | d | �} t d | | f } | j | j d k � t | d d � d d � f | � t | d d � d d � f | � t | j t j
d | | f � t | | f } | j | j d k � t | d d � d d � f | � t | d d � d d � f | � t | j t j
| | f � d S( Ni i R6 t 1i
( i i
( i
i ( R? t randomt randt round_R R Rb R R R6 t r_( R@ t a_1t a_2t m_1t m_2t b_1t b_2R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_2d s !!## ##( RZ R[ R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� s t
TestNotMaskedc B` s e Z d � Z d � Z RS( c C` sJ t t j d � j d d � d d d d d d g d d d d d g d d d d d g d d d d d g d d d d d g g �} t | d � } t | d d g � t | d � } t | d d d
g � t | d d d g � t | d � } t | d d d g � t | d d d g � t | j d � } t | d d g � t | j d � } t | d d d g � t | d d d g � t | j d
� } t | d d d g � t | d d d g � t | d <t | d � } t | d d d g � t | d d d g � t | d
� } t | d d d! g � t | d d" d# g � d S($ Ni i R6 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 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 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 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 ( i i i i ( i i i i ( i i i i ( i i i i ( R R? R Rc R0 RX R RJ R
( R@ RJ RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_edges7 s:
c
C` s� t t j d � j d d � d d d d d d d d d g d d d d d d d d g d d d d d d d d g g �} t | d � } t | d t d d d � � t | d t d
d d � � t | d t d d
d � � t | d � } | j t | d � d k � | j | d d k � t | d | d � | j t | d � d k � t | d � } t | d d t d d
d � � | j | d d k � t | d d t d d d � � t | d d t d d d � � d S( Ni i i R6 i i i����i i����i i i����i i i i (
R R? R Rc R/ RX R RT Rb t len( R@ R7 t tmp( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_contiguousY s$ $ !!( RZ R[ R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� 4 s "t TestCompressFunctionsc B` s> e Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s5 t j t t d@ � � � j d d d � } t j dA � j t � } t | dB <t | d | �} t | � } t
| d d d d g d d d
d g d d
d d g g d d d d g d d d d g d d d d g g g � t | d � } t
| d d d d d g d d d d d g d d d d
d g d d! d
d d g g d d" d d d g d# d$ d% d& d' g d d( d d d g d d) d d d g g g � t | d � } t
| d d d d d g d d d d
d g d d! d
d d g g d* d+ d, d- d. g d/ d0 d1 d2 d3 g d4 d5 d6 d7 d8 g g d d" d d d g d d( d d d g d d) d d d g g g � t | dC � } t | d9 � } t | dD � } t
| | � t
| | � t
| | � t | d � } t
| d d d d g d d d d g d d d
d g d d
d d g g d* d, d- d. g d: d; d< d= g d/ d1 d2 d3 g d4 d6 d7 d8 g g d d d d g d# d% d& d' g d d d d g d d d d g g g � t | dE � } t | d> � } t | dF � } t
| | � t
| | � t
| | � t | dG � } t
| d d d d d g d d d d
d g d d! d
d d g g d d" d d d g d d( d d d g d d) d d d g g g � t | dH � } t
| | � t | dI � } t
| d d d d g d d d
d g d d
d d g g d* d, d- d. g d/ d1 d2 d3 g d4 d6 d7 d8 g g d d d d g d d d d g d d d d g g g � t | dJ � } t | dK � } t | dL � } t
| | � t
| | � t
| | � t | dM � } t
| d d d d g d d d d g d d d
d g d d
d d g g d d d d g d# d% d& d' g d d d d g d d d d g g g � t | dN � } t
| | � d S(O Ni i i i R6 i i i
i i
i i i i i i( i* i+ i, i2 i4 i5 i6 i7 i9 i: i; i i i i i i i) i- i. i/ i0 i1 i3 i8 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 i i ( i ( i����( i ( i����( i i ( i i����( i i ( i����i ( i i����( i����i����( i i ( i i����( R? R t listRG Rc R t astypeR~ R` R) R ( R@ Rm R� R7 t a2t a3t a4( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_compress_nds s� *
c C` s� t t j d � j d d � d d d d g d d d g d d d g g �} t t | � d d g d d g g � t t | d � d d d g d
d d g g � t t | d � d d g d d g d d g g � t | j d d d d g d d d g d d d g g �} t t | � d d g d
d g g � t t | d � d d d g d
d d g g � t t | d � d d g d d g d
d g g � t | j d d d d g d d d g d d d g g �} t t | � d g g � t t | d � d
d d g g � t t | d � d g d g d g g � t | j d d d d g d d d g d d d g g �} t t | � j d � t t | d � j d � t t | d � j d � d S( Ni i R6 i i i i i i i i ( R R? R Rc R R* t _dataRW ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_compress_rowcols� s" -%.19%.19"(9c C` s� t t j d � j d d � d d d d g d d d g d d d g g �} t t | � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t | j d d d d g d d d g d d d g g �} t t | � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t | j d d d d g d d d g d d d g g �} t t | � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t t | d � j d d d g d d d g d d d g g � t | j d d d d g d d d g d d d g g �} | j t | � j � t
k � | j t | d � j � t
k � | j t | d � j � t
k � | j t | � j j � � | j t | d � j j � � | j t | d � j j � � d S( Ni i R6 i i ( R R? R Rc R R+ R6 R� Rb t allR
( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_mask_rowcols� s: -+++9+++9+++9 "" c C` s� t j d d � } d d d d d d g } t | d | �j d d � } t | d | �j d d � } t | | d t �} t | j d d g d d g g � t | | d t �} t | j d d d g d d d g d d d g g � t | | d t �} t | t j | j d � | j d � � � t | | d t �} t | t j | j d � | j d � � � d d d d d d g } t | d | �j d d � } t | d | �j d d � } t | | d t �} t | j d d g d d g g � t | | d t �} t | j d d d g d d d g d d d g g � t | | d t �} t | t j | j d � | j d � � � t | t | | � � t | | d t �} t | t j | j d � | j d � � � d d d d d d g } t | d | �j d d � } t | d | �j d d � } t | | � } t | j t
� t | | � } t | j t
� t | d d d d d d d g �j d d � } t | d d d d d d d g �j d d � } t | | d t �} t | j d d g d d g g � t | | d t �} t | t j | j d � | j d � � � t | | d t �} t | j d d d g d d d g d d d g g � t | | d t �} t | t j | j d � | j d � � � t | d d d d d d d g �j d d � } t | d d d d d d d g �j d d � } t | | d t �} t | j d d g d d g g � t | | � } t | t j | j d � | j d � � � t | | d t �} t | j d d d g d d d g d d d g g � t | | d t �} t | t j | j d � | j d � � � t | d d d d d d d g �j d d � } t | d d d d d d d g �j d d � } t | | d t �} t | j d d g d d g g � t | | d t �} t | t j | j d � | j d � � � t | | d t �} t | j d d d g d d d g d d d g g � t | | d t �} t | t j | j d � | j d � � � d S( Ni i i R6 i i t strict( R? R R Rc R R` R R6 Ra t filledR ( R@ t nR� R7 R8 R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_dot sz "4++ "4++ 00"+4+00"+4+00"+4c C` s� t j d � } t | � } t t t | | � � t k � t t t | | � � t k � t t t | | � � t k � t t t | | � � t k � d S( Ni ( R? t eyeR R t typeR R ( R@ R7 R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_dot_returns_maskedarray] s c C` s` t t j d � � } t t j d � � } t | | d | �} t | | k � t | | � d S( Ni t out( i i ( R R? R� R R R R ( R@ R7 R� t res( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_dot_outf s
( RZ R[ R� R� R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� q s p F t TestApplyAlongAxisc B` s e Z d � Z d � Z RS( c C` sY t d � j d d d � } d � } t | d | � } t | d d g d d g g � d S( Ng (@i i c S` s | d S( Ni ( ( R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt myfuncs s i i i i
( R Rc R( R ( R@ R7 R� t xa( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_3dp s c C` sb t d � j d d d � } d d � } t | d | d d �} t | d d g d d
g g � d S( Ni i i i c S` s | d | S( Ni ( ( R8 t offset( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� } s R� i i i i ( R Rc R( R ( R@ R7 R� R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_3d_kwargsz s ( RZ R[ R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� n s
t TestApplyOverAxesc B` s e Z d � Z RS( c C` s� t d � j d d d � } t t j | d d g � } t j d g d g d g g g � } t | | � t | | d j t j � <t t j | d d g � } t j d g d
g d g g g � } t | | � d S( Ni i i i i i< i\ i| i i, (
R Rc R' R? RK R R R
R� R~ ( R@ R7 RA t ctrl( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_basic� s $
$( RZ R[ R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s t
TestMedianc 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 RS(
c C` sJ t j j t j t j g t j t j g g d d �} t | t j � d S( NR] i����( R? t maR t infR ( R@ t r( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_pytype� s 6c C` s� t j d � } t t j j | � d � t t t j j | � � t k � t d � } t t j j | � d � t t t j j | � � t k � d } t t j j | � d � t t t j j | � � t k � d S( Ni g @i g @( R? R R R� R R R� R RG ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_non_masked� s " "c C` sx t t j d � d d g d d g d �} t t j j | � d � t t j j | � j d d � t t t j j | � � t k � t t j d � j
d d
� d d g d d g d �} t t j j | � d � t t j j | � j d d � t t t j j | � � t k � t j j | d
d d t �} t | d d g � t | j d d � t t | � t k � d S( s5 test the examples given in the docstring of ma.mediani R6 i i i g �?s shape mismatchi
i i i g @R] i����t overwrite_inputg @g @N( ( ( i ( R R? R R R� R R R R� R Rc R` ( R@ Rm t ma_x( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_docstring_examples� s - "9 " c C` sY d } xLt d � D]>} x5t t g D]'} t d d | d | �} t j t | | � t t g � } xZ | D]R \ } } y t j j | d | d | �Wqm t | | | | | f � � qm Xqm Wt j | d | g t t g � } xd | D]\ \ } } y t j j | d | d | �Wn t
k
r,q� Xt | | | | | f � � q� Wq& Wq Wd S( Ns5 mask = %s, ndim = %s, axis = %s, overwrite_input = %si i t ndminR6 R] R� ( RG Ra R` R t itertoolst productR? R� R t AssertionErrort
IndexError( R@ t msgR� R6 Rm t argsR] t over( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_axis_argument_errors� s" " $#
c C` s` t d d t �} t t j j | � d � t d d t �} t t j j | � t j j � d S( Ni R6 ( R Ra R R? R� R R` R
( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_masked_0d� s c C` s t t j d � d t �} t t j j | � t j j � t t j j | � j d d � t t
t j j | � � t j j j k � t t j d � d t
�} t t j j | � d � t t j j | � j d
d � t t
t j j | � � t k � t t j d � d d d d d d g �} t t j j | � d � t t j j | � j d d � t t
t j j | � � t k � t t j d � d d d d d d g �} t t j j | � d � t t j j | � j d d � t t
t j j | � � t k � d S(
Ni R6 s shape mismatchg @i i g @g ( ( ( ( ( R R? R R` R R� R R
R R R� t coret MaskedConstantRa R ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_masked_1d� s + "* "* c C` sk t t j j t d d d g d d d d g �� j t j j t d d d g d d d d g �� j � d S( Ni i i R6 i ( R R? R� R R R ( R@ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_1d_shape_consistency� s 3c C` sK d \ } } t t j d d | � � } t | d *| d )t t j | | f d t �� } | | d d � d f <t j t | � � } xA t d | � D]0 } t j j
| � | | | d d � | f <q� Wt t | d d � d f � d � t t | � d � t t | d
d �t j
| � � t t | j d
d �t j
| � � d S( Nie i g �g �?i
i����R5 i i R] ( ie i ( R R? t linspaceR
t emptyR> R R� RG R� t shuffleR R R t T( R@ R� t pRm Rt t idxRM ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s ! #"c C` sO t t j d � j d d � � } t | d *| d )t t | � d � t t t j j | � � t
k � t t | d d �d d d g � t t t j j | d d �� t
k � t t | d d
�d d d d d d d
d d d g
� t t t j j | d d
�� t
k � t t | d d
�j d
d
d
d d d d d
d
d
g
� d S( Ni i
i i����g -@R] i g +@g /@i i
i i ( R R? R Rc R
R R R R� R� R R6 ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_2d_waxis� s !""(7(c C` s t j j d � j d d d � } t | | d d k <t t | d � d d g d d g d d g d
d g g � d | _ t t | d � d d g d
d g d d g g � t j j d � j d d d � } t | | d d k <t t | d � d d g d d g d d g g � d S( Ni i i i i i i i i i ic i
i i
i i i i i ( i i i ( R? R� R Rc R
R R R ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� s !: 1!c C` s\ t t j d � j d d � � } t | d *| d )t t | d d �t | d d �� d S( Ni i
i i����R] i����i ( R R? R Rc R
R R ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_neg_axis s !c C` s� t t j d � j d d � � } t | d *| d )t t j d � � } t | d d d | �} t | | � t t | � t
k � d S( Ni i
i i����R] i R� ( R R? R Rc R
R R R R R� R ( R@ Rm R� R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_out s !
c C` se d d g d d g d d g g } t j j | d � } d d g } t t j j | d d �| � d S( Ng �?g g @i R] ( R? R� t masked_equalR R ( R@ RJ t
masked_arrR� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt$ test_single_non_masked_value_on_axis s ( RZ R[ R� R� R� R� R� R� R� R� R� R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s t TestCovc B` s5 e Z d � Z d � Z d � Z d � Z d � Z RS( c C` s t t j j d � � | _ d S( Ni ( R R? R� R� RJ ( R@ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt setUp% s c C` s� | j } t t j | � t | � � t t j | d t �t | d t �� t t j | d t d t �t | d t d t �� d S( Nt rowvart bias( RJ R
R? R Ra R` ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_1d_wo_missing( s
(c C` s� | j j d d � } t t j | � t | � � t t j | d t �t | d t �� t t j | d t d t �t | d t d t �� d S( Ni i R� R� ( RJ Rc R
R? R Ra R` ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_2d_wo_missing0 s
(c C` s� | j } t | d <| | j � 8} | j � } t t j | � t | � � t t j | d t �t | d t �� t t j | d t d t �t | d t d t �� y t | d t �Wn t k
r� n X| d d !} t t j | | d d d � � t | | d d d � � � t t j | | d d d � d t �t | | d d d � d t �� t t j | | d d d � d t d t �t | | d d d � d t d t �� d S( Ni����R� R� t allow_maskedi (
RJ R
t meanRL R
R? R Ra R` t
ValueError( R@ Rm t nx( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_1d_w_missing8 s$
(
<%#+c C` s� | j } t | d <| j d d � } t j t | � � j t � } t j | | j � } | | j
d � d d � d f j d � } t
t | � t j | � | j d d | d � t
t | d t �t j | d t �| j d | � t j | j | � } | | j
d � j d � } t
t | d t �t j | d t �| j d d | d � t
t | d t d t �t j | d t d t �| j d | � d S( Ni����i i i i g �?R� R� ( RJ R
Rc R? t logical_notR R� RH R R� R� RX R� R
R R R` Ra ( R@ Rm t validt fract xf( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_2d_w_missingP s
, '% -( RZ R[ R� R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� # s
t catch_warn_maec B` s e Z d Z e f Z RS( sB Context manager to catch, reset warnings in ma.extras module
( RZ R[ t __doc__t maet
class_modules( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� f s t TestCorrcoefc B` sG e Z d � Z d � Z d � Z d � Z d � Z d � Z d � Z RS( c C` s: t t j j d � � | _ t t j j d � � | _ d S( Ni ( R R? R� R� RJ t data2( R@ ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� n s c
C` s
| j | j } } t j | � } t j | | � } t � �� t j d � t t t | d d �t j d � t t j | d d �t | d d �� t t | d d �| � t t | | d d �| � t t | d d �| � t t | | d d �| � Wd QXd S( Nt alwayst ddofi����t ignorei i (
RJ R� R? R R� t warningst simplefilterR t DeprecationWarningR
( R@ Rm Rn R� t expected2( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ddofr s
( c C` s� | j | j } } t j | � } t � �� t j d � t t t | | t t
� t t t | | t t � t t t | d t
�t j d � t t | d d �| � Wd QXd S( NR� R� R� i ( RJ R� R? R R� R� R� R R� R` Ra R
( R@ Rm Rn R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_bias� s
c C` s� | j } t t j | � t | � � t t j | d t �t | d t �� t � �F t j d � t t j | d t d t �t | d t d t �� Wd QXd S( NR� R� R� ( RJ R
R? R Ra R� R� R� R` ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s
c C` s� | j j d d � } t t j | � t | � � t t j | d t �t | d t �� t � �F t j d � t t j | d t d t �t | d t d t �� Wd QXd S( Ni i R� R� R� (
RJ Rc R
R? R Ra R� R� R� R` ( R@ Rm ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s
c
C` s" | j } t | d <| | j � 8} | j � } t t j | � t | � � t t j | d t �t | d t �� t � �F t j
d � t t j | d t d t �t | d t d t �� Wd QXy t | d t �Wn t k
r� n X| d d !} t t j | | d d d � � t | | d d d � � � t t j | | d d d � d t �t | | d d d � d t �� t � �� t j
d � t t j | | d d d � � t | | d d d � d d �� t t j | | d d d � � t | | d d d � d d �� Wd QXd S( Ni����R� R� R� R� i R� i (
RJ R
R� RL R
R? R Ra R� R� R� R` R� ( R@ Rm R� ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s2
<%#
# c
C` si | j } t | d <| j d d � } t | � } t j | � } t | d d � d d � f | d d � d d � f � t � �� t j d � t t | d d �d d � d d � f | d d � d d � f � t t | d d �d d � d d � f | d d � d d � f � t t | d d �d d � d d � f | d d � d d � f � Wd QXd S( Ni����i i R� R� i����R� i ( RJ R
Rc R R? R
R� R� R� ( R@ Rm RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� � s
9
( ( (( RZ R[ R� R R R� R� R� R� ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR� l s
t TestPolynomialc B` s e Z d � Z d � Z RS( c C` s� t j j d � } t j j d � j d d � } t t | | d � t j | | d � � | j t � } t | d <| j t � } t | d
<| d <t | | d d � d f d d t �\ } } } } } t j | d | d d � d f j
� d d t �\ } } }
} } xH t | | | | | f | | |
| | f � D] \ }
} t |
| � q;Wt | | d d � d f d d t �\ } } } } } t j | d d !| d d � d f d d t �\ } } }
} } xH t | | | | | f | | |
| | f � D] \ }
} t |
| � q�Wt | | d d t �\ } } } } } t j | d d !| d d � d d � f d d t �\ } } }
} } xH t | | | | | f | | |
| | f � D] \ }
} t |
| � q�Wt j j d � d } | j � } | d d !} | d d !} | d d !} t | | d d t d | �\ } } } } } t j | | d d t d | �\ } } }
} } t
| | � xH t | | | | | f | | |
| | f � D] \ }
} t |
| � q�Wd S( Ni
i i����i i i RI i t w( i i ( i����i����( R? R� R� Rc R
R t viewR R
R` RL t zipt copyR ( R@ Rm Rn t Ct Rt Kt St DR� R� RO RQ R� R7 t a_R t wot xst yst ws( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_polyfit� s< (
7,:7A:'G:
-0
:c C` s9 t j j d � } t j j d � j d d � } t j | d <t j | d <| j t � } | j t � } t | d <t | d
<t | | d d t �\ } } } } } t j | d d !| d d � d d � f d d t �\ } } }
} } xH t
| | | | | f | | |
| | f � D] \ }
} t |
| � qWd S( Ni
i i����i i i RI i ( i����i����( i����i����( R? R� R� Rc t nanR R R
R R` R R
( R@ Rm Rn R R R R
R R� R� RO RQ R� R7 R ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_polyfit_with_masked_NaNs s
'G:( RZ R[ R R ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR � s 't TestArraySetOpsc 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 RS( c C` s� d d d d d d g } t | d t d t �} | j t | d t � � t | d t d d d g d d d d g �� t | d d d d g � t | d d d d d d d g � d S( Ni i i t return_indext return_inversei R6 i ( R R` Rb R� R R R ( R@ RJ RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_unique_onlist s /c
C` s� t d d d d d d g d d d d d d d g �} t | d t d t �} t | d t d d d d g d d d d d g �� t | d d d d d g � t | d d d d d d d g � d | _ t d
d d d d d d g d d d d d d d g d d � } t | d t d t �} t | d t d d d d g d d d d d g �� t | d d d d d g � t | d d d d d d d g � d S( Ni i i R6 i R R i����i RJ t
fill_value( R R R` R R ( R@ RJ RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_unique_onmaskedarray s 65 # !5 c C` s� t d d d g d t �} t | d t d t �} t | d t d g d t g �� t | d d g � t | d d d d g � t } t | d t d t �} t | d t t � � t | d d g � t | d d g � d S( Ni R6 R R i i ( R R` R R R
( R@ RJ RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_unique_allmasked, s #c C` s� t t j d � d d d d d d g �} t d d d d g d d d d d g �} t | � } t | | � t | j | j � t | j | j � d S( Ni R6 i i i ( R R? R R R& R RJ R6 ( R@ Rm RB RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ediff1d; s **
c
C` s- t t j d � d d d d d d g �} t | d t �} t d d d d d g d d d d d d g �} t | | � t | j | j � t | j | j � t | d d d d g �} t d d d d d d d g d d d d d d d d g �} t | | � t | j | j � t | j | j � d S( Ni R6 i i t to_begini i i ( R R? R R& R
R R RJ R6 ( R@ Rm RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ediff1d_tobeginD s *0
<
c
C` s- t t j d � d d d d d d g �} t | d t �} t d d d d d g d d d d d d g �} t | | � t | j | j � t | j | j � t | d d d d g �} t d d d d d d d g d d d d d d d d g �} t | | � t | j | j � t | j | j � d S( Ni R6 i i t to_endi i i ( R R? R R& R
R R RJ R6 ( R@ Rm RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ediff1d_toendS s *0
<
c C` sE t t j d � d d d d d d g �} t | d t d t �} t d d d d d d g d d d d d d d g �} t | | � t | j | j � t | j | j � t | d d d d g d t �} t d d d d d d d d g d d d d d d d d d g �} t | | � t | j | j � t | j | j � d S(
Ni R6 i i R R i i i ( R R? R R& R
R R RJ R6 ( R@ Rm RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ediff1d_tobegin_toendb s *6
!!!
c C` s t j d � } t | � } t d d d d g d d d d d g �} t | | � | j t | t � � t | j | j � t | j | j � t | d t
d t
�} t d d d d d d g d d d d d d d g �} | j t | t � � t | j | j � t | j | j � d S( Ni i R6 i R R ( R? R R& R R Rb R� R RJ R6 R
( R@ Rm RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_ediff1d_ndarrayr s *
6c C` s� t d d d d g d d d d d g �} t d d d d g d d d d d g �} t | | � } t d d d g d d d d g �} t | | � d S( Ni i R6 i i����( R R$ R ( R@ Rm Rn RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_intersect1d� s
**$c C` s0 t d d d d d g d d d d d d g �} t d d d d d d g d d d d d d d g �} t | | � } t | t d d d g � � t d d d d d g d d d d d d g �} d d d d d g } t | | � } t | t d d d d g d d d d d g �� t d d d g � } t d
d d g � } t | | � } t t | t � � t | d d d d d d
g � t d d d d g d d d d d g �} t d
d d d g d d d d d g �} t | | � } t t | t � � t | d d d d d d
g � t g t g g � � d S( Ni i i i i����R6 i i i i i ( R R! R R R� R R ( R@ R7 R8 RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt
test_setxor1d� s& 06 01 ** c C` s* t d d d d d g d d d d d d g �} t d d d d d d g d d d d d d d g �} t | | � } t | t t t t t g � t d d d d d g d d d d d d g �} t d d d g d d d d g �} t | | � } t | t t t t t g � t g t g g � � d S(
Ni i i i i����R6 i i i ( R R% R R` Ra R ( R@ R7 R8 RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_in1d� s 06 0$ c C` s6 t d d d d d g d d d d d d g �} t d d d d d d g d d d d d d d g �} t t j t | | � � t | | d
t �� t d d d d d g d d d d d d g �} t d d d g d d d d g �} t t j t | | � � t | | d
t �� t g t g g d
t �� d S( Ni i i i i����R6 i i i t invert( R R R? R% R% R` R ( R@ R7 R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_in1d_invert� s 06.0$.c
C` s� t d d d d d d g d d d d d d d g �} t d d d d d d g d d d d d d d g �} t | | � } t d d d d d d d g d d d d d d d d g �} t | | � t g t g g � � d S(
Ni i i i i����R6 i i i ( R R# R R ( R@ R7 R8 RA RB ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_union1d� s 66<
c C` s t d d d d d d d d g d d d d d d d d d g �} t d d d d d d d g � } t | | � } t | t d d d
g d d d d g �� t d � } t d � } t t | | � t d d
g � � t g t j d g �} t t | g � j t j � d S( Ni i i i i i R6 i i i����i
i i ( R R" R R R? t uint32R5 ( R@ R7 R8 RA ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_setdiff1d� s B!+ "c C` sV t j d d d g � } t j d d d g � } t t | | � t j d g � � d S( NR7 R8 R� RQ ( R? R R R" ( R@ R7 R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_setdiff1d_char_array� s ( RZ R[ R R R R R R R R! R" R# R$ R& R' R) R* ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR s
t
TestShapeBasec B` s e Z d � Z d � Z RS( c C` s� t d d d g d d d d g �} t | � } t | j d � t | j j | j j � t | j d � t | j j | j j � d S( Ni i i R6 i ( i i ( i ( R R R R R6 RJ ( R@ R7 R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_atleast2d� s $ c C` s� t d � } t | j d � t | j j | j j � t d � } t | j d � t | j j | j j � t d � } t | j d � t | j j | j j � t d � } t | j d � t | j j | j j � d S( Ng �?i ( i ( i i ( i i i ( i i ( R R R R6 RJ R R R3 ( R@ R8 ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt test_shape_scalar� s ( RZ R[ R, R- ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyR+ � s t __main__(P R� t
__future__R R R R� R� t numpyR? t
numpy.testingR R R R t numpy.ma.testutilsR R R R
t
numpy.ma.coreR R R
R R R R R R R R t numpy.ma.extrasR R R R R R R R R R R R! R" R# R$ R% R& R' R( R) R* R+ R, R- R. R/ R0 R1 R2 R3 R� t extrasR� R4 R\ R� R� R� R� R� R� R� R� R� R R R+ RZ ( ( ( sM /opt/alt/python27/lib64/python2.7/site-packages/numpy/ma/tests/test_extras.pyt <module> s2 ""L�i�#=�
�Ck:�
Mini Shell