yZ> G d d Z? G d
d Z@ G d d
ZA G d d ZB G d d ZC G d d ZD G d d ZE G d d ZF G d d ZG G d d ZH G d d ZI G d d ZJ G d d! ZK G d" d# ZL G d$ d% ZMdS )&zTests 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 $
N)normalize_axis_tuple) assert_warnssuppress_warnings)assert_assert_array_equal assert_equalassert_almost_equal) arrayarangemasked MaskedArray masked_array getmaskarrayshapenomaskoneszeroscount)"
atleast_1d
atleast_2d
atleast_3dmr_dotpolyfitcovcorrcoefmedianaverageuniquesetxor1d setdiff1dunion1d intersect1din1dediff1dapply_over_axesapply_along_axis compress_ndcompress_rowcols mask_rowcols clump_maskedclump_unmaskedflatnotmasked_contiguousnotmasked_contiguousnotmasked_edges
masked_allmasked_all_likeisindiagflat ndenumeratestackvstackc > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestGenericc ( t dt }t ddgddgt }t || t j ddgddgd }t d| }t d d gd
d
g| }t || t d | }t d d gd d ggd
d
gd
d
gg| }t || t j d dd
dgfg }t d| }t ddgddg| }t || t d| }t ddgddg| }t || t d
| }t dggdgg| }t || d S )N dtype maskr= abfnamesformatsr r r> r> r; r; rA rC barC bbrC r> rH )r0 floatr
r npr= )selftestcontroldts n/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_extras.pytest_masked_allzTestGeneric.test_masked_all& s $e,,,AaV5999T7###
Xc
SzBB
C
C$b)))(/?rJJJT7###&+++&&)FF+;< %v .0@ A " " " T7###
XzC+{)C#DE
F
F$b)))k2 ); 7rC C CT7###$b)))k2 ); 7rC C CT7###&++++ }oRHHHT7##### c P t j ddt fgdffg }t j d| }t t
|d t j j j t t
|d d t j j j t t |d d d t |d d j
d t |d d j j
d d S )NrB cr> r> rH rQ r= objectmar0 r typecorer
lenr _fill_valuerR my_dtype
masked_arrs rV "test_masked_all_with_object_nestedz.TestGeneric.test_masked_all_with_object_nestedD s 8cc6]OT#: ; <==U%%dH55
T*S/**BEJ,BCCCT*S/#.//1GHHHSC -..222Z_S)/888Z_S)5;R@@@@@rX c t j dt dffg }t j d| }t t
|d t j j j t t |d d t |d j
d t |d j j
d d S )NrB r[ r> rH r\ r] rd s rV test_masked_all_with_objectz'TestGeneric.test_masked_all_with_objectP s 8cFD> 2 344U%%dH55
T*S/**BEJ,BCCCSC ))1---Z_*F333Z_06;;;;;rX c t ddgt }t | }t ddgddgt }t || t j ddgddgd }t d d gd
d
g| }t | }t d d gd
d
g| }t || t j d dd
dgfg }t ddgddg| }t | }t || d S )Nr> r; r< r? rA rB rC rD rG rH )
rk rJ rK rM rO )r
rP r1 r rQ r= )rR baserS rT rU s rV test_masked_all_likez TestGeneric.test_masked_all_likeZ s4 aV5))) t$$AaV5999T7###
Xc
SzBB
C
Cff%VV, r; r< r ) rangerQ r intfullr r@ datasumr, r
compressed) rR rC ijkjarA ssls rV check_clumpzTestGeneric.check_clumpn s, q!
:
:A1a4[[
:
:Ias+++WQ--- A&& 1+!+!A$$ * *B )))AA&& !3!3!5!5q9999 fWAF !3!3!5!5q9999
:
:
:rX c 0 t t j d }t |g d< t | }t dd t dd t dd g}t
|| | t d S )Nrk r r> r; r r ro r )r rQ r r r+ slicer r| rR rA rS rT s rV test_clump_maskedz TestGeneric.test_clump_masked~ s 2 ' ' &
AA;;a U1b\\:T7### &&&&&rX c t t j d }t |g d< t | }t dd t dd g}t
|| | t d S )Nrk r~ r r ro r )r rQ r r r, r r r| r s rV test_clump_unmaskedz TestGeneric.test_clump_unmasked sy 2 ' ' &
a A;;a .T7### (((((rX c t d }t | }t |t d|j g t j dt |_ t |t d|j g t ||dk |dk z |dk z < t | }t |t dd t dd g t |d d < t | }t |g d S ) Nrk r r< r r r r )
r r- r r sizerQ r boolr@ r )rR rA rS s rV test_flatnotmasked_contiguousz)TestGeneric.test_flatnotmasked_contiguous s 2JJ'**TE!QV , ,-..."D)))TE!QV , ,-...*01q5QU
qAv
&''**TE!QKKq!5666!!!'**T2 rX N) __name__
__module__ __qualname__rW rg ri rm r| r r r r\ rX rV r8 r8 $ s $ $ $<
A
A
A< < <$ $ $(: : : ' ' ') ) ) rX r8 c e Zd Zd Zd Zd Zd Zd Zd Ze j
dg dd d
gg d d gd
gfg dg dgdg dgddgg dgg dgfg d Z d Z
d S ) TestAveragec t g dg d }t dt |d t dt |g d t |g dd
\ }}t d| t |d k t |d d < t t |d j d g t g dg d }| d d }t |d d d
f< t t |d ddg t t |d
j d d g t ddgt |d t |dd \ }}t |ddg d S )N) ? @ @)TFFFr@ r r axis)r r r r weightsTr returned @r; r> r r r r )r
r r r r r@ reshape)rR ottresultwtss rV test_testAverage1z TestAverage.test_testAverage1 s $$$+F+F+FGGGS'#A...///S'#/?/?/?@@@AAA c+;+;+;dKKK S&!!!s
AAAWSq))).777$$$+F+F+FGGGkk!Q AAAqD WSq)))C:666WSq))).q1D6:::b"Xws 3 3 3444 cD999 S2r(#####rX c
N g d}g dg dg}t dt j }t t |d d t t |d| d t t dt j d t d z g }t t |d t j t j d d
z d z t t |d t j d d
z d z t t |d t |d t |d d z g t t |d |
d t t |d| g d t t |d t |d t |d d z g t d }g d}g dg dg}t d }g d} t t t || d d t t t || d d t t t || d j dg t t t || d d t t t t || d d t || }
t t |
d d t t |
d g d t t |
d ddg t t |
d| g d d S )Nr r> r> r> r> r r> r r r r r> r r< r r @)r r r r (@r> r g
@)r r r r r $@)r r r> r> r r )r r> r> r> r> r> Tr )r r X@r r g @ @)r r r r r r )
r rQ float_r r r
addreducer r r r@ r ) rR w1w2xym1m2m3m4m5zs rV test_testAverage2z TestAverage.test_testAverage2 s
"4"4"4
5 1BI & & &WQQ'''---WQQ333S999 6!29---sVAYY? @ @WQ%%rv}}RYq\\'B'BR'G#'MNNNWQQ'''1 ):R)?@@@WQQ''' aa((('!!*<*<*GGGW\!R 0 0q9993???U7<2#6#6Q???@@!DDD B WQ%%x000WQQ''')E)E)EFFFWQQ'''#s444WQQ333222 4 4 4 4 4rX c b t d }t d dz }t ||g||ggdd \ }}t t | t | t |j |j t t d dddgd \ }}t t | t | t t d d \ }}t t | t | t t d t d d
\ }}t t | t | t dd gdd ggt }t |d
d
gdd
gg }t |d } t | ddg t |d }
t |
ddg t |d }
t |
d t |d }
t |
ddg d S )Nr r r> Tr )r; r; r r r r r )r r r; Fr ?r r g@ ?r )r r r r r r
rP r ) rR rA rB r1r r2r a2da2dma2daa2dmas rV test_testAverage3z TestAverage.test_testAverage3 s 1II 1IIM1a&1a&)DAAABU2YYb ***RXrx(((i q1a&4PPPBU2YYb ***i 4888BU2YYb ***i $y//DQQQBU2YYb ***aVaV$e,,C5%.4-!@AAs###TC:&&&1%%%US#J'''4(((UG$$$1%%%US#J'''''rX c z t j g d dd }t j |dgdgdgg }t j g d dd }t ||dd }t d gd
gd ggdgdgdgg }t
|| d S ) Nr; r r r r> FTr )r r r )r r keepdimsr r r )rQ r
r r_ r r r )rR r rB wactualdesireds rV test_testAverage4z TestAverage.test_testAverage4 s HYYY ' '1 - - EKK%5'D6 :K ; ; HYYY ' '1 - -AA=== rdRD 1UGeWtf3MNNVW%%%%%rX c t t ddg }t |d t t g dg d }t |d d S )Nr> r; r )r> r; r r FFTTr )r r
r rR rA s rV test_onintegers_with_maskz%TestAverage.test_onintegers_with_mask sd E1a&MM " "Q E,,,-G-G-GHHH I IQ rX c t j g dg dgt }t g dg dg| }t | }t j | }t
|j |j t
|j |j t |d }t |j d t |j d d z z }t
|j |j t
|j |j t |d
}t |j d
t |j d
d z z }t
|j |j t
|j |j t j g d g d g } t ||
}
t j | | |
}t
|
j |j t
|
j |j t || d } t |j | d t |j | d d z z }t
| j |j t
| j |j t || d
} t |j | d
t |j | d
d z z }t
| j |j t
| j |j d S )N)r r r r> r r r> r r r r< )r y ? @y @ @y @ @y @ @)y "@ ?y @ @y @ @y @ @r r r r r> )r r r r r )r r r r r r r r ) rQ r
r r r ru r realimag)
rR r@ rA avexpectedav0 expected0av1 expected1r wavwav0wav1s
rV test_complexzTestAverage.test_complex s x ( *157 7 7 5556668 "
$
$
$ QZZ:allnn--BGX]333BGX]333aa AF+++gaf1.E.E.Eb.HH CHin555CHin555aa AF+++gaf1.E.E.Eb.HH CHin555CHin555 h1111113 4 4a%%%:allnnc4%jAAACHhm444CHhm444q#A... QVSq999 QVSq999"<= DIy~666DIy~666q#A... QVSq999 QVSq999"<= DIy~666DIy~66666rX z r; r Nr )r r r> g ? @)r> r; r )r> r r )r r r r> r )r r g #@)r r r c V t j ||d }|j t j | k sJ t || t j |||d }|j t j | k sJ t || t j |||dd \ }} |j t j | k sJ t || | j t j | k sJ t | | d S )NT)r r )r r r )r r r r )rQ r_ r r r )
rR r r expected_avgr
expected_wavg
expected_wsumavgwavgwsums
rV test_basic_keepdimsz TestAverage.test_basic_keepdims1 s emmAD4m88yBH\ 2 222223 ---u}}QT7T}JJzRXm 4 444444 ///U]]14,04 # A A
dzRXm 4 444444 ///zRXm 4 444444 /////rX c t j t j d dd g dg dg dg }t g dd }t g dg d }t
|d|d }t j g d
}t || t
|d|d }t j g d }t || t j g d g d
}t
|dddt j g }t |d t j g d g dg dgg dg dg dg }t j |dt j dgd }t j dt j t j dgg d }t || t |j
|j
d S )Nr r r> r r r r r r )r Fr r ) @g @g @)r g0N@g0N@)r r r r r r> r r )r r @r )g "@r r r )FTTF)TFTT)TFTFr r @) rQ r_ r
r r r r r nanr r@ ) rR rA weights_unmaskedweights_masked avg_unmaskedexpected_unmasked
avg_maskedexpected_masked avg_expecteds rV test_masked_weightsz TestAverage.test_masked_weightsH s>
EKK ! ,,Q22 'iiIII >
@
@' %@@@%kkk BBB qq'7%I I I H%<%<%< = =L*;<<< QQOOO
(#N#N#NOOJ888
EKK111222
4
4 q1aBF*;<<< L#... EKK
' & & % % % & & &
U]]1q"&!n1]EE
u{{C#=#=#=#= # ? ? J 555Z_l&788888rX )r r r r r r r r r pytestmark parametrizer r r\ rX rV r r s
$ $ $& 4 4 4D( ( (0& & & *7 *7 *7X [ F
))TC5)))dVcU ; 99jjj
!1&7 a&???#iii[
2 3 0 0
0")9 )9 )9 )9 )9rX r c e Zd Zd Zd Zd ZdS )TestConcatenatorc t t t d t g d t d }g d}t || }t |dd|f }t t
|t t |g d t |j t |dd|f d S )Nr> 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> ) r r r
r r r
isinstancer
r@ )rR rB mdrZ s rV test_1dzTestConcatenator.test_1dw s 3 /0%8J8J8J2K2KLLL GG OO # # # 1a
O
1k**+++1 B B BCCC163q!Qz?33333rX c t j dd }t j dd }t j t j dd d }t j t j dd d }t || }t || }t
d||f }t
|j dk t |d d d df | t |d d dd f | t |j t j
d||f t
||f }t
|j dk t |d dd d f | t |dd d d f | t |j t j
||f d S )Nr r r 1)r rk )rk r ) rQ randomrandroundr r r r r r@ r_)rR a_1a_2m_1m_2b_1b_2r s rV test_2dzTestConcatenator.test_2d s innQ""innQ""hry~~a++Q//hry~~a++Q//3S)))3S))) S#
7"###1QQQU8S)))1QQQU8S)))1625c3 #7888 SM7"###1RaRT7C(((1QRRT7C(((1625c?33333rX c R t t j j df }t |j ddg t |j d d t ddgt j j f }t |j g d t |j d d ddg d S )Nr> TFr; )FFT)r rQ r_ r r r@ rs )rR r s rV test_masked_constantz%TestConcatenator.test_masked_constant s RU\1_%V[4-000V[^Q'''aVRU\)*V["6"6"6777V[!_q!f-----rX N)r r r r r r r\ rX rV r r t sA 4 4 44 4 4(. . . . .rX r c e Zd Zd Zd ZdS )
TestNotMaskedc L t t j d dd g dg dg dg dg dg }t |d }t |d d
g t |d }t |d dd g t |d d
d g t |d }t |d d dg t |d d dg t |j d }t |d d
g t |j d }t |d dd g t |d dd g t |j d }t |d d dg t |d d dg t |d< t |d }t |d dd g t |d dd g t |d }t |d ddg t |d ddg d S )N 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 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 r r r r )r> r> r; r r )r r> r; r )r r r; r )r r; r r )r rQ r r r/ r rs r rR rs rS s rV
test_edgeszTestNotMasked.test_edges s; BIbMM 1 1!Q 7 7"1//"1//"1//"1//"1// "35 5 5 tT**TAr7### tQ''T!W @AAAT!W @AAA tQ''T!W @AAAT!W @AAA ty$//TAr7### ty!,,T!W @AAAT!W @AAA ty"--T!W @AAAT!W @AAAR tQ''T!W @AAAT!W @AAA tR((T!W|\ :;;;T!W|\ :;;;;;rX c | t t j d dd g dg dg dg }t |d }t |t
dd d t
d
d d t
d dd g t |d }t |t
dd
d t
ddd gt
dd
d t
ddd gt
dd
d t
ddd gt
dd
d t
ddd gt
ddd gt
ddd gg t
ddd gg t |d
}t |t
dd d gg t
ddd t
ddd gg d S )Nr 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> r r r r r> r; r ro )r rQ r r r. r r )rR rA tmps rV test_contiguousz TestNotMasked.test_contiguous s 2 ..q!44 7 7 7 7 7 7 7 7 7 9
:
:
: #1d++S !Q "b$ "b$
#1a((S
1a
aD 1 1 2
1a
aD 1 1 2
1a
aD 1 1 2
1a
aD 1 1 2
1a
1a
1a
#1a((S
1a
1a
aD 1 1 2
rX N)r r r r! r' r\ rX rV r r s3 < < r> r> r )r r; r r )rk
) )( * + , )2 4 5 6 )7 9 : ; r r )r r ro r r )rk r r. r/ r0 )r1 r# r2 r3 r4 )r5 ) r6 r7 r8 )- . / 0 1 )r9 3 r: r; r< )r= 8 r> r? r@ r> ) r$ r% r ) r ! " )# $ % & ' r[ r )r r; )r ro r r )rI r$ r% r )r r )rK rL rM rN )rO rQ rR rS )rB rD rE rF r: r )r r r> )r r r> r; )r r; )r> r )r r r r; )r r )
rQ r
listrp r r astyper r( r )rR r r rA a2a3a4s rV test_compress_ndz&TestCompressFunctions.test_compress_nd s^ HT%,,'' ( ( 0 0Aq 9 9 HW $ $T * *% !!
NNQ*********, +********,- . . .
1 Q............0 /...........0 1 2 2 2
1 Q.........0 /........0 /........0
1 2 2 2 D
!
!
B
E
"
"QQQ
1 Q///)//******, +***********, +***********, - . . . D
!
!
B
E
"
"QQQ
6 " "Q.........0 /........01 2 2 2 G
$
$Q
6 " "Q*********, +********, +********,
- . . . G
$
$
G
$
$
H
%
%QQQ
6 " "Q************, +***********, - . . . G
$
$QrX c t t j d dd g dg dg dg }t t | ddgdd gg t t |d
g d g d g t t |d
d
dgddgdd gg t |j g dg dg dg }t t | d
dgdd gg t t |d
g dg d g t t |d
d
dgddgdd gg t |j g dg dg dg }t t | d gg t t |d
g d g t t |d
dgdgd gg t |j g dg dg dg }t t | j d
t t |d
j d
t t |d
j d
d S )Nr r r r r r r ro r r r, )r ro r r> r; r r> r r r r> r; r r r> )r
rQ r r r r) _datar rR r s rV test_compress_rowcolsz+TestCompressFunctions.test_compress_rowcolsX sv ")A,,&&q!,,! 999iii8
:
:
:%a((Aq6Aq6*:;;;%a++iii-CDDD%a++q!fq!fq!f-EFFF !'IIIyyy A B B B%a((Aq6Aq6*:;;;%a++iii-CDDD%a++q!fq!fq!f-EFFF !'IIIyyy A B B B%a((A3%000%a++iii[999%a,,sQC!o>>> !'IIIyyy A B B B%a((-q111%a++0!444%a++0!44444rX c t t j d dd g dg dg dg }t t | j g dg dg dg t t |d j g dg dg dg t t |d j g dg dg dg t |j g dg d g dg }t t | j g d g dg d g t t |d j g dg dg dg t t |d j g d g d g d g t |j g dg d g dg }t t | j g dg dg d
g t t |d j g dg dg dg t t |d j g d
g d
g d
g t |j g dg d g d g }t t | t u t t |d t u t t |d t u t t | j t t |d j t t |d j d S ) Nr r r r r r- r r> r` )r> r> r rb ) r
rQ r r r r* r@ rc r allr rd s rV test_mask_rowcolsz'TestCompressFunctions.test_mask_rowcolsl s ")A,,&&q!,,! 999iii8
:
:
:\!__) iiIII6 8 8 8\!Q'', iiIII6 8 8 8\!Q'', iiIII6 8 8 8 !'IIIyyy A B B B\!__) iiIII6 8 8 8\!Q'', iiIII6 8 8 8\!Q'', iiIII6 8 8 8 !'IIIyyy A B B B\!__) iiIII6 8 8 8\!Q'', iiIII6 8 8 8\!Q((- iiIII6 8 8 8 !'IIIyyy A B B B Q##%%/000 Q""&&((F2333 Q""&&((F2333 Q$((**+++ Q""'++--... Q""'++--.....rX r )Nr r> func rowcols_axisr r> c 6 t t j d dd g dg dg dg }t t
5 ||| }t
|t || d d d d S # 1 swxY w Y d S )Nr r r r r r )r
rQ r r r DeprecationWarningr r* )rR r ri rj r ress rV #test_mask_row_cols_axis_deprecationz9TestCompressFunctions.test_mask_row_cols_axis_deprecation s
")A,,&&q!,,! 999iii8
:
:
: ,
-
- = =$qt$$$C l1l ; ; < < < = = = = = = = = = = = = = = = = = =s ,BBBc x t j dd }g d}t || dd }t || dd }t ||d }t |j ddgdd gg t ||d }t |j g d
g d g d g t ||d }t |t j | d | d t ||d }t |t j | d | d g d
}t || dd }t || dd }t ||d }t |j d dgddgg t ||d }t |j g dg dg d
g t ||d }t |t j | d | d t |t || t ||d }t |t j | d | d g d}t || dd }t || dd }t || }t |j t t || }t |j t t |g d dd }t |g d dd }t ||d }t |j ddgd d gg t ||d }t |t j | d | d t ||d }t |j g d g d g d g t ||d }t |t j | d | d t |g d
dd }t |g d dd }t ||d }t |j d d gddgg t || }t |t j | d | d t ||d }t |j g dg dg dg t ||d }t |t j | d | d t |g d
dd }t |g d dd }t ||d }t |j dd gddgg t ||d }t |t j | d | d t ||d }t |j g dg d
g dg t ||d }t |t j | d | d t t j d ddd dd gd d ggd d gd d ggg }t t j d ddd d d gd d ggd d gd dggg }t ||d }t |j ddgddggd d gd dgggd d gd dggd d gd dgggg t ||d }t |j d d gd dggd d gd d gggd d gd d ggd d gd d gggg t ||d }t |j dd gd d ggdd gd d gggdd gd d ggddgddgggg t ||d }t |j d d gd d ggd d gd d gggd d gd d ggdd gd d gggg t t j d ddd dd gd d ggd d gd d ggg }d}t ||d }t |j dd gd d ggd d gd d ggg t ||d }t |j dd gd d ggd d gd d ggg t ||d }t |j dd gd d ggd d gd d ggg t ||d }t |j dd gd d ggd d gd d ggg t t j d ddd dd gd d ggd d gd d ggg }t t j d d dg }t ||d }t |j ddgddgg t ||d }t |j dd gd d gg d S )Nr> ro )r> r r r r r r r; r T)strictr r- r Fr r r r r r> rb )r r r r r r )r r r> r r r r r ) rQ r r r r r r@ filledr )rR nr rA rB rZ s rV test_dotz TestCompressFunctions.test_dot s IaOO # # # + +Aq 1 1 # # # + +Aq 1 1 1T " " "QVq!fq!f -... 1T " " "QViiiIII >??? 1U # # #Qqxx{{AHHQKK88999 1U # # #Qqxx{{AHHQKK88999 # # # + +Aq 1 1 # # # + +Aq 1 1 1T " " "QVq!fq!f -... 1T " " "QViiiIII >??? 1U # # #Qqxx{{AHHQKK88999QAq """ 1U # # #Qqxx{{AHHQKK88999 # # # + +Aq 1 1 # # # + +Aq 1 1 1IIQVV$$$ 1IIQVV$$$ !3!3!3 4 4 4 < ??? 1U # # #Qqxx{{AHHQKK88999 !3!3!3 4 4 4 < ??? 1U # # #Qqxx{{AHHQKK88999 !3!3!3 4 4 4 < ??? 1U # # #Qqxx{{AHHQKK88999 1 --aA66!"AA /1a&1a&1A B
D
D
D 1 --aA66!"AA /1a&1a&1A B
D
D
D 1T " " "QVAA'1a&1a&)9:AA'1a&1a&)9:< = = =
1U # # #QVAA'1a&1a&)9:AA'1a&1a&)9:< = = =
1T " " "QVAA'1a&1a&)9:AA'1a&1a&)9:< = = =
1U # # #QVAA'1a&1a&)9:AA'1a&1a&)9:< = = =
1 --aA66!"AA /1a&1a&1A B
D
D
D 1T " " "QV1v1v .!Q!Q0@ ABBB 1U # # #QV1v1v .!Q!Q0@ ABBB 1T " " "QV1v1v .!Q!Q0@ ABBB 1U # # #QV1v1v .!Q!Q0@ ABBB 1 --aA66!"AA /1a&1a&1A B
D
D
D 1 QF 3 3 3 1T " " "QVq!fq!f -... 1U # # #QVq!fq!f -.....rX c t j d }t | }t t t || t u t t t || t u t t t || t u t t t || t u d S )Nr )rQ eyer
r r` r r
rR rA rB s rV test_dot_returns_maskedarrayz2TestCompressFunctions.test_dot_returns_maskedarray s F1II !HHSAYY;.///SAYY;.///SAYY;.///SAYY;./////rX c t t j d }t t j d }t ||| }t ||u t
|| d S )Nr )r r out)r
rQ rv r r r r )rR rA r{ rm s rV test_dot_outz"TestCompressFunctions.test_dot_out sj "&)) BHV$$%%!QC s
Q rX N)r r r r^ re rh r r r rQ r_ mask_rows mask_colsrn rt rx r| r\ rX rV r) r) s n n n `5 5 5( / / /@ [ V\\\22 [ fn 5 !u 2RU_a4H IK K= =K K 32=m/ m/ m/^0 0 0 rX r) c e Zd Zd Zd ZdS )TestApplyAlongAxisc t d ddd }d }t |d| }t |ddgddgg d S ) Nr r; r c | d S Nr> r\ )rB s rV myfuncz*TestApplyAlongAxis.test_3d..myfunc s Q4KrX r> r ro rk r r r' r rR rA r xas rV test_3dzTestApplyAlongAxis.test_3d sd 3KK 1a ( ( fa
+
+R1a&1b'*+++++rX c t d ddd }d d}t |d|d }t |ddgd d
gg d S ) Nr. r; r r c | d|z S r r\ )rB offsets rV r z1TestApplyAlongAxis.test_3d_kwargs..myfunc' s QvX; rX r> )r r r r )r r r s rV test_3d_kwargsz!TestApplyAlongAxis.test_3d_kwargs$ sn 2JJ q!Q ' ' fa1
5
5
5R1a&1b'*+++++rX N)r r r r r r\ rX rV r r s2 , , ,, , , , ,rX r c e Zd Zd ZdS )TestApplyOverAxesc t d ddd }t t j |ddg }t j dgdgdggg }t
|| t ||dz t < t t j |ddg }t j d gd
gdggg }t
|| d S ) Nr r; r r r r+ \ | r r8 )
r r r& rQ rt r
r r rZ r )rR rA rS ctrls rV
test_basicz TestApplyOverAxes.test_basic0 s 2JJ q!Q ' ' rvq1a&11x2$se,-..T4 "(1q5..
rvq1a&11x2$rd+,--T4 rX N)r r r r r\ rX rV r r . s# ! ! ! ! !rX r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zej dg d d Zd Zd Zd Zd Zd Zd Zd Zd Z dS )
TestMedianc t j t j t j gt j t j ggd }t |t j d S )Nr r )rQ r_ r infr rR rs rV test_pytypezTestMedian.test_pytype< sH ELL2626*RVRV,<=BL G GQ rX c * t j t j t j t j gt j t j gg d }t |t j t j t j t j t j gt j t j gg d }t |t j t j t j t j t j gt j t j ggd d }t |j d t j t j t j t j gt j t j ggd d }t |j d d S )Nr r Tr )rQ r_ r r r r r@ r s rV test_infzTestMedian.test_inf@ s ELL++bfbf-=.0fbf-=-? @ @FH
J
JQ ELL++bfbf-=.0fbf-=-? @ @FJ
L
LQ ELL++bfbf-=.0fbf-=-?EI , K K
"
" QVT""" ELL++bfbf-=.0fbf-=-?EI , K K "
$
$ QVT"""""rX c j t j d }t t j | d t t
t j | t u t d }t t j | d t t
t j | t u d}t t j | d t t
t j | t u t j d dd }t t j |d t j |d t t j |d
t j |d
t t j |d
t u t j d dd }t t j |d t j |d t t j |d
t j |d
t t j |d
t u d S ) Nr r r r r r H r r r> g R@)
rQ r r r_ r r r` r
rp r rd s rV test_non_maskedzTestMedian.test_non_maskedR s: IaLLRU\\!__b)))RU\\!__%%[8999 !HHRU\\!__c***RU\\!__%%[8999
RU\\!__b)))RU\\!__%%[8999 Ie $ $Q * *RU\\!!\,,bi.B.B.BCCCRU\\!!\,,bi.B.B.BCCC QQ ''{:;;; If % %a + +RU\\!!\,,bi.B.B.BCCCRU\\!!\,,bi.B.B.BCCC QQ ''{:;;;;;rX c @ t t j d dgdz dgdz z }t t j | d t t j | j dd t t t j | t u t t j d
d
d dgd z dgdz z }t t j | d
t t j | j dd t t t j | t u t j |dd }t |ddg t |j dd t t | t u dS )z5test the examples given in the docstring of ma.medianr r r r> r r r\ shape mismatchrk r; r r r r Tr overwrite_inputr r r: N) r
rQ r r r_ r r r r` r
r )rR r ma_xs rV test_docstring_examplesz"TestMedian.test_docstring_examplesg s ")A,,aSUaSU] 3 3 3RU\\!__c***RU\\!__*B0@AAARU\\!__%%[8999 ")B--''1--QCEQCEM B B BRU\\!__c***RU\\!__*B0@AAARU\\!__%%[8999u||AB|==TB8$$$TZ'7888T
k)*****rX c . d}t d D ]}dD ]}t d|| }t j t | | ddg }|D ]L\ }} t j ||| )# t $ r t |||||fz w xY wt j |dz |gddg }|D ]R\ }} t j ||| t |||||fz # t j $ r Y Ow xY wd S ) Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr )FTr> )ndminr@ FTr )
rp r
itertoolsproductrQ r_ r ExceptionAssertionError AxisError)rR msgr r@ r argsr overs rV test_axis_argument_errorsz$TestMedian.test_axis_argument_errorsv s E1XX N NE%
N
N!5t444 !(vu)=)=t}MM"& N NJD$N QT4 HHHH$ N N N ,SD%t3L-L M MMN !(EAI,)>
NN"& N NJD$N QT4 HHH -SD%t3L-L M MM < N
N N Ns "A< Fr T)r
r rQ r_ r r rd s rV test_masked_0dzTestMedian.test_masked_0d sh !% RU\\!__a((( !$ RU\\!__bel33333rX c t t j d d }t t j | t j j t t j | j dd t t t j | t j j
j u t t j d d }t t j | d t t j | j dd t t t j | t u t t j d g d }t t j | d t t j | j dd t t t j | t u t t j d g d
}t t j | d t t j | j dd t t t j | t u t t j d g d }t t j | d
t t j | j dd t t t j | t u t t j d g d }t t j | d
t t j | j dd t t t j | t u t t j d g d }t t j | d t t j | j dd t t t j | t u t t j d g d }t t j | d t t j | j dd t t t j | t u d S )Nr Tr r\ r Fr r r )r r> r> r> r> r )r r> r> r r r r r r r )
r
rQ r r r_ r r r r r` ra MaskedConstantr
rd s rV test_masked_1dzTestMedian.test_masked_1d s# ")A,,T * * *RU\\!__bel333RU\\!__*B0@AAARU\\!__%%)BBCCC ")A,,U + + +RU\\!__b)))RU\\!__*B0@AAARU\\!__%%[8999 ")A,,[[[ 1 1 1RU\\!__c***RU\\!__*B0@AAARU\\!__%%[8999 ")A,,[[[ 1 1 1RU\\!__b)))RU\\!__*B0@AAARU\\!__%%[8999 ")A,,[[[ 1 1 1RU\\!__b)))RU\\!__*B0@AAARU\\!__%%[8999 ")B--kkk 2 2 2RU\\!__b)))RU\\!__*B0@AAARU\\!__%%[8999 ")A,,]]] 3 3 3RU\\!__c***RU\\!__*B0@AAARU\\!__%%[8999 ")B--mmm 4 4 4RU\\!__c***RU\\!__*B0@AAARU\\!__%%[899999rX c
t t j t g dg d j t j t g dg d j d S )Nr r r r` )r rQ r_ r r
r rR s rV test_1d_shape_consistencyz$TestMedian.test_1d_shape_consistency st RU\\%WWW"="="=>>DU\\%WWW"="="=>>D G G G G GrX c d\ }}t t j dd| }t x|d d< |dd < t t j ||ft
}|d d |d d df< t j t | }t d| D ]0}t j
| || |d d |f< 1t t |d d df d t t | d t t |d t j
| t t |j d t j
| d S )
N)e rK r rk r< r r> r )r rQ linspacer emptyrP r rb rp r shuffler r r T)rR rs pr r idxrv s rV r zTestMedian.test_2d s\ A S"a00 2 2!!#2#344 1a&666 7 7AAA$!!!Q$iA q! A I c " " "fAaaadGGVAaaadG__a(((VAYY"""VAA&&& 444VACa((("(1++66666rX c ( t t j d dd }t x|d d< |dd < t t
| d t t t j | t u t t
|d g d t t t j |d t u t t
|d g d
t t t j |d t u t t
|d j g d d S ) NrK rk r -@r r ) +@r g /@r> )
r r r rk r/ r# r4 r r r )
r> r> r> r r r r r> r> r> ) r rQ r r r r r r r` r_ r
r@ rd s rV
test_2d_waxiszTestMedian.test_2d_waxis sM 2 ..r155 6 6 "1""##VAYY%%%RU\\!__%%[8999VAA&&&(:(:(:;;;RU\\!!\,,--<===VAA&&&(J(J(JKKKRU\\!!\,,--<===VAA&&&+-K-K-KLLLLLrX c t j d ddd }t ||dz dk < t t
|d ddgdd gddgd
d gg d |_ t t
|d d d
gdd gddgg t j d ddd }t ||dz dk < t t
|d dd
gddgddgg d S )Nr r r r; r r. r r r1 r3 )r r r; c rk r r/ r0 r r r# r2 )rQ r_ r r r r r r rd s rV r zTestMedian.test_3d s ELL $ $Q1 - - !a%1*
VAq\\RGaWr1gBx#HIIIVAq\\RHr2hR#ABBB ELL $ $Q1 - - !a%1*
VAq\\RHq!fr2h#?@@@@@rX c t t j d dd }t x|d d< |dd < t t
|d t
|d d S )NrK rk r r r r r> )r rQ r r r r r rd s rV
test_neg_axiszTestMedian.test_neg_axis sq 2 ..r155 6 6 "1""##VAB'''):):):;;;;;rX c dD ]}t t j | }t x|d d< |dd < t t j d }t || }|dk rt
|d nt
|d t ||u t t | t u d S ) N)rK >@r g ?@r r r\ rz rK r g .@)
r rQ r r r r r r r` r
)rR vr r{ r s rV test_out_1dzTestMedian.test_out_1d s #
,
,A RYq\\**A # #AbqbEAbccF rwr{{++Cqc"""ABww S$'''' S#&&& AH DGG{* + + + +
,
,rX c l dD ]/}t t j | dd }t x|d d< |dd < t t j d }t
|d| }|dk r4t d gdz g d
z d gdz z d gdz d gd
z z d gdz z }n3t d gdz g dz d gdz z d gdz d gd
z z d gdz z }t || t ||u t t | t u 1d S )N)r5 g D@rK r rk r r r r> )r r{ rK r )rk r/ r# r4 TFr r )r g 1@g 5@g 9@) r rQ r r r r r r r r` r
)rR r r r{ r es rV test_outzTestMedian.test_out sx #
,
,A RYq\\ 1 1"b 9 9::A # #AbqbEAbccF rwr{{++Cqqc***ABww "a*:*:*:!:bT!V!C'+fqjE7Q;&>$!&KM M M !"a*B*B*B!BbT!V!K'+fQh%&:dVAX&EG G G A AH DGG{* + + + +
,
,rX r )Nr> r[ rV )r r )argnames argvaluesc t j dt }t j d t j |j d d d f z }| t j }t j |t | < t t j |j | | dj z }nCt |j t fdt j D }t t j | }t! |d| }||u sJ t# |j | d S ) N)r r ro r r< )r r r[ c 3 > K | ]}|v rd n j | V dS )r> N)r ).0rv axis_normr s rV z/TestMedian.test_keepdims_out.. sN L L89Q)^^ L L L L L LrX T)r r r{ )rQ r r r r
r rZ intpr tupler r ndimr rp r r r ) rR r r@ r shape_outr{ r r r s @@rV test_keepdims_outz TestMedian.test_keepdims_out
sP x
T222 I X & &$*)=)=aaag)F F
HHRW U1XX ,,4 8 8 8 < qv
II,T16::I L L L L L=B16]] L L L L LI28I..//t===}}}}V\9-----rX c ddgddgddgg}t j |d }ddg}t t j |d | d S )Nr r r r r )rQ r_ masked_equalr r )rR rs rf r s rV $test_single_non_masked_value_on_axisz/TestMedian.test_single_non_masked_value_on_axis& ss RRR U''a00
825<<
<;;# % % % % %rX c x dt j dt fD ]"}t j dt j dgg dg }||_ t j |d }t t j | t |t j t j | d }t t j | t |t j t j |d }t t | t t |dt j dg t j |d }t t | t t |t j d g t j |d
}t t | t t |t j d g $t j dt j dgg dg }t j j |d d d f< t t j |d t j t t j |d dt j dg t t j |d t j d g d S ) NFr r< r> r r r r r; r r )rQ r r r_ r
r r@ r r isscalarr ravelr r` r
r )rR r@ dmr s rV test_nanzTestMedian.test_nan/ s[ BHQd 3 3 34 / /Dq"&!niii 899BBG Rd ++A BKNN # # # q"& ) ) ) RXXZZa 00A BKNN # # # q"& ) ) ) Ra ((A a+ . . . q1bfa. 1 1 1 Ra ((A a+ . . . q261+ . . . Rb ))A a+ . . . q261+ . . . .
U[[1bfa.)))4
5
55<111a425<<<66???25<<<33a^DDD25<<<33bfc]CCCCCrX c t j t j d }t j t j d }t j |d< t j j |d< t t j |d| | t j t j d }t t j |d| | t j t j d }t t j || | d S )
N)r r r r; r> rI r rz r r> r\ ) rQ r_ r r r r r r r )rR or s rV test_out_nanzTestMedian.test_out_nanL s E rx~~ . . E rwv / /&$%,$RU\\!QA\..222 E rx~~ . .RU\\!QA\..222 E rx|| , ,RU\\!\++Q/////rX c b t j t j dt }t j j |d d d< t j |d< t t j | t j t t j |d t j t j t j dt ddd }t j |j
dz dk |_ | }t j |d < t j |d
< t t j | t j t t j t j | t j |d }t j |d < t j |d < t t j |d | t j |d }t j |d
< t j |d < t t j |d | t j |d }t j |d< t j |d< t t j |d | d S )Nr r< r r; r r r r> r )r> r> r; r; r rW r> r rX )rQ r_ r r rP r r r r r r r@ copyr r r )rR rA aorigrB s rV test_nan_behaviorz TestMedian.test_nan_behaviorW s3 E ry5 9 9 9 : : ##A#v!25<<??BF33325<<<22BF;;; E ry5 9 9 9 A A!Q J J K K16""Q&!+V'
V'
25<<??BF333 BELLOO,,---
ELLQL ' '&$&$RU\\!Q''+++
ELLQL ' '&$&$RU\\!Q''+++
ELLVL , ,v!v!RU\\!V,,a00000rX c & t j g dg dgt j }t j ||dk }t t j |d d t t j |d j d t t j |d |d t t j | d d S )
N)r r r< r r r> r r Fr )rQ r
uint8r_ r r r r@ r s rV test_ambigous_fillz TestMedian.test_ambigous_fillz s Hkkk;;;/rx @ @ @ E qqAv . .25<<<22C88825<<<227???25<<<22AaD99925<<??C00000rX c t j t j fD ]}t j |t j gt j t j gg }t j |t j | }t t j |d |t j g t t j |d |t j g t t j | | t j t j t j |gt j t j |gg }t j |t j | }t t j |d | t t j |d j
d t t j |d |d t t j | | t j ||g||gg }t t j | | t t j |d | t t j |d | t j |d| dgdt j t j d gd
t j t j |ggt j }t j |t j | }|dk rbt t j |d d d
| dg t t j | d nat t j |d dd
| dg t t j | d t t j |d dd|g t dd D ]}t dd D ]}t j t j g|z |g|z z gdz }t j |t j | }t t j | | t t j |d | t t j |d t j g|z |g|z z d S )Nr r r r> Fro ir r r r< r r r g @g $g "g r rk r; )
rQ r r
r r_ r isnanr r r r@ float32rp )rR r rA rv rw s rV test_specialzTestMedian.test_special sh FRVG$ ( = ( =C3.2626*:;<!?!!CDDA**128A;;*??A a #666 aa !8!8#>>> aa !8!8#%6(Q,3%!)!;= = = = =
=C( = ( =rX c x t j t j g t }t 5 }| t t t j | t j
t |j d j
t u d d d n # 1 swxY w Y t j t j g t d }t 5 }| t t j ddt t t j | t j
t |j d j
t u d d d n # 1 swxY w Y t j t j g t d }t! t j |d | t! t j |d | t j t j t j
t d }t j d
5 }t j ddt t! t j |d | t |d j
t u d d d d S # 1 swxY w Y d S ) Nr< r r )r= r always r; r r> T)record)rQ r_ r r
rP r r RuntimeWarningr r r r logcategorywarningsfilterwarningsr catch_warnings)rR rA r rB s rV
test_emptyzTestMedian.test_empty s E rx% 8 8 8 9 9
9A
HH^ $ $ $ ru||A 7 7 7 AE!H% 7 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9
E rx%q A A A B B
9A
HH^ $ $ $ #Hb. A A A ru||A 7 7 7 AE!H% 7 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9
E rx%q A A A B BRU\\!!\,,a000RU\\!!\,,a000
E rxe1 E E E F F
$D
1
1
1 5Q #Hb. A A A aa 00! 4 4 4 AaDM^3 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s9 A8C CC BF<<G G 6A,L//L36L3c t j t j d }t t t j | t t t j
|d< t t t j | t t d S )Nr r; ) rQ r_ r r r r` r rZ r^ rP r )rR r s rV test_objectzTestMedian.test_object s E ry}} - -RU\\!((6"2"23344e<<<v!RU\\!((6"2"23344e<<<< > > %d ; ; ; = = = = =rX c j | j dd }t t j | t | t t j |d t |d t t j |dd t |dd d S )Nr r Fr Tr )rs r r rQ r rd s rV test_2d_without_missingz TestCov.test_2d_without_missing s I a # #BF1IIs1vv...BF1U 3 3 3S55I5I5IJJJBF1U > > > %d ; ; ; = = = = =rX c
| j }t |d< || z }| }t t j | t
| t t j |d t
|d t t j |dd t
|dd t
|d n# t $ r Y nw xY w|dd }t t j ||d d d t
||d d d t t j ||d d d d t
||d d d d t t j ||d d d dd t
||d d d dd d S )Nr Fr Tr allow_maskedr> )rs r meanru r rQ r
ValueError)rR r nxs rV test_1d_with_missingz TestCov.test_1d_with_missing s I" QVVXX
\\^^BF2JJA///BF2e 4 4 4c!E6J6J6JKKKBF2e$ ? ? ? %d ; ; ; = = = & & & & & D qtWBF2r$$B$x 0 0#a44R4//BBBBF2r$$B$x > > > 1TTrT75 9 9 9 ; ; ;BF2r$$B$xD I I I 1TTrT75t D D D F F F F Fs C' '
C43C4c " | j }t |d< | dd }t j t | t }t j ||j }||
d d d d f z
d }t t | t j
| |j d dz
z |dz
z t t |d t j
|d |j d z |z t j |j | }||
d z
d }t t |d
t j
|d
|j d dz
z |dz
z t t |d d t j
|d d |j d z |z d S ) Nr r r r> r r Tr Fr r )rs r r rQ logical_notr rZ rq r r r rr r r r )rR r validfracxfs rV test_2d_with_missingz TestCov.test_2d_with_missing s I"
IIaOO|A//66s;;veUW%%!&&))AAAtG$$
,
,Q
/
/CFF F2JJ!'!*q. 9TBY G I I IC - - - F2D 1 1 1AGAJ > E G G Gvegu%%!&&))m
#
#A
&
&C% 0 0 0 VBu 5 5 5 gaj1n .15 < > > > C%d ; ; ; VBu4 @ @ @ WQZ (*. / 1 1 1 1 1rX N)r r r r
r r r r# r\ rX rV r
r
sb . . .= = == = =F F F01 1 1 1 1rX r
c 8 e Zd Zd Zd Zd Zd Zd Zd Zd Z dS ) TestCorrcoefc t t j d | _ t t j d | _ d S r )r
rQ r r rs data2r s rV r
zTestCorrcoef.setup_method s@ ")..,,-- 29>>"--..
rX c | j | j }}t j | }t j || }t 5 }t j d t t t |d | t d t t j |d t |d t t |d | t t ||d | t t |d | t t ||d | d d d d S # 1 swxY w Y d S )Nr r ddof bias and ddof have no effectr r rs r' rQ r r r simplefilterr rl filterr )rR r r r expected2sups rV test_ddofzTestCorrcoef.test_ddof s y$*1;q>>K1%%
CC !( + + + +Xqr B B B B JJ)+I J J J AA 6 6 68K8K8K L L L 4 4 4h ? ? ? AB 7 7 7 C C C 3 3 3X > > > AA 6 6 6 B B B C C C C C C C C C C C C C C C C C Cs C>EEEc | j | j }}t j | }t 5 }t j d t t t ||dd t t t ||dd t t t |d | t d t t |d | d d d d S # 1 swxY w Y d S )Nr TFr r+ r> r, )rR r r r r0 s rV test_biaszTestCorrcoef.test_bias+ s y$*1;q>>
?C !( + + + +Xq!T5 I I I +Xq!T4 H H H +Xqu E E E E JJ)+I J J J 3 3 3X > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?s B(C&&C*-C*c | j }t t j | t | t t j |d t |d t 5 }| t d t t j |dd t |dd d d d d S # 1 swxY w Y d S )NFr r+ Tr )rs r rQ r r r. rl rR r r0 s rV r z$TestCorrcoef.test_1d_without_missing8 s* IBKNNHQKK888BK% 8 8 8 $Qu 5 5 5 7 7 7
FC JJ)+I J J J Ae$ G G G (5t D D D
F
F
F F F F F F F F F F F F F F F F F F Fs 8ACC Cc | j dd }t t j | t | t t j |d t |d t 5 }| t d t t j |dd t |dd d d d d S # 1 swxY w Y d S )Nr r Fr r+ Tr )rs r r rQ r r r. rl r5 s rV r z$TestCorrcoef.test_2d_without_missingC s: I a # #BKNNHQKK888BK% 8 8 8 $Qu 5 5 5 7 7 7
FC JJ)+I J J J Ae$ G G G (5t D D D
F
F
F F F F F F F F F F F F F F F F F F Fs AC**C.1C.c 4 | j }t |d< || z }| }t t j | t
| t t j |d t
|d t 5 }| t d t t j |dd t
|dd d d d n # 1 swxY w Y t
|d n# t $ r Y nw xY w|dd }t t j ||d d d t
||d d d t t j ||d d d d t
||d d d d t 5 }| t d t t j ||d d d t
||d d d d t t j ||d d d t
||d d d d
d d d d S # 1 swxY w Y d S ) Nr Fr r+ Tr r r> r r; r) ) rs r r ru r rQ r r r. rl r )rR r r r0 s rV r z!TestCorrcoef.test_1d_with_missingN s I" QVVXX
\\^^BKOOXa[[999BK5 9 9 9 $Qu 5 5 5 7 7 7
FC JJ)+I J J J Bu4 H H H (5t D D D
F
F
F F F F F F F F F F F F F F F F QU + + + + + D qtWBKBtttH 5 5x1TTrT77K7KLLLBKBtttHU C C C $Q$$B$ > > > @ @ @
>C JJ)+I J J J B44R4 9 9 (AdddG! < < <
>
>
> B44R4 9 9 (AdddG! < < <
>
>
> > > > > > > > > > > > > > > > > > >s8 -AD
DDD( (
D54D5B&J
JJc | j }t |d< | dd }t | }t j | }t |d dd df |d dd df t
5 }| t d t t |d d dd df |d dd df t t |d d dd df |d dd df t t |d d dd df |d dd df d d d d S # 1 swxY w Y d S ) Nr r r r+ r r) r> r ) rs r r r rQ r r r. rl )rR r rS rT r0 s rV r# z!TestCorrcoef.test_2d_with_missingl s I"
IIaOO{{+a..D"crcNGCRC"H,=>>>
3C JJ)+I J J J 4 4 4SbS#2#X > 'SbS 1
3
3
3 3 3 3CRC"H = 'SbS 1
3
3
3 3 3 3CRC"H = 'SbS 1
3
3
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s CEEEN)
r r r r
r1 r3 r r r r# r\ rX rV r% r% s / / /C C C ? ? ? F F F F F F > > ><3 3 3 3 3rX r% c e Zd Zd Zd ZdS )TestPolynomialc j t j d }t j d dd }t t ||d t j ||d | t }t |d< | t }t x|d< |d< t ||d d df dd
\ }}}}}t j |d d |d d df dd
\ }} }
} } t |||||f|| |
| | f D ]\ }
}t |
| t ||d d df dd
\ }}}}}t j |d d |d ddf dd
\ }} }
} } t |||||f|| |
| | f D ]\ }
}t |
| t ||dd
\ }}}}}t j |d d |d dd d f dd
\ }} }
} } t |||||f|| |
| | f D ]\ }
}t |
| t j d d z }| }|d d }|d d }|d d }t ||dd | \ }}}}}t j ||dd | \ }} }
} } t || t |||||f|| |
| | f D ]\ }
}t |
| d S )
Nrk rI r r; r r rG r r Trr r> )rr r )
rQ r r r r r viewr
r ru zipr r )rR r r CRKSDrZ r rx rz r rA a_r woxsyswss rV test_polyfitzTestPolynomial.test_polyfit s
INN2 INN2 & &r1 - -GAq! , ,bjAq.A.ABBB
FF; !
FF; $$$!F)!!Qqqq!tWad;;;Aq!Q *QqrrUAabb!eH,?,?,A,A1*.0 0 0Aq!QAq!Q?Q1aO<< ' 'GQ 2 & & & &!!Qqqq"uXqt<<<Aq!Q *QqtWa"bk14HHHAq!QAq!Q?Q1aO<< ' 'GQ 2 & & & &!!Q555Aq!Q *QqtWa"QQQiFFFAq!QAq!Q?Q1aO<< ' 'GQ 2 & & & & INN2 "
VVXX
qtW
qtW
qtW!!Q:::Aq!Q *RQTR@@@Aq!QQAq!Q?Q1aO<< ' 'GQ 2 & & & & ' 'rX c l t j d }t j d dd }t j |d< t j |d< | t }| t }t |d< t |d< t ||dd \ }}}}}t j |d
d |d
dd d f dd \ }} }
} } t |||||f|| |
| | f D ]\ }
}t |
| d S ) Nrk rI r r; r r< r Tr= r> ) rQ r r r r r> r
r r r? r )rR r r r@ rA rB rC rD rZ r rx rz r rA rE s rV test_polyfit_with_masked_NaNsz,TestPolynomial.test_polyfit_with_masked_NaNs s* INN2 INN2 & &r1 - -v!6%
FF;
FF; !%!!Q555Aq!Q *QqtWa"QQQiFFFAq!QAq!Q?Q1aO<< ' 'GQ 2 & & & & ' 'rX N)r r r rJ rL r\ rX rV r: r: s3 %' %' %'N' ' ' ' 'rX r: c n e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd ZdS )TestArraySetOpsc 4 g d}t |dd }t t |d t t |d t g dg d t |d g d t |d
g d d S ) Nr> r> r> r; r; r T return_indexreturn_inverser r r r r> )r r r r; )r r r r> r> r; )r r r r
r r r s rV test_unique_onlistz"TestArraySetOps.test_unique_onlist s !!!ddCCC
47K00111T!Wl999999 E E EFFFT!Wiii(((T!W 0 0 011111rX c t g dg d }t |dd }t |d t g dg d t |d g d
t |d g d d
|_ t g dg dd
}t |dd }t |d t g dg d t |d g d
t |d g d d S )NrP )r r r> r r> r r TrQ r )r> r; r r r r r r> r> )r r r r; r; )r r r r> r r; r )rs r@
fill_value)r r r rW r s rV test_unique_onmaskedarrayz)TestArraySetOps.test_unique_onmaskedarray s: . . .5G5G5GHHHddCCCT!Wl===||| L L LMMMT!Wlll+++T!W 0 0 0111!3!3!3!3!3!3C C CddCCCT!Wl===||| L L LMMMT!Wlll+++T!W 0 0 011111rX c t g dd }t |dd }t |d t dgdg t |d dg t |d g d t }t |dd }t |d t t t |d dg t |d dg d S ) Nr- Tr rQ r r> r; r )r r r r r s rV test_unique_allmaskedz%TestArraySetOps.test_unique_allmasked s IIID111ddCCCT!WlA5v > > >???T!Wqc"""T!Wiii((( ddCCCT!Wl6 2 2333T!Wqc"""T!Wqc"""""rX c X t t j d g d }t g dg d }t | }t || t | d | d t |j |j d S )Nr r> r r r r> r )r> r> r> r )r> r r r> r )r rQ r r
r% r rr r@ )rR r rT rS s rV test_ediff1dz TestArraySetOps.test_ediff1d s 1 OOO < < < <<<888qzzT7###T[[^^W^^A%6%6777TY -----rX c v t t j d g d }t |t }t g dg d }t
|| t
| d | d t
|j |j t |g d }t g d g d
}t
|| t
| d | d t
|j |j d S ) Nr r\ r )to_begin)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 r> r rQ r r% r r
r rr r@ rR r rS rT s rV test_ediff1d_tobeginz$TestArraySetOps.test_ediff1d_tobegin s 1 OOO < < <q6***ooo>>>T7###T[[^^W^^A%6%6777TY ---q999------4I4I4IJJJT7###T[[^^W^^A%6%6777TY -----rX c v t t j d g d }t |t }t g dg d }t
|| t
| d | d t
|j |j t |g d }t g d g d
}t
|| t
| d | d t
|j |j d S ) Nr r\ r )to_end)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 r r` ra s rV test_ediff1d_toendz"TestArraySetOps.test_ediff1d_toend s 1 OOO < < <q(((ooo>>>T7###T[[^^W^^A%6%6777TY ---q+++---4I4I4IJJJT7###T[[^^W^^A%6%6777TY -----rX c t t j d g d }t |t t }t g dg d }t
|| t
| d | d t
|j |j t |g dt }t g d g d
}t
|| t
| d | d t
|j |j d S ) Nr r\ r rd 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> r r r> r r r r` ra s rV test_ediff1d_tobegin_toendz*TestArraySetOps.test_ediff1d_tobegin_toend s, 1 OOO < < <q&999***1C1C1CDDDT7###T[[^^W^^A%6%6777TY ---qV<<<000 5 5 57 7 7T7###T[[^^W^^A%6%6777TY -----rX c t j d }t | }t g dg d }t || t t
|t t | d | d t |j |j t |t t }t g dg d }t t
|t t | d | d t |j |j d S ) Nr r> r> r> r> r r r r r r rg r r ) rQ r r% r
r r r r
rr r@ r ra s rV test_ediff1d_ndarrayz$TestArraySetOps.test_ediff1d_ndarray s" IaLLqzz <<<888T7###
4--...T[[^^W^^A%6%6777TY ---q&999***1C1C1CDDD
4--...T[[^^W^^A%6%6777TY -----rX c t g dg d }t g dg d }t || }t g dg d }t || d S )N)r> r r r rV r )r r> r> r> )r> r r rb )r
r# r )rR r r rS rT s rV test_intersect1dz TestArraySetOps.test_intersect1d+ su ,,,\\\ 2 2 2 ,,,\\\ 2 2 21a
333T7#####rX c 4 t g dg d }t g dg d }t || }t |t g d t g dg d }g d}t || }t |t g dg d t g d
}t g d }t || }t t |t
t |g d t g d
g d }t g dg d }t || }t t |t
t |g d t
g t g g d S )Nr> r; r ro r r r r r r> r r> r; r r r r rq )r r ro )r> r; r r r )r r ro r rV r )r r r r )r> r r; r )r r> r r )r r r r )r
r r r r r
r rR rA rB rS s rV
test_setxor1dz TestArraySetOps.test_setxor1d3 s """ 9 9 9 %%%,>,>,> ? ? ?1~~T5++,,, """ 9 9 9 OO1~~T5 \\\BBBCCC ))) ))) 1~~
4--...T---... ,,,\\\ 2 2 2 ,,,\\\ 2 2 21~~
4--...T---...2xB / /00000rX c t j d g d }t j g d }d|d< t || }t g dg d }t dt
}d
|d < d
|d < d
|d
< t
|| }t t |t t || t j |||j |j z }t || d S )Nr r r> )r> r; r r ) r rk rI rK r> r r r$ rM ) r r> r r> r r> r r> r )rs r@ r< Tr rb )r r; r ) rQ r r r r
r r2 r r r
r r@ )rR rA r@ rB ecrZ r s rV test_isinzTestArraySetOps.test_isinM s
IbMM ! !))) , ,x ""W
!$ ::::::
<
<
<
9D
)
)
)7 7 7 AJJ
1k**+++1b!!! GAq!&z " "afW ,1a rX c r t g dg d }t g dg d }t || }t |g d t g dg d }t g dg d }t || }t |g d
t g t g g d S ) Nrp rq r rr rq )TTTFTr r r; r> r r> r r rb )TTFTT)r
r$ r r rs s rV test_in1dzTestArraySetOps.test_in1db s """ 9 9 9 %%%,>,>,> ? ? ?AqzzT:::;;; """ 9 9 9 ***999 - - -AqzzT:::;;;2tB||,,,,,rX c t g dg d }t g dg d }t t j t || t ||d t g dg d }t g d g d
}t t j t || t ||d t g t g g d d S ) Nrp rq r rr rq T)invertry rz rb )r
r rQ r} r$ r rw s rV test_in1d_invertz TestArraySetOps.test_in1d_invertp s """ 9 9 9 %%%,>,>,> ? ? ?RYtAqzz**DAd,C,C,CDDD """ 9 9 9 ***999 - - -RYtAqzz**DAd,C,C,CDDD2tB4 8 8 899999rX c t g dg d }t g dg d }t || }t g dg d }t || t g dg dgg d g d
g }t g d g d }t g d
g d }t || }t || t g t g g d S )N)r> r; r ro r r rq r rr )r> r; r r r ro r )r r r r r r r> ra r, r rb r rq )r r> r; r r r )r
r" r r ) rR rA rB rS rT r r ezr s rV test_union1dz TestArraySetOps.test_union1d| s" %%%,>,>,> ? ? ? %%%,>,>,> ? ? ?q!}}...5J5J5JKKKT7###
999iii( 999/E F F F /// 8 8 8
%%%,>,>,>
?
?
? AqMMQ2wr2/////rX c t g dg d }t g d }t || }t |t g dg d t d }t d }t t || t dd g t g t j g }t t |g j t j d S )
N)r r r ro ro r> r; r> )r r r r r r r r> r )r; r r r r; r> r )r ro r rb rk r r )r
r! r r rQ uint32r= rs s rV test_setdiff1dz TestArraySetOps.test_setdiff1d s ***1I1I1I J J J ''' ( (AT5)))<<<=== 2JJ 1IIYq!__eQFmm444 "bib ) ) )Yq"%%+RY77777rX c t j g d }t j g d }t t || t j dg d S )N)rA rB rZ )rA rB rz rZ )rQ r
r r! rw s rV test_setdiff1d_char_arrayz)TestArraySetOps.test_setdiff1d_char_array sP H___ % % H___ % %9Q??BHcUOO<<<< r r; r< r Fru ) rQ r r
zeros_liker r r rY r4 )rR ordinary
empty_mask with_masks rV test_ndenumerate_nomaskedz)TestNDEnumerate.test_ndenumerate_nomasked s 9R==((33]84888
;;; T".2233+h//00 2 2 2T+h//00+i0011 3 3 3T+i0011+iEBBBCC E E E E ErX c
t d }t d }t d }t t t | g t t t | g t t t |d t t t j d dt gz t t t | g t t t |d t t t j d dt gz d S )Nr\ )d r Fr r r )r0 r rY r4 r? rQ ndindexr )rR rA rB rZ s rV test_ndenumerate_allmaskedz*TestNDEnumerate.test_ndenumerate_allmasked s0 rNN v y ! !T+a..))2...T+a..))2...T+aE:::;;#bj 0 0#.AABB D D DT+a..))2...T+aE:::;;#bj 3 3Y&5IJJKK M M M M MrX c
t t j d d g dg dg dg }g d}t t t
| | t t t t
|d |j t
|d D ]\ }}t || | d S )
Nr. r rj )r> r> r r> rk r ))rW r ))r; r r )r r )rI rk )r r Fr ) r rQ r r r rY r4 rb r )rR rA items
coordinatevalues rV test_ndenumerate_mixedmaskedz,TestNDEnumerate.test_ndenumerate_mixedmasked s 2 ..v66 +|| +|| +|| -
.
.
.G G GT+a..))5111Sk! > > >??@@!&III!,Q5!A!A!A / / J : . . . . / /rX N)r r r r r r r\ rX rV r r sG E E E
M
M
M
/
/
/
/
/rX r c e Zd Zd Zd Zd ZdS ) TestStackc t g dg d }t g dg d }t ||gd }t |j d t |j |d j t |j |d j t
||g }t |j |j t |j |j t ||gd }t |j d
t |j |d d df j t |j |d d d f j d S ) Nra r` r r r ro r r r r r> r r; r r5 r r r r@ r6 rs rR rA rB rZ r s rV
test_stack_1dzTestStack.test_stack_1d s3 3 3 3 3 3 3 1a&q ! ! !QWf%%%161Q49---161Q49--- Aq6NN1616***1616*** 1a&q ! ! !QWf%%%161QQQT7<000161QQQT7<00000rX c t g dd }t g dd }t ||gd }t |j d t |j |d j t |j |d j t
||g }t |j |j t |j |j t ||gd }t |j d
t |j |d d df j t |j |d d d f j d S ) Nra Tr r Fr r r r> r r r s rV test_stack_maskszTestStack.test_stack_masks s+ . . . / / / 1a&q ! ! !QWf%%%161Q49---161Q49--- Aq6NN1616***1616*** 1a&q ! ! !QWf%%%161QQQT7<000161QQQT7<00000rX c d}t j dd| }t j dd| }t j dd| t }t j dd| t }t || }t || }t
||gd }d|z } t |j | t |j
|d j
t |j
|d j
t
||gd }|dz } t |j | t |j
|d
j
t |j
|d j
d }t j dd| }t j dd| }t j dd| t }t j dd| t }t || }t || }t
||gd }d|z } t |j | t |j
|d j
t |j
|d j
t
||gd }|dz } t |j | t |j
|d
j
t |j
|d j
d S )
Nr r rk r; r r r: r> r ).r ).r> )r r; r r ) rQ r randintrZ r r r5 r r r r@ )
rR shpd1d2r r a1r[ rZ c_shps
rV
test_stack_ndzTestStack.test_stack_nd% s
Y
q"c
*
*
Y
q"c
*
*
Y
q!S
)
)
0
0
6
6
Y
q!S
)
)
0
0
6
6
"2
&
&
&
"2
&
&
& 2r( # # #s
QWe$$$27AaDI...27AaDI... 2r( $ $ $d
QWe$$$27AfIN33327AfIN333
Y
q"c
*
*
Y
q"c
*
*
Y
q!S
)
)
0
0
6
6
Y
q!S
)
)
0
0
6
6
"2
&
&
&
"2
&
&
& 2r( # # #s
QWe$$$27AaDI...27AaDI... 2r( $ $ $d
QWe$$$27AfIN33327AfIN33333rX N)r r r r r r r\ rX rV r r sA 1 1 1$1 1 1$)4 )4 )4 )4 )4rX r )N__doc__r r r numpyrQ numpy.core.numericr
numpy.testingr r numpy.ma.testutilsr r r r
numpy.ma.corer
r r r
r r r r r r r 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. r/ r0 r1 r2 r3 r4 r5 r6 r8 r r r r) r r r r
r% r: rN r r r r\ rX rV r sS
3 3 3 3 3 3 DK9 K9 K9 K9 K9 K9 K9 K9\). ). ). ). ). ). ). ).XC C C C C C C C Lo o o o o o o o d , , , , , , , ,,
!
!
!
!
!
!
!
!T= T= T= T= T= T= T= T=n @1 @1 @1 @1 @1 @1 @1 @1Fh3 h3 h3 h3 h3 h3 h3 h3V7' 7' 7' 7' 7' 7' 7' 7'te= e= e= e= e= e= e= e=P41 41 41 41 41 41 41 41n#/ #/ #/ #/ #/ #/ #/ #/LO4 O4 O4 O4 O4 O4 O4 O4 O4 O4rX