bgj$dZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z m Z mZmZddlmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZm 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>Gdd Z?Gd d Z@Gd d ZAGddZBGddZCGddZDGddZEGddZFGddZGGddZHGddZIGddZJGd d!ZKGd"d#ZLGd$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>eZdZdZdZdZdZdZdZdZ dZ d S) TestGenericc(tdt}tddgddgt}t||t jddgddgd}td|}td d gd d g|}t||td |}td d gd d ggd d gd d gg|}t||t jd dd dgfg}td|}tddgddg|}t||td|}tddgddg|}t||td |}tdggdgg|}t||dS)Ndtypemaskr=abfnamesformatsrrr>r>r;r;rArCbarCbbrCr>rH)r0floatr rnpr=)selftestcontroldts m/builddir/build/BUILD/cloudlinux-venv-1.0.7/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);7rCCCT7###$b)))k2);7rCCCT7###&++++ }oRHHHT7#####cPtjddtfgdffg}tjd|}t t |dtjjjt t |ddtjjjt t|dddt |ddj dt |ddj j ddS)NrBcr>r>rH rQr=objectmar0rtypecorer lenr _fill_valuerRmy_dtype masked_arrs rV"test_masked_all_with_object_nestedz.TestGeneric.test_masked_all_with_object_nestedDs8cc6]OT#:;<==U%%dH55 T*S/**BEJ,BCCCT*S/#.//1GHHHSC-..222Z_S)/888Z_S)5;R@@@@@rXctjdtdffg}tjd|}t t |dtjjjt t|ddt |dj dt |dj j ddS)NrBr[r>rHr\r]rds rVtest_masked_all_with_objectz'TestGeneric.test_masked_all_with_objectPs8cFD>2344U%%dH55 T*S/**BEJ,BCCCSC))1---Z_*F333Z_06;;;;;rXctddgt}t|}tddgddgt}t||t jddgddgd}td d gd d g|}t|}td d gd d g|}t||t jd dd dgfg}tddgddg|}t|}t||dS)Nr>r;r<r?rArBrCrDrGrH) rkrJrKrMrO)r rPr1rrQr=)rRbaserSrTrUs rVtest_masked_all_likez TestGeneric.test_masked_all_likeZs4aV5)))t$$AaV5999T7### Xc SzBB C Cff%VV,r;r<r) rangerQr intfullrr@datasumr,r compressed) rRrCijkjarAssls rV check_clumpzTestGeneric.check_clumpns,q! : :A1a4[[ : :Ias+++WQ--- A&&1+!+!A$$**B)))AA&& !3!3!5!5q9999fWAF !3!3!5!5q9999 : : :rXc0ttjd}t|gd<t |}t ddt ddt ddg}t |||tdS)Nrkrr>r; rrror)rrQr r r+slicerr|rRrArSrTs rVtest_clump_maskedzTestGeneric.test_clump_masked~s 2 ' ' &   AA;;a U1b\\:T7### &&&&&rXcttjd}t|gd<t |}t ddt ddg}t |||tdS)Nrkr~rrror)rrQr r r,rrr|rs rVtest_clump_unmaskedzTestGeneric.test_clump_unmaskedsy 2 ' ' &   a  A;;a .T7### (((((rXctd}t|}t|td|jgt jdt|_t|td|jgt||dk|dkz|dkz<t|}t|tddtddgt|dd<t|}t|gdS) Nrkrr<rrrr) r r-rrsizerQrboolr@r )rRrArSs rVtest_flatnotmasked_contiguousz)TestGeneric.test_flatnotmasked_contiguouss 2JJ'**TE!QV,,-..."D)))TE!QV,,-...*01q5QU qAv &''**TE!QKKq!5666!!!'**T2rXN) __name__ __module__ __qualname__rWrgrirmr|rrrr\rXrVr8r8$s$$$< A A A<<<$$$(::: ' ' ')))rXr8c eZdZdZdZdZdZdZdZe j dgdd d ggd d gd gfgdgdgdgdgddggdggdgfgdZ dZ d S) TestAveragec tgdgd}tdt|dtdt|gdt|gdd \}}td|t|d kt|dd<tt|djd gtgdgd}|d d }t|ddd f<tt|dddgtt|d jdd gtddgt|dt|dd \}}t|ddgdS)N)?@@)TFFFr@rraxis)rrrrweightsTrreturned@r;r>rrrr)r rrrr r@reshape)rRottresultwtss rVtest_testAverage1zTestAverage.test_testAverage1s$$$+F+F+FGGGS'#A...///S'#/?/?/?@@@AAAc+;+;+;dKKK S&!!!s AAAWSq))).777$$$+F+F+FGGGkk!QAAAqD WSq)))C:666WSq))).q1D6:::b"Xws333444cD999 S2r(#####rXc Ngd}gdgdg}tdtj}tt |ddtt |d|dt tdtjd tdzg}tt |dtjtjdd zd z tt |dtjdd zd z tt |d t |dt |dd zgtt |d| dtt |d|gdtt |d t |dt |dd zgtd}gd}gdgdg}td}gd} tt t||ddtt t||ddtt t||dj dgtt t|| ddttt t||ddt||} tt | ddtt | dgdtt | d ddgtt | d|gddS)Nrr>r>r>r>rr>rrrrr>rr<rr@)rrrr(@r>rg @)rrrrr$@)rrr>r>rr)rr>r>r>r>r>Tr)rrX@rrg@@)rrrrrr) r rQfloat_rrr addreducerrrr@r) rRw1w2xym1m2m3m4m5zs rVtest_testAverage2zTestAverage.test_testAverage2s      "4"4"4 5 1BI & & &WQQ'''---WQQ333S999 6!29---sVAYY? @ @WQ%%rv}}RYq\\'B'BR'G#'MNNNWQQ'''1):R)?@@@WQQ'''aa((('!!*<*<*GGGW\!R00q9993???U7<2#6#6Q???@@!DDD B  WQ%%x000WQQ''')E)E)EFFFWQQ'''#s444WQQ333222 4 4 4 4 4rXcbtd}tddz}t||g||ggdd\}}tt|t|t|j|jtt ddddgd\}}tt|t|tt dd \}}tt|t|tt dt dd \}}tt|t|t dd gdd ggt }t|d d gdd gg}t|d} t| ddgt|d} t| ddgt|d} t| dt|d} t| ddgdS)Nrrr>Tr)r;r;rrrrr)rrr;Fr?rrg@?r)r rrrrr rPr) rRrArBr1rr2ra2da2dma2daa2dmas rVtest_testAverage3zTestAverage.test_testAverage3s 1II 1IIM1a&1a&)DAAABU2YYb ***RXrx(((iq1a&4PPPBU2YYb ***i4888BU2YYb ***i$y//DQQQBU2YYb ***aVaV$e,,C5%.4-!@AAs###TC:&&&1%%%US#J'''4(((UG$$$1%%%US#J'''''rXcztjgddd}tj|dgdgdgg}tjgddd}t ||dd}t d gd gd ggdgdgdgg}t ||dS) Nr;rrrr>FTr)rrr)rrkeepdimsrrr)rQr rr_rrr)rRrrBwactualdesireds rVtest_testAverage4zTestAverage.test_testAverage4s HYYY   ' '1 - - EKK%5'D6 :K ; ; HYYY   ' '1 - -AA===rdRD1UGeWtf3MNNVW%%%%%rXcttddg}t|dttgdgd}t|ddS)Nr>r;r)r>r;rrFFTTr)rr rrRrAs rVtest_onintegers_with_maskz%TestAverage.test_onintegers_with_masksd E1a&MM " "Q E,,,-G-G-GHHH I IQrXctjgdgdgt}tgdgdg|}t |}tj|}t |j|jt |j|jt |d}t |jdt |jdd zz}t |j|jt |j|jt |d }t |jd t |jd d zz}t |j|jt |j|jtjgd gd g} t || } tj|| | }t | j|jt | j|jt || d} t |j| dt |j| dd zz}t | j|jt | j|jt || d } t |j| d t |j| d d zz}t | j|jt | j|jdS)N)rrrr>rrr>rrrr<)ry?@y@@y@@y@ @)y"@?y@@y@@y@@rrrrr>)rrrrr)rrrrrrrr) rQr rrrrur realimag) rRr@rAavexpectedav0 expected0av1 expected1rwavwav0wav1s rV test_complexzTestAverage.test_complexsx(*15777 5556668" $ $ $QZZ:allnn--BGX]333BGX]333aa   AF+++gaf1.E.E.Eb.HH CHin555CHin555aa   AF+++gaf1.E.E.Eb.HH CHin555CHin555h111111344a%%%:allnnc4%jAAACHhm444CHhm444q#A...QVSq999QVSq999"<= DIy~666DIy~666q#A...QVSq999QVSq999"<= DIy~666DIy~66666rXzr;rNr)rrr>g? @)r>r;r)r>r r)rrrr>r)rrg#@)rrrcVtj||d}|jtj|ksJt ||tj|||d}|jtj|ksJt ||tj|||dd\}} |jtj|ksJt ||| jtj|ksJt | |dS)NT)rr)rrr)rrrr)rQr_rrr) rRrr expected_avgr expected_wavg expected_wsumavgwavgwsums rVtest_basic_keepdimszTestAverage.test_basic_keepdims1semmAD4m88yBH\2222223 ---u}}QT7T}JJzRXm4444444///U]]14,04#AA dzRXm4444444///zRXm4444444/////rXctjtjdddgdgdgdg}t gdd}t gdgd}t |d|d }tjgd }t||t |d|d }tjgd }t||tjgd gd }t |dddtjg}t|dtjgd gdgdggdgdgdg}tj|dtjdgd}tjdtjtjdggd}t||t|j |j dS)Nrrr>rrrrrr)rFrr)@g@g@)rg0N@g0N@)rrrrrr>rr)rr@r)g"@rrr)FTTF)TFTT)TFTFrr @) rQr_r r rrrr nanrr@) rRrAweights_unmaskedweights_masked avg_unmaskedexpected_unmasked avg_maskedexpected_masked avg_expecteds rVtest_masked_weightszTestAverage.test_masked_weightsHs> EKK ! ,,Q22'iiIII>  @ @' %@@@%kkk BBBqq'7%III H%<%<%<==L*;<<<QQOOO (#N#N#NOOJ888 EKK111222  4 4q1aBF*;<<< L#... EKK  ' & & % % % & & &   U]]1q"&!n1]EE u{{C#=#=#=#=#??  J 555Z_l&788888rX)rrrrrrrrrpytestmark parametrizerrr\rXrVrrs $$$& 4 4 4D(((0&&&*7*7*7X [F ))TC5)))dVcU ; 99jjj !1&7 a&???#iii[ 2 3 00  0")9)9)9)9)9rXrc eZdZdZdZdZdS)TestConcatenatorctttdtgdtd}gd}t ||}t|dd|f}t t |tt|gdt|jt|dd|fdS)Nr>r;rrrrr)r>rrrrrr) r>r>r>r>r>rrr>r>r>r>r>) rrr rrr isinstancer r@)rRrBmdrZs rVtest_1dzTestConcatenator.test_1dws3/0%8J8J8J2K2KLLL GG OO  # # # 1a O 1k**+++1BBBCCC163q!Qz?33333rXctjdd}tjdd}tjtjddd}tjtjddd}t ||}t ||}t d||f}t |jdkt|ddddf|t|ddddf|t|j tj d||ft ||f}t |jdkt|ddddf|t|ddddf|t|j tj ||fdS)Nrrr1)rrk)rkr) rQrandomrandroundrrrrrr@r_)rRa_1a_2m_1m_2b_1b_2rs rVtest_2dzTestConcatenator.test_2dsinnQ""innQ""hry~~a++Q//hry~~a++Q//3S)))3S))) S# 7"###1QQQU8S)))1QQQU8S)))1625c3#7888 SM7"###1RaRT7C(((1QRRT7C(((1625c?33333rXcRttjjdf}t |jddgt |jddtddgtjjf}t |jgdt |jddddgdS)Nr>TFr;)FFT)rrQr_r rr@rs)rRrs rVtest_masked_constantz%TestConcatenator.test_masked_constantsRU\1_%V[4-000V[^Q'''aVRU\)*V["6"6"6777V[!_q!f-----rXN)rrrrrrr\rXrVrrtsA 4 4 4444(.....rXrceZdZdZdZdS) TestNotMaskedc Lttjdddgdgdgdgdgdg}t |d}t |d d gt |d }t |d dd gt |d d d gt |d }t |d d dgt |d d dgt |jd}t |d d gt |jd }t |d dd gt |d dd gt |jd}t |d d dgt |d d dgt|d<t |d }t |d dd gt |d dd gt |d}t |d ddgt |d ddgdS)Nr)rrr>rr)rrrr>r>)r>r>rrr)rrrrr)r>r>r>rrrrrr>r;rrr>)rrrrr)rrr;rr)rr;rrr)rrrrr)r>r>r;rr)rr>r;r)rrr;r)rr;rr)rrQr rr/rrsr rRrsrSs rV test_edgeszTestNotMasked.test_edgess;BIbMM11!Q77"1//"1//"1//"1//"1// "3555 tT**TAr7###tQ''T!W@AAAT!W@AAAtQ''T!W@AAAT!W@AAAty$//TAr7###ty!,,T!W@AAAT!W@AAAty"--T!W@AAAT!W@AAARtQ''T!W@AAAT!W@AAAtR((T!W|\:;;;T!W|\:;;;;;rXc|ttjdddgdgdgdg}t |d}t |t dd dt d d dt d ddgt |d}t |t dd dt dddgt dd dt dddgt dd dt dddgt dd dt dddgt dddgt dddggt dddggt |d }t |t dd dggt dddt dddggdS)Nrrr)rrrrr>r>r>r>)r>r>r>r>r>r>r>r>)rrrrrrr>rrrrr>r;rro)rrQr rr.rr)rRrAtmps rVtest_contiguouszTestNotMasked.test_contiguouss 2..q!447777777779 : : :#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      rXN)rrrr!r'r\rXrVrrs3 < < r>r>r)rr;rr)rk ))(*+,)2456)79:;rr)rrrorr)rkrr.r/r0)r1r#r2r3r4)r5)r6r7r8)-./01)r93r:r;r<)r=8r>r?r@r>)r$r%r)r !")#$%&'r[r)rr;)rrorr)rIr$r%r)rr)rKrLrMrN)rOrQrRrS)rBrDrErFr:r)rrr>)rrr>r;)rr;)r>r)rrrr;)rr) rQr listrprrastyperr(r)rRrrrAa2a3a4s rVtest_compress_ndz&TestCompressFunctions.test_compress_nds^ HT%,,'' ( ( 0 0Aq 9 9 HW   $ $T * *% !!    NNQ*********,+********,- . . . 1  Q............0/...........0 1 2 2 2 1  Q.........0/........0/........0 1 2 2 2D ! ! 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 $ $QrXcttjdddgdgdgdg}t t |ddgdd ggt t |d gd gd gt t |d d dgddgdd ggt|jgdgdgdg}t t |d dgdd ggt t |d gdgd gt t |d d dgddgdd ggt|jgdgdgdg}t t |d ggt t |d gd gt t |d dgdgd ggt|jgdgdgdg}t t |jd t t |d jd t t |d jd dS)Nrrrrrrrrorrr,)rrorr>r;rr>rrrr>r;rrr>)r rQr rrr)_datarrRrs rVtest_compress_rowcolsz+TestCompressFunctions.test_compress_rowcolsXsv ")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!44444rXcttjdddgdgdgdg}t t |jgdgdgdgt t |djgdgdgdgt t |djgdgdgdgt|jgdgd gdg}t t |jgd gdgd gt t |djgdgdgdgt t |djgd gd gd gt|jgdgd gdg}t t |jgdgdgd gt t |djgdgdgdgt t |djgd gd gd gt|jgdgd gd g}tt | tutt |d tutt |d tutt |j tt |dj tt |dj dS) Nrrrrrr-rr>r`)r>r>rrb) r rQr rrr*r@rcrallr rds rVtest_mask_rowcolsz'TestCompressFunctions.test_mask_rowcolsls ")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""'++--.....rXr)Nrr>func rowcols_axisrr>c6ttjdddgdgdgdg}t t 5|||}t |t||ddddS#1swxYwYdS)Nrrrrrr)r rQr rrDeprecationWarningrr*)rRrrirjrress rV#test_mask_row_cols_axis_deprecationz9TestCompressFunctions.test_mask_row_cols_axis_deprecations ")A,,&&q!,,! 999iii8 : : :, - - = =$qt$$$C l1l;; < < < = = = = = = = = = = = = = = = = = =s,BBBc xtjdd}gd}t||dd}t||dd}t ||d}t |jddgdd ggt ||d}t |jgd gd gd gt ||d }t |tj|d |d t ||d }t |tj|d |d gd }t||dd}t||dd}t ||d}t |jd dgddggt ||d}t |jgdgdgd gt ||d }t |tj|d |d t |t ||t ||d }t |tj|d |d gd}t||dd}t||dd}t ||}t |jtt ||}t |jtt|gddd}t|gddd}t ||d}t |jddgd d ggt ||d }t |tj|d |d t ||d}t |jgd gd gd gt ||d }t |tj|d |d t|gd dd}t|gddd}t ||d}t |jd d gddggt ||}t |tj|d |d t ||d}t |jgdgdgdgt ||d }t |tj|d |d t|gd dd}t|gddd}t ||d}t |jdd gddggt ||d }t |tj|d |d t ||d}t |jgdgd gdgt ||d }t |tj|d |d ttjdddddd gd d ggd d gd d ggg}ttjddddd d gd d ggd d gd dggg}t ||d}t |jddgddggd d gd dgggd d gd dggd d gd dggggt ||d }t |jd d gd dggd d gd d gggd d gd d ggd d gd d ggggt ||d}t |jdd gd d ggdd gd d gggdd gd d ggddgddggggt ||d }t |jd d gd d ggd d gd d gggd d gd d ggdd gd d ggggttjdddddd gd d ggd d gd d ggg}d}t ||d}t |jdd gd d ggd d gd d gggt ||d }t |jdd gd d ggd d gd d gggt ||d}t |jdd gd d ggd d gd d gggt ||d }t |jdd gd d ggd d gd d gggttjdddddd gd d ggd d gd d ggg}ttjdd dg}t ||d}t |jddgddggt ||d }t |jdd gd d ggdS)Nr>ro)r>rrrrrrr;rT)strictrr-rFrrrrrr>rb)rrrrrr)rrr>rrrrr) rQr rrrrr@filledr)rRnrrArBrZs rVtest_dotzTestCompressFunctions.test_dots 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&1AB D D D 1--aA66!"AA/1a&1a&1AB 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&1AB 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&1AB D D D 1QF 3 3 3 1T " " "QVq!fq!f-... 1U # # #QVq!fq!f-.....rXctjd}t|}tt t ||t utt t ||t utt t ||t utt t ||t udS)Nr)rQeyer rr`rr rRrArBs rVtest_dot_returns_maskedarrayz2TestCompressFunctions.test_dot_returns_maskedarrays F1II !HHSAYY;.///SAYY;.///SAYY;.///SAYY;./////rXcttjd}ttjd}t |||}t ||ut ||dS)Nr)rrout)r rQrvrrrr)rRrAr{rms rV test_dot_outz"TestCompressFunctions.test_dot_outsj "&))  BHV$$%%!QC   s QrXN)rrrr^rerhrrrrQr_ mask_rows mask_colsrnrtrxr|r\rXrVr)r)snnn`555(///@ [V\\\22 [fn5!u2RU_a4HIKK==KK32=m/m/m/^000rXr)ceZdZdZdZdS)TestApplyAlongAxisctdddd}d}t|d|}t|ddgddggdS) Nrr;rc|dSNr>r\)rBs rVmyfuncz*TestApplyAlongAxis.test_3d..myfuncs Q4KrXr>rrorkr rr'rrRrArxas rVtest_3dzTestApplyAlongAxis.test_3dsd 3KK  1a ( (   fa + +R1a&1b'*+++++rXctdddd}d d}t|d|d}t|ddgd d ggdS) Nr.r;rrc|d|zSrr\)rBoffsets rVrz1TestApplyAlongAxis.test_3d_kwargs..myfunc'sQvX; rXr>)rrrr)rrrs rVtest_3d_kwargsz!TestApplyAlongAxis.test_3d_kwargs$sn 2JJ  q!Q ' '    fa1 5 5 5R1a&1b'*+++++rXN)rrrrrr\rXrVrrs2,,,,,,,,rXrceZdZdZdS)TestApplyOverAxesctdddd}ttj|ddg}tjdgdgdggg}t ||t||dzt<ttj|ddg}tjd gd gdggg}t ||dS) Nrr;rrrr+\|rr8) r rr&rQrtr rr rZr)rRrArSctrls rV test_basiczTestApplyOverAxes.test_basic0s 2JJ  q!Q ' 'rvq1a&11x2$se,-..T4   "(1q5..  rvq1a&11x2$rd+,--T4     rXN)rrrrr\rXrVrr.s#!!!!!rXrceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZejdgddZdZdZdZdZdZdZdZdZdS) TestMedianctjtjtjgtjtjggd}t |tjdS)Nrr)rQr_rinfrrRrs rV test_pytypezTestMedian.test_pytype<sH ELL2626*RVRV,<=BL G GQrXc*tjtjtjtjgtjtjggd}t |tjtjtjtjtjgtjtjggd}t |tjtjtjtjtjgtjtjggdd}t |jdtjtjtjtjgtjtjggdd}t |jddS)NrrTr)rQr_rrrrr@rs rVtest_infzTestMedian.test_inf@s ELL++bfbf-=.0fbf-=-?@@FH  J JQ ELL++bfbf-=.0fbf-=-?@@FJ  L LQ ELL++bfbf-=.0fbf-=-?EI,KK   " " QVT""" ELL++bfbf-=.0fbf-=-?EI,KK"  $ $ QVT"""""rXcjtjd}ttj|dt t tj|tutd}ttj|dt t tj|tud}ttj|dt t tj|tutjd dd}ttj|d tj|d ttj|d tj|d t tj|d tutjd  dd}ttj|d tj|d ttj|d tj|d t tj|d tudS) NrrrrrrHrrr>gR@) rQr rr_rrr`r rprrds rVtest_non_maskedzTestMedian.test_non_maskedRs: 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 ''{:;;;;;rXc@ttjddgdzdgdzz}ttj|dttj|jddtttj|tuttjd  d d dgd zdgdzz}ttj|d ttj|jddtttj|tutj|dd}t|ddgt|jddtt|tudS)z5test the examples given in the docstring of ma.medianrrrr>rrr\shape mismatchrkr;rrrrTroverwrite_inputrrr:N) r rQr rr_rrrr`r r)rRrma_xs rVtest_docstring_examplesz"TestMedian.test_docstring_examplesgs ")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)*****rXc .d}tdD]}dD]}td||}tjt| |ddg}|D]L\}} tj|||)#t$rt|||||fzwxYwtj|dz |gddg}|D]R\}} tj|||t|||||fz#tj $rYOwxYwdS) Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr)FTr>)ndminr@FTr) rpr itertoolsproductrQr_r ExceptionAssertionError AxisError)rRmsgrr@rargsrovers rVtest_axis_argument_errorsz$TestMedian.test_axis_argument_errorsvsE1XX N NE% N N!5t444!(vu)=)=t}MM"&NNJD$N QT4 HHHH$NNN,SD%t3L-LMMMN!(EAI,)> NN"&NNJD$N QT4 HHH-SD%t3L-LMMM<N N N Ns"A<FrT)r rrQr_rr rds rVtest_masked_0dzTestMedian.test_masked_0dsh !% RU\\!__a((( !$   RU\\!__bel33333rXc ttjdd}ttj|tjjttj|jddtttj|tjj j uttjdd}ttj|dttj|jddtttj|tuttjdgd}ttj|d ttj|jddtttj|tuttjdgd }ttj|d ttj|jddtttj|tuttjdgd }ttj|d ttj|jddtttj|tuttjdgd }ttj|d ttj|jddtttj|tuttjdgd}ttj|d ttj|jddtttj|tuttjdgd}ttj|d ttj|jddtttj|tudS)NrTrr\rFrrr)rr>r>r>r>r)rr>r>rrrrrrr) r rQr rr_rr rrr`raMaskedConstantr rds rVtest_masked_1dzTestMedian.test_masked_1ds# ")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\\!__%%[899999rXc ttjt gdgdjtjt gdgdjdS)Nrrrr`)rrQr_rr rrRs rVtest_1d_shape_consistencyz$TestMedian.test_1d_shape_consistencystRU\\%WWW"="="=>>DU\\%WWW"="="=>>D G G G G GrXc d\}}ttjdd|}tx|dd<|dd<ttj||ft }|dd|dddf<tjt|}td|D]0}tj ||||dd|f<1tt|dddfdtt|dtt|d tj |tt|jd tj |dS) N)erKrrkr<rr>r)rrQlinspacer emptyrPr rbrpr shufflerrrT)rRrsprridxrvs rVrzTestMedian.test_2ds\A S"a00 2 2!!#2#344 1a&666 7 7AAA$!!!Q$iAq!  A I  c " " "fAaaadGGVAaaadG__a(((VAYY"""VAA&&& 444VACa((("(1++66666rXc(ttjddd}tx|dd<|dd<t t |dtttj |tut t |dgdtttj |dtut t |d gd tttj |d tut t |d j gd dS) NrKrkr-@rr)+@rg/@r>) rrrrkr/r#r4rrr) r>r>r>rrrrr>r>r>) rrQr rr rrrr`r_r r@rds rV test_2d_waxiszTestMedian.test_2d_waxissM 2..r155 6 6"1""##VAYY%%%RU\\!__%%[8999VAA&&&(:(:(:;;;RU\\!!\,,--<===VAA&&&(J(J(JKKKRU\\!!\,,--<===VAA&&&+-K-K-KLLLLLrXctjdddd}t||dzdk<t t |dddgdd gddgd d ggd |_t t |dd d gdd gddggtjdddd}t||dzdk<t t |ddd gddgddggdS)Nrrrr;rr.rrr1r3)rrr;crkrr/r0rrr#r2)rQr_r rr rrrrds rVrzTestMedian.test_3ds ELL   $ $Q1 - -!a%1* VAq\\RGaWr1gBx#HIIIVAq\\RHr2hR#ABBB ELL   $ $Q1 - -!a%1* VAq\\RHq!fr2h#?@@@@@rXcttjddd}tx|dd<|dd<t t |dt |ddS)NrKrkrrrrr>)rrQr rr rrrds rV test_neg_axiszTestMedian.test_neg_axissq 2..r155 6 6"1""##VAB'''):):):;;;;;rXcdD]}ttj|}tx|dd<|dd<ttjd}t ||}|dkrt |dnt |dt||utt|tudS) N)rK>@rg?@rrr\rzrKrg.@) rrQr r rrrrr`r )rRvrr{rs rV test_out_1dzTestMedian.test_out_1ds# , ,ARYq\\**A# #AbqbEAbccFrwr{{++Cqc"""ABwwS$''''S#&&& AH    DGG{* + + + + , ,rXcldD]/}ttj|dd}tx|dd<|dd<ttjd}t |d|}|dkr4td gdzgd zd gdzzd gdzd gd zzd gdzz}n3td gdzgdzd gdzzd gdzd gd zzd gdzz}t||t||utt|tu1dS)N)r5gD@rKrrkrrrr>)rr{rKr)rkr/r#r4TFrr)rg1@g5@g9@) rrQr rr rrrrr`r )rRrrr{res rVtest_outzTestMedian.test_outsx# , ,ARYq\\11"b99::A# #AbqbEAbccFrwr{{++Cqqc***ABww "a*:*:*:!:bT!V!C'+fqjE7Q;&>$!&KMMM!"a*B*B*B!BbT!V!K'+fQh%&:dVAX&EGGG A    AH    DGG{* + + + + , ,rXr)Nr>r[rV)rr)argnames argvaluesctjdt}tjdtj|jdddfz}|tj}tj|t|<ttj |j|| dj z}nCt|j tfdtj D}ttj|}t!|d|}||usJt#|j|dS) N)rrrorr<)rrr[c3>K|]}|vrdn j|VdS)r>N)r).0rv axis_normrs rV z/TestMedian.test_keepdims_out..sNLL89Q)^^LLLLLLrXT)rrr{)rQrrr r rrZintprtuplerrndimrrprrr) rRrr@r shape_outr{rrrs @@rVtest_keepdims_outzTestMedian.test_keepdims_out sPx T222 I  X & &$*)=)=aaag)F F HHRW  U1XX ,,4 8 8 8 <qv II,T16::ILLLLL=B16]]LLLLLI28I..//t===}}}}V\9-----rXcddgddgddgg}tj|d}ddg}ttj|d|dS)Nrrrrr)rQr_ masked_equalrr)rRrsrfrs rV$test_single_non_masked_value_on_axisz/TestMedian.test_single_non_masked_value_on_axis&ssRRRU''a00 825<< <;;# % % % % %rXcxdtjdtfD]"}tjdtjdggdg}||_tj|d}ttj |t|tjtj| d}ttj |t|tjtj|d}tt|tt|dtjdgtj|d}tt|tt|tjd gtj|d }tt|tt|tjd g$tjdtjdggdg}tjj|ddd f<ttj|dtjttj|ddtjdgttj|dtjd gdS) NFrr<r>rrrrr;rr)rQrrr_r rr@rrisscalarrravelrr`r r )rRr@dmrs rVtest_nanzTestMedian.test_nan/s[BHQd3334 / /Dq"&!niii899BBG 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]CCCCCrXctjtjd}tjtjd}tj|d<tjj|d<ttj|d||tjtjd}ttj|d||tjtjd }ttj|||dS) N)rrrr;r>rIrrzrr>r\) rQr_rrrrr rr)rRors rV test_out_nanzTestMedian.test_out_nanLs E  rx~~ . . E  rwv / /&$%,$RU\\!QA\..222 E  rx~~ . .RU\\!QA\..222 E  rx|| , ,RU\\!\++Q/////rXcbtjtjdt}tjj|ddd<tj|d<ttj|tjttj|dtjtjtjdt ddd}tj|j dzdk|_ | }tj|d <tj|d <ttj|tjttjtj|tj|d}tj|d <tj|d <ttj|d|tj|d}tj|d <tj|d <ttj|d|tj|d}tj|d<tj|d<ttj|d|dS)Nrr<rr;rrrr>r)r>r>r;r;rrWr>rrX)rQr_rr rPr rrrrrr@copyrrr)rRrAaorigrBs rVtest_nan_behaviorzTestMedian.test_nan_behaviorWs3 E  ry5999 : :##A#v!25<<??BF33325<<<22BF;;; E  ry5999AA!QJJ K K16""Q&!+V' V'  25<<??BF333 BELLOO,,--- ELLQL ' '&$&$RU\\!Q''+++ ELLQL ' '&$&$RU\\!Q''+++ ELLVL , ,v!v!RU\\!V,,a00000rXc&tjgdgdgtj}tj||dk}t tj|ddt tj|djdt tj|d |d t tj|ddS) N)rrr<rrr>rrFr)rQr uint8r_rrrr@rs rVtest_ambigous_fillzTestMedian.test_ambigous_fillzs Hkkk;;;/rx @ @ @ E  qqAv  . .25<<<22C88825<<<227???25<<<22AaD99925<<??C00000rXc  tjtj fD]}tj|tjgtjtjgg}tj|tj|}ttj|d|tjgttj|d|tjgttj||tjtjtj|gtjtj|gg}tj|tj|}ttj|d|ttj|dj dttj|d|dttj||tj||g||gg}ttj||ttj|d|ttj|d|tj|d| dgdtjtjd gd tjtj|ggtj }tj|tj|}|dkrbttj|dd d | dgttj|dnattj|ddd | dgttj|dttj|ddd|gtddD]}tddD]}tjtjg|z|g|zzgdz}tj|tj|}ttj||ttj|d|ttj|dtjg|z|g|zzdS)Nrrrr>Froirrrr<rrrg@g$g"grrkr;) rQrr rr_risnanrrrr@float32rp)rRrrArvrws rV test_specialzTestMedian.test_specialshFRVG$( =( =C3.2626*:;<!?!!CDDA**128A;;*??A a#666 aa!8!8#>>> aa!8!8#%6(Q,3%!)!;==== = =C( =( =rXcxtjtjgt}t 5}|tttj |tj t|j dj tudddn #1swxYwYtjtjgtd}t 5}|ttjddtttj |tj t|j dj tudddn #1swxYwYtjtjgtd}t!tj |d|t!tj |d |tjtjtj td}tjd 5}tjddtt!tj |d|t|dj tuddddS#1swxYwYdS) Nr<rr)r=ralwaysr;rr>T)record)rQr_rr rPrrRuntimeWarningrrrrlogcategorywarningsfilterwarningsrcatch_warnings)rRrArrBs rV test_emptyzTestMedian.test_emptys E  rx%888 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%qAAA 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%qAAA B BRU\\!!\,,a000RU\\!!\,,a000 E  rxe1EEE F F  $D 1 1 1 5Q  #Hb. A A A aa00! 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 5s9A8C  CCBF<<GG6A,L//L36L3c tjtjd}t t tj|tttj |d<t t tj|ttdS)Nrr;) rQr_rr rr`rrZr^rPr)rRrs rV test_objectzTestMedian.test_objects E  ry}} - -RU\\!((6"2"23344e<<<v!RU\\!((6"2"23344e<<<<>>%d;;; = = = = =rXcj|jdd}ttj|t |ttj|dt |dttj|ddt |dddS)NrrFrTr)rsrr rQrrds rVtest_2d_without_missingzTestCov.test_2d_without_missings I  a # #BF1IIs1vv...BF1U333S55I5I5IJJJBF1U>>>%d;;; = = = = =rXc |j}t|d<||z}|}t t j|t |t t j|dt |dt t j|ddt |dd t |dn#t$rYnwxYw|dd}t t j||dddt ||dddt t j||ddddt ||ddddt t j||dddddt ||ddddddS)NrFrTr allow_maskedr>)rsr meanrur rQr ValueError)rRrnxs rVtest_1d_with_missingzTestCov.test_1d_with_missings I" QVVXX  \\^^BF2JJA///BF2e444c!E6J6J6JKKKBF2e$???%d;;; = = =   & & & & &    D qtWBF2r$$B$x00#a44R4//BBBBF2r$$B$x>>>1TTrT75999 ; ; ;BF2r$$B$xDIII1TTrT75tDDD F F F F FsC'' C43C4c"|j}t|d<|dd}tjt |t}tj||j }|| ddddfz d}tt|tj ||jddz z|dz z tt|dtj |d|jdz|z tj|j |}|| dz d}tt|d tj |d |jddz z|dz z tt|d d tj |d d |jdz|z dS) Nrrrr>rrTrFrr)rsr rrQ logical_notrrZrqrrrrrr rr)rRrvalidfracxfs rVtest_2d_with_missingzTestCov.test_2d_with_missings I" IIaOO|A//66s;;veUW%%!&&))AAAtG$$ , ,Q / /CFFF2JJ!'!*q.9TBYG I I IC---F2D111AGAJ>E G G Gvegu%%!&&))m # #A & &C%000VBu555gaj1n.15< > > > C%d;;;VBu4@@@WQZ(*./ 1 1 1 1 1rXN)rrrr rrrr#r\rXrVr r sb...======FFF011111rXr c8eZdZdZdZdZdZdZdZdZ dS) TestCorrcoefcttjd|_ttjd|_dSr )r rQr r rsdata2rs rVr zTestCorrcoef.setup_methods@")..,,-- 29>>"--.. rXc|j|j}}tj|}tj||}t 5}t jdttt|d| tdttj|dt|dtt|d|tt||d|tt|d|tt||d|ddddS#1swxYwYdS)Nrrddofbias and ddof have no effectrr rsr'rQrrr simplefilterrrlfilterr )rRrrr expected2sups rV test_ddofzTestCorrcoef.test_ddofsy$*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 CsC>EEEc |j|j}}tj|}t 5}t jdttt||ddttt||ddttt|d| tdtt|d|ddddS#1swxYwYdS)NrTFrr+r>r,)rRrrrr0s rV test_biaszTestCorrcoef.test_bias+sy$*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 > > > ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sB(C&&C*-C*c |j}ttj|t|ttj|dt|dt 5}|t dttj|ddt|ddddddS#1swxYwYdS)NFrr+Tr)rsr rQrrr.rlrRrr0s rVrz$TestCorrcoef.test_1d_without_missing8s* IBKNNHQKK888BK%888$Qu555 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 Fs8ACCCc |jdd}ttj|t |ttj|dt |dt 5}|tdttj|ddt |ddddddS#1swxYwYdS)NrrFrr+Tr)rsrr rQrrr.rlr5s rVrz$TestCorrcoef.test_2d_without_missingCs: I  a # #BKNNHQKK888BK%888$Qu555 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|dt |dt5}|tdt t j|ddt |dddddn #1swxYwY t |dn#t$rYnwxYw|dd}t t j||dddt ||dddt t j||ddddt ||ddddt5}|tdt t j||dddt ||dddd t t j||dddt ||dddd ddddS#1swxYwYdS) NrFrr+Trrr>rr;r)) rsr rrur rQrrr.rlr)rRrrr0s rVrz!TestCorrcoef.test_1d_with_missingNs  I" QVVXX  \\^^BKOOXa[[999BK5999$Qu555 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 qtWBKBtttH55x1TTrT77K7KLLLBKBtttHUCCC$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 |ddddf|ddddft 5}|tdt t|dddddf|ddddft t|dddddf|ddddft t|dddddf|ddddfddddS#1swxYwYdS) Nrrrr+rr)r>r) rsr rrrQr rr.rl)rRrrSrTr0s rVr#z!TestCorrcoef.test_2d_with_missingls 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 3sCEEEN) rrrr r1r3rrrr#r\rXrVr%r%s///CCC ? ? ? F F F F F F>>><33333rXr%ceZdZdZdZdS)TestPolynomialcjtjd}tjddd}t t ||dtj||d|t}t|d<|t}tx|d<|d<t ||dddfdd \}}}}}tj|d d|d ddf dd \}} } } } t|||||f|| | | | fD]\} }t | |t ||dddfdd \}}}}}tj|d d|d ddfdd \}} } } } t|||||f|| | | | fD]\} }t | |t ||dd \}}}}}tj|d d|d dddfdd \}} } } } t|||||f|| | | | fD]\} }t | |tjdd z}| }|d d}|d d}|d d}t ||dd | \}}}}}tj||dd | \}} } } } t||t|||||f|| | | | fD]\} }t | |dS) NrkrIrr;rrrGrrTrrr>)rrr) rQr r rr rviewr r ruziprr)rRrrCRKSDrZrrxrzrrAa_rwoxsyswss rV test_polyfitzTestPolynomial.test_polyfits INN2   INN2   & &r1 - -GAq!,,bjAq.A.ABBB FF;  ! FF;  $$$!F)!!Qqqq!tWad;;;Aq!Q*QqrrUAabb!eH,?,?,A,A1*.000Aq!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 & & & & ' 'rXcltjd}tjddd}tj|d<tj|d<|t }|t }t|d<t|d<t||dd \}}}}}tj|d d|d dddfdd \}} } } } t|||||f|| | | | fD]\} }t| |dS) NrkrIrr;rr<rTr=r>) rQr r rrr>r r rr?r )rRrrr@rArBrCrDrZrrxrzrrArEs rVtest_polyfit_with_masked_NaNsz,TestPolynomial.test_polyfit_with_masked_NaNss* INN2   INN2   & &r1 - -v!6% FF;   FF;  !%!!Q555Aq!Q*QqtWa"QQQiFFFAq!QAq!Q?Q1aO<< ' 'GQ 2 & & & & ' 'rXN)rrrrJrLr\rXrVr:r:s3%'%'%'N'''''rXr:cneZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)TestArraySetOpsc4gd}t|dd}tt|dtt |dt gdgdt |dgd t |d gd dS) Nr>r>r>r;r;rT return_indexreturn_inverserrrrr>)rrrr;)rrrr>r>r;)rrrr rrr s rVtest_unique_onlistz"TestArraySetOps.test_unique_onlists!!!ddCCC 47K00111T!Wl999999EEEFFFT!Wiii(((T!W00011111rXctgdgd}t|dd}t|dtgdgdt|d gd t|d gd d |_tgdgdd }t|dd}t|dtgdgdt|d gd t|d gd dS)NrP)rrr>rr>rrTrQr)r>r;rrrrrr>r>)rrrr;r;)rrrr>rr;r)rsr@ fill_value)rrrrWr s rVtest_unique_onmaskedarrayz)TestArraySetOps.test_unique_onmaskedarrays:...5G5G5GHHHddCCCT!Wl===|||LLLMMMT!Wlll+++T!W000111!3!3!3!3!3!3CCCddCCCT!Wl===|||LLLMMMT!Wlll+++T!W00011111rXctgdd}t|dd}t|dtdgdgt|ddgt|dgdt}t|dd}t|dttt|ddgt|ddgdS) Nr-TrrQrr>r;r)rrrr r s rVtest_unique_allmaskedz%TestArraySetOps.test_unique_allmaskedsIIID111ddCCCT!WlA5v>>>???T!Wqc"""T!Wiii(((ddCCCT!Wl622333T!Wqc"""T!Wqc"""""rXcXttjdgd}tgdgd}t |}t ||t |d|dt |j|jdS)Nrr>rrrr>r)r>r>r>r)r>rrr>r)rrQr r r%rrrr@)rRrrTrSs rV test_ediff1dzTestArraySetOps.test_ediff1ds 1OOO < < < <<<888qzzT7###T[[^^W^^A%6%6777TY -----rXcvttjdgd}t|t}t gdgd}t ||t |d|dt |j|jt|gd}t gd gd }t ||t |d|dt |j|jdS) Nrr\r)to_begin)rr>r>r>r)r>r>rrr>rr)r>r;rr>r>r>r)rrrr>rrr> rrQr r%r r rrrr@rRrrSrTs rVtest_ediff1d_tobeginz$TestArraySetOps.test_ediff1d_tobegins 1OOO < < <q6***ooo>>>T7###T[[^^W^^A%6%6777TY ---q999------4I4I4IJJJT7###T[[^^W^^A%6%6777TY -----rXcvttjdgd}t|t}t gdgd}t ||t |d|dt |j|jt|gd}t gd gd }t ||t |d|dt |j|jdS) Nrr\r)to_end)r>r>r>rr)r>rrr>r>rr)r>r>r>rr>r;r)r>rrr>rrrr`ras rVtest_ediff1d_toendz"TestArraySetOps.test_ediff1d_toends 1OOO < < <q(((ooo>>>T7###T[[^^W^^A%6%6777TY ---q+++---4I4I4IJJJT7###T[[^^W^^A%6%6777TY -----rXcttjdgd}t|tt}t gdgd}t ||t |d|dt |j|jt|gdt}t gd gd }t ||t |d|dt |j|jdS) Nrr\rrdr_)rr>r>r>rr)r>r>rrr>r>rr)rr>r>r>rr>r;r)r>r>rrr>rrrr`ras rVtest_ediff1d_tobegin_toendz*TestArraySetOps.test_ediff1d_tobegin_toend s, 1OOO < < <q&999***1C1C1CDDDT7###T[[^^W^^A%6%6777TY ---qV<<<000555777T7###T[[^^W^^A%6%6777TY -----rXctjd}t|}tgdgd}t ||t t |tt |d|dt |j |j t|tt}tgdgd}t t |tt |d|dt |j |j dS) Nrr>r>r>r>rrrrrrrgrr) rQr r%r rrrr rrr@r ras rVtest_ediff1d_ndarrayz$TestArraySetOps.test_ediff1d_ndarrays" IaLLqzz <<<888T7### 4--...T[[^^W^^A%6%6777TY ---q&999***1C1C1CDDD 4--...T[[^^W^^A%6%6777TY -----rXctgdgd}tgdgd}t||}tgdgd}t||dS)N)r>rrrrVr)rr>r>r>)r>rrrb)r r#r)rRrrrSrTs rVtest_intersect1dz TestArraySetOps.test_intersect1d+su ,,,\\\ 2 2 2 ,,,\\\ 2 2 21a   333T7#####rXc4tgdgd}tgdgd}t||}t|tgdtgdgd}gd}t||}t|tgdgd tgd }tgd }t||}tt |t t|gd tgd gd}tgdgd }t||}tt |t t|gd t gtggdS)Nr>r;rrorrrrrr>rr>r;rrrrrq)rrro)r>r;rrr)rrrorrVr)rrrr)r>rr;r)rr>rr)rrrr)r r rrrr rrRrArBrSs rV test_setxor1dzTestArraySetOps.test_setxor1d3s """ 9 9 9 %%%,>,>,> ? ? ?1~~T5++,,, """ 9 9 9 OO1~~T5\\\BBBCCC )))   )))  1~~ 4--...T---... ,,,\\\ 2 2 2 ,,,\\\ 2 2 21~~ 4--...T---...2xB//00000rXctjdgd}tjgd}d|d<t ||}t gdgd}tdt }d |d <d |d <d |d <t ||}tt|tt||tj|||j |j z}t||dS)Nrrr>)r>r;rr) rrkrIrKr>rrr$rM) rr>rr>rr>rr>r)rsr@r<Trrb)rr;r) rQr rrr rr2rrr rr@)rRrAr@rBecrZrs rV test_isinzTestArraySetOps.test_isinMs IbMM ! !))) , ,x ""W !$    :::::: < < < 9D ) ) )7 7 7 AJJ 1k**+++1b!!! GAq!&z " "afW ,1a     rXcrtgdgd}tgdgd}t||}t|gdtgdgd}tgdgd }t||}t|gd tgtggdS) Nrprqrrrrq)TTTFTrrr;r>rr>rrrb)TTFTT)r r$rrrss rV test_in1dzTestArraySetOps.test_in1dbs """ 9 9 9 %%%,>,>,> ? ? ?AqzzT:::;;; """ 9 9 9 ***999 - - -AqzzT:::;;;2tB||,,,,,rXctgdgd}tgdgd}ttjt ||t ||dtgdgd}tgd gd }ttjt ||t ||dt gt ggddS) NrprqrrrrqT)invertryrzrb)r rrQr}r$rrws rVtest_in1d_invertz TestArraySetOps.test_in1d_invertps """ 9 9 9 %%%,>,>,> ? ? ?RYtAqzz**DAd,C,C,CDDD """ 9 9 9 ***999 - - -RYtAqzz**DAd,C,C,CDDD2tB488899999rXctgdgd}tgdgd}t||}tgdgd}t||tgdgdggd gd g}tgd gd }tgd gd}t||}t||tgtggdS)N)r>r;rrorrrqrrr)r>r;rrrror)rrrrrrr>rar,rrbrrq)rr>r;rrr)r r"rr) rRrArBrSrTrrezrs rV test_union1dzTestArraySetOps.test_union1d|s" %%%,>,>,> ? ? ? %%%,>,>,> ? ? ?q!}}...5J5J5JKKKT7### 999iii( 999/E F F F /// 8 8 8 %%%,>,>,> ? ? ? AqMMQ2wr2/////rXctgdgd}tgd}t||}t|tgdgdtd}td}tt||tdd gtgtjg}tt|gjtjdS) N)rrrroror>r;r>)rrrrrrrr>r)r;rrrr;r>r)rrorrbrkrr)r r!rr rQuint32r=rss rVtest_setdiff1dzTestArraySetOps.test_setdiff1ds ***1I1I1I J J J ''' ( (AT5)))<<<=== 2JJ 1IIYq!__eQFmm444 "bib ) ) )Yq"%%+RY77777rXctjgd}tjgd}tt||tjdgdS)N)rArBrZ)rArBrzrZ)rQr rr!rws rVtest_setdiff1d_char_arrayz)TestArraySetOps.test_setdiff1d_char_arraysP H___ % % H___ % %9Q??BHcUOO<<<<rr;r<rFru) rQr r zeros_likerrrrYr4)rRordinary empty_mask with_masks rVtest_ndenumerate_nomaskedz)TestNDEnumerate.test_ndenumerate_nomaskeds9R==((33]84888   ;;; T".2233+h//00 2 2 2T+h//00+i0011 3 3 3T+i0011+iEBBBCC E E E E ErXc td}td}td}ttt|gttt|gttt|dtt t jddtgzttt|gttt|dtt t jddtgzdS)Nr\)drFrrr)r0rrYr4r?rQndindexr )rRrArBrZs rVtest_ndenumerate_allmaskedz*TestNDEnumerate.test_ndenumerate_allmaskeds0 rNN v   y ! !T+a..))2...T+a..))2...T+aE:::;;#bj00#.AABB D D DT+a..))2...T+aE:::;;#bj33Y&5IJJKK M M M M MrXc ttjddgdgdgdg}gd}t t t ||t tt t |d |jt |d D]\}}t |||dS) Nr.rrj)r>r>rr>rkr))rWr))r;rr)rr)rIrk)rrFr) rrQr rrrYr4rbr)rRrAitems coordinatevalues rVtest_ndenumerate_mixedmaskedz,TestNDEnumerate.test_ndenumerate_mixedmaskeds 2..v66+||+||+||- . . .GGGT+a..))5111Sk!>>>??@@!&III!,Q5!A!A!A / / J : . . . . / /rXN)rrrrrrr\rXrVrrsG E E E M M M / / / / /rXrc eZdZdZdZdZdS) TestStackctgdgd}tgdgd}t||gd}t|jdt |j|djt |j|d jt ||g}t |j|jt |j|jt||gd }t|jd t |j|dddfjt |j|ddd fjdS) Nrar`rrrrorrrrr>rr;rr5rrrr@r6rsrRrArBrZrs rV test_stack_1dzTestStack.test_stack_1ds3  3 3 3  3 3 3 1a&q ! ! !QWf%%%161Q49---161Q49--- Aq6NN1616***1616*** 1a&q ! ! !QWf%%%161QQQT7<000161QQQT7<00000rXctgdd}tgdd}t||gd}t|jdt |j|djt |j|d jt ||g}t |j|jt |j|jt||gd }t|jd t |j|dddfjt |j|ddd fjdS) NraTrrFrrrr>rrrs rVtest_stack_maskszTestStack.test_stack_maskss+  . . .  / / / 1a&q ! ! !QWf%%%161Q49---161Q49--- Aq6NN1616***1616*** 1a&q ! ! !QWf%%%161QQQT7<000161QQQT7<00000rXcd}tjdd|}tjdd|}tjdd|t}tjdd|t}t ||}t ||}t ||gd}d|z} t|j| t|j |dj t|j |dj t ||gd }|dz} t|j| t|j |d j t|j |d j d }tjdd|}tjdd|}tjdd|t}tjdd|t}t ||}t ||}t ||gd}d|z} t|j| t|j |dj t|j |dj t ||gd }|dz} t|j| t|j |d j t|j |d j dS) Nrrrkr;rrr:r>r).r).r>)rr;rr) rQr randintrZrrr5rrrr@) rRshpd1d2rra1r[rZc_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$$$27AfIN33327AfIN33333rXN)rrrrrrr\rXrVrrsA111$111$)4)4)4)4)4rXr)N__doc__rrrnumpyrQnumpy.core.numericr numpy.testingrrnumpy.ma.testutilsrrrr numpy.ma.corer r r r rrrrrrrnumpy.ma.extrasrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r8rrrr)rrrr r%r:rNrrrr\rXrVrsS 333333DK9K9K9K9K9K9K9K9\).).).).).).).).XC C C C C C C C Looooooood ,,,,,,,,, ! ! ! ! ! ! ! !T=T=T=T=T=T=T=T=n @1@1@1@1@1@1@1@1Fh3h3h3h3h3h3h3h3V7'7'7'7'7'7'7'7'te=e=e=e=e=e=e=e=P4141414141414141n#/#/#/#/#/#/#/#/LO4O4O4O4O4O4O4O4O4O4rX