bgddlmZddlZddlZddlmcmZddlmcm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm 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>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYddlZm[Z[ej\Z\ddZ]Gdd Z^Gd d Z_Gd d Z`dZadS))reduceN)assert_ assert_raises assert_equal)KMaskType MaskedArrayabsoluteaddallallcloseallequalalltruearangearccosarcsinarctanarctan2arrayaveragechoose concatenate conjugatecoscoshcountdivideequalexpfilledgetmaskgreater greater_equalinner isMaskedArrayless less_equalloglog10 make_maskmasked masked_array masked_equalmasked_greatermasked_greater_equal masked_inside masked_lessmasked_less_equalmasked_not_equalmasked_outsidemasked_print_option masked_values masked_wheremaximumminimummultiplynomasknonzero not_equalonesouterproductputravelrepeatresizeshapesinsinhsometruesortsqrtsubtractsumtaketantanh transposewherezeros)picklecZt||}|std|d|d||S)NzNot eq: z ----)r print)vwmsgresults m/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/ma/tests/test_old_ma.pyeqr\sC a^^F - +++q++++,,, MceZdZdZdZejdddgdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"S)#TestMactjddddtdz dddddddg }tjgd }d}gd }gd }t|| }t|| }tjgd }t|gd } tj|d|} |j} |d||||||||| | | f |_dS)N?g@@@$$@@) rdrgrbgrhrerfrarhrg) rrrrrrjrrrrr) rrrjrrrjrjrrrrrjmask)grh?g?rrjrr@xD)nprpirPrDset_fill_valued selfxya10m1m2xmymzzmxfss r[ setup_methodzTestMa.setup_method&s Hb"b#r#vr2tS"b"M N N HJJJ K K 1 1 1 1 1 1 12    12    H&&& ' ' 1<<< ( ( ( Xb% # # G %   QRRQB:r]c |j\ }}}}}}}}} } } tt| tt|tt || t|j| t|j|jt|jtd| tt|t|td|z tt|| ttt|d| tt||dS)Nc ||zSNrvrws r[z)TestMa.test_testBasic1d..=s Qr]c ||zSrrrs r[rz)TestMa.test_testBasic1d..>s QUr]ro) rsrr$rrDdtypesizerrlenr\rrts r[test_testBasic1dzTestMa.test_testBasic1d5s.48F1AsBBAr2qM!$$$%%% b!!"""U2YY"""RXq!!!RXqw'''RWf%6%6::;;;U2YYB&1B1BB*G*G GHHH2r 6"e$$b))***1b r]r)c |j\ }}}}}}}} } } }||_||_||_||_|| _tt| tt|t t||t |j|t |jt d|t t|t|t d|z tt|| ttt|d| tt||dS)Nc ||zSrrrs r[rz)TestMa.test_testBasic2d..Qs !a%r]c ||zSrrrs r[rz)TestMa.test_testBasic2d..Rs a!er]ro) rsrDrr$rrrrrr\r) rurrvrwrxryrzr{r|r}r~rs r[test_testBasic2dzTestMa.test_testBasic2dCs?59F1AsBBAr2qM!$$$%%% b!!"""U2YY"""RXq!!!RWf%7%7;;<<<U2YYB&1C1CR*H*H HIII2r 6"e$$b))***1b r]c  |j\ }}}}}}}}} } } tddgddgg} t| ddgddgg} tt | | z| | ztt | | z| | ztt | | z | | z dD]4} || }|| }|| }|| }| | } tt | | tt ||z||ztt ||z ||z tt ||z||zt jdd5tt ||z ||z dddn #1swxYwYtt ||z||ztt ||z ||z tt ||z||zt jdd5tt ||z ||z dddn #1swxYwYtt ||z||ztt ||z ||z tt ||z||ztt ||z ||z tt |dz|dztt t|dzt|dztt ||z||ztt t j ||t||tt t j ||t||tt t j ||t||t jdd5tt t j ||t||dddn #1swxYwY6dS) Nrjrrr)) r)rrignorerinvalid@) rsrr+rr\reshaperperrstateabsr rJr9r)rurvrwrxryrzr{r|r}r~rra2da2dms r[test_testArithmeticzTestMa.test_testArithmeticWs48F1AsBBAr2qaVaV$%%C1a&1a&!12239cDj))***39cDj))***39cDj))***( = =A ! A ! AABABAB BrB3KK Bq1ub2g&& ' ' ' Bq1ub2g&& ' ' ' Bq1ub2g&& ' ' 'Hh??? , ,1q5"r'**+++ , , , , , , , , , , , , , , , BsQwb)) * * * BsQwb)) * * * BsQwb)) * * *Hh??? / /37C"H--... / / / / / / / / / / / / / / / Bq3wS)) * * * Bq3wS)) * * * Bq3wS)) * * * Bq3wS)) * * * BqAvrQw'' ( ( ( Bs1vv}c"ggn55 6 6 6 BqAvrRx(( ) ) ) Brva||SR[[11 2 2 2 Br{1a(((2r*:*:;; < < < Br{1a(((2r*:*:;; < < <Hh??? = =29Q??F2rNN;;<<< = = = = = = = = = = = = = = =7 = =s64$G$$G( +G( .$JJ" %J" ?SS S ctjdg}tdg}tt||zttt||ztdS)Nrj)rprr isinstancer)runamas r[test_testMixedArithmeticzTestMa.test_testMixedArithmetic}s\ Xqc]] A3ZZ 27K00111 27K0011111r]c F |j\ }}}}}}}}} } } tttj|t |tttj|t |tttj|t |tttj|t|tttj|t|tttj |t|tj dd5tttj t|t|tttj t|t|tttjt|t|dddn #1swxYwYtttj|t|tttj|t!| tttj|t#| tttj|t%| tttj||t'||tttj|t)|tttj||t+||tttj||t-||tttj||t/||tttj||t1||tttj||t3||tttj||t5||tttj|t7|tttj||ft9||ftttj||ft9||ftttj||ft9||ftttj|||ft9|||fdS)Nrr)rsrr\rprrrErFrMrNrrIrr'r(rrrrrr rr<r%r!r&r"rrrts r[test_testUfuncs1zTestMa.test_testUfuncs1s48F1AsBBAr2q26!99c"gg&&'''271::tBxx(()))26!99c"gg&&'''271::tBxx(()))26!99c"gg&&'''271::tBxx(())) [( ; ; ; 5 5 Brws1vvR11 2 2 2 Brvc!ff~~s2ww// 0 0 0 BrxA''r33 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 26!99c"gg&&'''29Q<<,,---29Q<<,,---29Q<<,,---2:a##WR__556662;q>>8B<<0011128Aq>>5R==112222<1%%yR'8'899:::271a==$r2,,//0002:a##WR__556662=A&& 2r(:(:;;<<<2#Aq))=R+@+@AABBB2<??IbMM223332>1a&));Bx+@+@AABBB2>1a&));1v+>+>??@@@2>1a&));Aw+?+?@@AAA2>1a),,k1b!*.E.EFFGGGGGsCI::I>I>c tgdgd}tt|jjt jutdt|tdtdttdtddg| d}tt|jjt juttt|dt j tt|jjt juttdt|ttt|dtuttddgt|ddS) Nrhrarbrgrjrrrrkrrjr)rrr)rrrrtyperpintprr\rrndarrayr r:)ruotts r[test_xtestCountzTestMa.test_xtestCountss$$$<<<888c  %0111Qc ###Qa!!!1eAQC((())***kk&!!c  %0111 5a=="*55666c  %01111eCjj!!"""c1 &&&0111Aq65a==))*****r]c b|j\ }}}}}}}}} } } tj|} t|} tt t | t j| tt t| tj| dSr) rsrprArr\maxr7rminr8)rurvrwrxryrzr{r|r}r~rrxrxmrs r[test_testMinMaxzTestMa.test_testMinMaxs48F1AsBBAr2q Xa[[Bii 3r77GN3//001113r77GN3//0011111r]c |j\ }}}}}}}}} } } tttj|t j|tttj|t j|ttdttddttdttddtttj|dt|dtttjt|ddt|dtttj|dt|dtttj |dt|dtttj |dt|dtttj t|ddt|dt| dkrtttj ||fdt||fdtttj|dt j|dtttj|dt|dtttj |dt|ddSdS)Nrraxisrj)rsrr\rpr r accumulaterKrrprodr?rrrts r[test_testAddSumProdzTestMa.test_testAddSumProds48F1AsBBAr2q26==##SZ]]3344426$$Q''):):;;<<<1c%((+++,,---1c%((+++,,---26!!$$$c!!nnn5566626&Q--a000#bq///BBCCC26!Q<<Q++,,,2711%%%wqq'9'9'9::;;;271a=='!Q--00111276"a==q111"2A...00 1 1 1 q66A:: Br~q!fa00*B8Q7799 : : : Brv}}Q**CJq!,<,<== > > > Brva||SAYY// 0 0 0 Brwq!}}gamm44 5 5 5 5 5 :r]c  tjgd}t|gd}t|gd}t|}t|t|t t tj|t |dt t|dt|dut |d|dkt |dtut t |d|dt t |dd |dd t t |dd|ddt t |dd|ddd |d<d |d<t t ||d |dd <d |dd <t t ||t|d<t t ||t|dd <t t ||||dd<t|d<t tt|tgd tgdgd|dd<t tt|tgdtgdgd|dd<t tt|tgdt t|tgdtj d dz}t|d}t t ||t ttgdt|jt t d|jtgdt"}tjgdt"}|d}|d}t%t|tt%t|tt%||t |ddjdkdS)N)rjrrrrrk)rrjrrjr) fill_valuerjr crrnrjrrr)rrjrjrrarg)rrrrjr)rjhellorr)r)rprstrreprrr\rHrr*r r r+rr5rrlrobjectrrD)rux1x2x3x4s1s2s r[ test_testCIzTestMa.test_testCIs& Xlll # # 2LLL ) ) ) 2LLL ) ) ) 2YY B R272;;RA 6 6 677888RU tBqE{{*+++1A1   2a5"Q%  !!!2ac7BqsG$$%%%2aaa5"QQQ%  !!!2abb62abb6""###112r 1Q31Q32r 12r 1Q32r 1111eLLL&9&9::;;;\\\<<<88111eLLL&9&9::;;;\\\<<<88111eLLL&9&9::;;;U<<<0011222 Yq\\C  2s # #2r 9927CCDDD3 &&''' %%%v . . X(((& 1 1 U UT"XXs###T"XXs###R1Q3 %&&&&&r]cgd}t|}t|}t||ut|d}t||utjd}t ||}t|j|utt ||jt|j|ut |d}t|jj|jjkt ||d}t|j|ut|dtud |d<t|dtut|j|utt |j dt ||d } t| j|ut| dtud | d<t| dtut| j|utt | j dt |d z|} tt| j |d zj utd } t| d<t| d } ttt| | g| ttt!| gdt#| dd} tt| gdt#| dd}tt| |dS)N)rrrjrrT)copyrrkrrlrrrrjrar))rrrjrrrrjr)rrrrr)rrrjrjrrrr)r)rrprr_datar _mask__array_interface__r*rlrrrCr\rr rB)runmrzm3ry1y1ay2y2ay3ry4y5y6s r[test_testCopySizezTestMa.test_testCopySizes OO aLL q\\R qt $ $ $  Yq\\ 2A   "###RX&&'''A BQ  -,- . . .2BQ ' ' 'B1   11V#$$$B!$$%%%BQQ''' "###A& !!!AAf$%%% "###1%%&&& 28! $ $ $r  R#X$44555 AYY1 B  ;Bx(("--...72;; 8 8 899::: B 1 - - -2///00111 B " " "2r r]c"td}gd}t|}|}t||}t |dt ut |dt uddg|ddg<t |j|ut |dt ut |dt ut t|gd t||d }|gd gd t |j|ut |dt ut |dt ut t|gddS)Nr)rrrrjrjrkrr (rj)rrrrTr)rrjr)rd)rrrrr) rr)rrrr*rr\r@)rursrrrzrvs r[ test_testPutzTestMa.test_testPut5ss 1II OO aLL VVXX !!   !!H1a& 1 !!F"###1((())*** !"4 ( ( ( iii(((r!"""!!1***++,,,,,r]ctd}t|gd}tddgddg}t|dtut|d tu||dd <t|dtut|d tutt |gd td}t|gd}|dd }tddgddg}t|dtut|d tu||dd<t|dtut|dtutt |ddgt|dtut|d tutt |gd dS) Nr)rrrrrrkrrrjrrrr)rrjrrr)rrrr*r\)rursrvr}rws r[ test_testPut2zTestMa.test_testPut2Ks 1II !/// * * * 2r(!Q ( ( (!F"###!F"###!A#!!F"###1'''(())) 1II !/// * * * acF 2r(!Q ( ( (!F"###!F"###!!!!!F"###1r2h   !!F"###1'''(()))))r]c |j\ }}}}}}}}} } } gd} tj| d} t|| | t t t || d| kdS)N) rjrrrrrrjrrrjrrjrr)rsrpr;r@rr rL)rurvrwrxryrzr{r|r}r~rrris r[test_testMaPutzTestMa.test_testMaPutes|48F1AsBBAr2q 0 0 0 JqMM!  B2DQQ'''2-../////r]c td}|dd}d|jd<t|ddk|d|zz}tt |j|tt |jd|ztt |t|zjd|z|zd |jd <td}t|d <tt|d ttk|d k}ttt|ttd kttt|tt|j kt||t}t|j |j ut|d tut|dtut|dtut|d tut|dtutt ||t|t|}t|j |j ut|d tut|dtut|dtut|d tut|dtut||}t|j |j ut|d tut|dtut|dtut|d tut|dtutt ||tgd}tgd}t|d<t||| }tt |gdt|d <t||| }tt |gdt|d tut|dtut|dtutt tt!|d|t#|dtt tt%|d|t'|dtt tt)|d|t+|dtt tt-|d|t/|dtt tt1|d|t3|dtt tt5|d|t7|dtt tt1|d|t3|dtt t9t;t=ddd gdtt t?t;t=ddd gdtt t9tt;t=dgddd j gdtt t?tt;t=dgddd j gdtt t7tt;t=dgddj gdtt t3tgdgddj gdtt tgdgd gd!tCd"tDj##}tI|j tJ}t||}tt ||tM|| |f}tt |gdt|d tut|dtut|dtutd$}t|d<td$dz}t|d<tgd%gd&}|'d}t|||}t|||} tt || ttQ| tRutt | gd't|td}tt |gd(t|dt}tt |gd)dS)*Nrrr)rjry$@rerh.rrrr)rarbrgrcrd)rjrjrjrrr)rarbrhrirj)rrrr)rrjrrr)rjrrrrrk)rjrjrjrjr)rrjrrr)rjrjrrrj)rjrrjrr)rrrjrrj)rjrrjrrj)rjrjrrr)rjrrrr)rrrrr)rrr)rr)rjrjrjrrr)rjrrrrr)rrjrr2)rrrrjrjrj)rrjrjrrr)*rrflatrr\realimagrr*rrrPrDrr6rr!r-r"r.r%r0r&r1r<r2rr,r/listranger3rlr=rpfloat32rQrrrr r:) rurvr}catestbtestctestrwcmr~s r[test_testOddFeatureszTestMa.test_testOddFeaturesls 2JJ IIaOOq $2  aK161 1626""###A ! $*C!GaK88999s 2JJ!AaD S[[())) FeAvv..//14555eAvv..//17:;;; !Q  17"###!!!!F"###!F"###1a !VQ  17"###!!F"###!F"###!! A  17"###!!F"###!F"###!!1a &&& ' ' /// " "! !QOO1+++,,---! !QOO1+++,,---!!F"###!<1 q11>!Q3G3GHHIII< a 3 3Q77'1--// 0 0 0<Q A.. Aq0A0ABBCCC< 1a 0 0!446G16M6MNNOOO< !Q335Ea5K5KLLMMM<a Q//a1C1CDDEEE< !Q335Ea5K5KLLMMM=eAhhA668M8M8MNNOOO>$uQxx..!Q779L9L9LMMNNN=tE!HH~~,;OO"="="=>?DDDH"??$$ % % % >%U1XX-<__#>#>#>?@!EEEI"??$$ % % % <d588nn+:??!<!<!<=>@@@D"??$$ % % % #E////>%@%@%@ABDDDH"??$$ % % % <AA$$$&& ' ' '\444ek8,,UE**5%  !!! 1r1g  1+++,,---!!F"###! 1II! 1IIN! $$$+=+=+= > > > XXa[[ !QNN 2q!__1b  v%&&&2,,,--... !VQ  1+++,,--- !Q  1,,,--.....r]c tttgdgdgdtttgdgdgdt d}t ddz }t |d<t |d<ttt||t t||||ttt||t t||||ttj |dkttj |d kdS) N)rjrr)rrr)rjrr)rrrrrrrr) rr\r8r7rr*rPr%r!r)rurvrws r[test_testMinMax2zTestMa.test_testMinMax2s27999iii00)))<<===7999iii00)))<<=== 1II 1IIM!!71a==%Q Aq"9"9::;;;71a==%1 q!"<"<==>>>q!!Q&'''q!!Q&'''''r]c  td}tjd}t|dd<|ddd}|ddd}t t tj|dt |dt t tj|ddt|ddt t tjt|d t|d t||t t tj t|d t|d t||tgd t}t|d<t|gd }t |d d kt |ddkt |ddkdS) Nrrrrr)rrrjrjr)abcrjdefrr)rrrr) rrpr*rrr\rOrLr#rr>rr)rurvrwts r[ test_testTakeTransposeInnerOuterz'TestMa.test_testTakeTransposeInnerOuters 2JJ IbMM!A# IIaA   IIaA  2<9--yI/F/FGGHHH271i++T!Y-B-BCCDDD28F1aLL&A,,77A;;   ! ! !28F1aLL&A,,77A;;   ! ! ! )))6 2 2! III  ! ! ! r]ctd}td}td}t|d<|dz }tt||dz|dz }tt||dztd}td}t|d<|dz}tt||dz |dz}tt||dz tddz}tddz}t|d<|dz}tt||dz|dz}tt||dztddz}td}t|d<|dz}tt|||dz}tt||tddz}tddz}t|d<|dz}tt||dz |tdz}tt|t dtdt j}td}t|d<|dz }tt||dzdS)Nrrrjrarb)r)rr*rr\r=astyperpr)rurwrvr{s r[test_testInplacezTestMa.test_testInplacesy 2JJ 2JJ BZZ1 Q1a!e  a1a!e  2JJ BZZ1 Q1a!e  a2q1u  2JJ  BZZ# 1 S1a!e  c 2q1u  2JJN BZZ1 a1a q1a 2JJ  BZZ# 1 S1a#g fRjj2tE{{##$$$ 2JJ  bj ) ) BZZ1 R1a"f r]c8td}t|ddd<|dd}tdtjdzD]I}t j||}t j|}tt||JdS)Nrrrrrrj)protocol) rr*rrrRHIGHEST_PROTOCOLdumpsloadsrr\)rurvprotorrws r[test_testPicklezTestMa.test_testPickle$s 2JJ!Bq& IIaOO1f59::  E Q///A QA Bq!HH      r]ctd}t|d<tttdkt|dtut t |ddddS)Nrrjz--r)rr*rrrr)ruxxs r[test_testMaskedzTestMa.test_testMasked.si AYY1F t#$$$1   VBqE1%%q)))))r]c tgdgd}ttdt|dttdt|gdt|gdd \}}ttd|t|d kt|dd<tt|dtutgdgd}|d d }t|ddd f<ttt|dddgtt|d dtuttddgt|dt|dd \}}tt|ddgdS)Nrrrkrbrr)rararbraweightsTrreturnedrcrrjrhrrra)rrr\rr*r)rurrZwtss r[test_testAverage1zTestMa.test_testAverage16s$$$<<<8883!,,,--...3-=-=-=>>>??@@@c+;+;+;dKKK 3   s AAA!$$$.///$$$<<<888kk!QAAAqD 73Q'''#s44555!$$$Q'61222B8WSq11122333cD999 3R!!"""""r]c T gd}gdgdg}td}ttt|ddttt|d|dt tddtdzg}ttt|dt jt jdd zd z ttt|dt jdd zdz ttt|d t|dt|ddzgttt|d| d ttt|d|gdttt|d t|dt|ddzgtd}gd}gdgdg}td}gd} tttt||ddtttt||ddttt||dtuttt|| ddtttt||ddt||} ttt| dd ttt| dgdttt| d ddgttt| d|gdtd} tddz} t| | g| | ggd d\} }tt| t|t| j|jttdddd gd\}}tt|t|ttdd\}}tt|t|ttdtdd\}}tt|t|kt d dgddggt}t|ddgd dgg}t|d}tt!|dd gt|d}tt!|dd gt|d}tt!|d t|d }tt!|d!d"gdS)#N)rrjrjrjrjr)rjrrrrrjrrrr)rrrbrgg(@rjrg @)rhrarbrgrcrf)rrrjrjrr)rrjrjrjrjrjrh)rhraX@rrcg@rd)rhrarrrcrfrTr)rrr)rrr)rrrrrmrag@g?rc)rrr rrrpr rrQr=r+r*rrrDfloatr\)ruw1w2rvrwryrzrm4m5r}abr1r2rra2daa2dmas r[test_testAverage2zTestMa.test_testAverage2IsF      "4"4"4 5 1II+++S11222B777==>>> 6!99cF1IIo. / /D))!#ry||!-CDDEEE+++")!!"4"4"4gaa6H6H6H6L!MOO P P PD"555x@@AAAB777!:!:!:<< = = =+++")!!"4"4"4gaa6H6H6H6L!MOO P P P 1XX      "4"4"4 5 !WW   a!4!41===sCCDDDa!4!41===sCCDDD Q++!444>???W\!R00q9993???U7<2#6#6Q???@@!DDD B  D))844555+++!=!=!=?? @ @ @+++c3Z88999B777!>!>!>@@ A A A 1II 1IIM1a&1a&)DAAABU2YYb ***RXrx(((iq1a&4PPPBU2YYb ***i4888BU2YYb ***i$y//DQQQBb U2YY&'''aVaV$e,,C1a&1a&!122s###4#s$$%%%1%%%53*%%&&&4(((5'""###1%%%53*%%&&&&&r]c dtdttdtdttdtdttdgggtdttdggt t ttddgt t ttddgt t ttddgddgdS)Nrjrarrk)rintrr r TypeError ValueErrorboolrus r[test_testToPythonzTestMa.test_testToPythonsQE!HH &&&S%a//***QEQC5'NN++,,,S%se --...iq!f 666j$q!f 666j$q!fAq6(B(B(BCCCCCr]c tdd}tjd5tdtdz jdddn #1swxYwYtd|zjt| jt| jtt ||jtt ||jt|j|j jutdd}t||j ktt|ttdS)Nrrjrkrr) rrprrrlr7r8rrrrrr4)rur{rvs r[test_testScalarArithmeticz TestMa.test_testScalarArithmeticsw 11    [ ) ) ) ) ) Qq\' ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )R " " B$%%%B$%%% !RX^3444 !!    ag%&&&SWWc"56677777s%AAAc tgd}tt||jtt||jtt||jtt||jtt|ddddd|jdddddtt| gd|j gdtt| |j tt| |j tddgddgg}tt| |j tt| |j tt|ddg|jddgtt||jdS)Nrjrrrrjrrrrjrrj)rrr\anyrr argmaxargminrcompressconjrdiagonalrKrLrO)rur%rs r[test_testArrayMethodszTestMa.test_testArrayMethodss~ )))  15577AGKKMM**+++15577AGKKMM**+++188::qw~~//00111188::qw~~//00111188Aq!Q**7>>!Q1a88:: ; ; ;1::iii((!'*:*:999*E*EFFGGG16688QW\\^^,,---1;;==!'"3"3"5"566777 Aq6Aq6" # #1::<<!1!1!3!34455515577AGKKMM**+++1661a&>>17<<A#7#7889991;;==!'"3"3"5"56677777r]cRtgd}t|jddS)Nr7rj)rrndimrur%s r[test_testArrayAttributeszTestMa.test_testArrayAttributess+ )))  QVQr]chtdttjD dS)Nchg|]/}|ttv|d-|0S)_)dirr startswith).0rs r[ z'TestMa.test_testAPI..sH,,,1[!1!111c**2111r])rrGrprr1s r[ test_testAPIzTestMa.test_testAPIsF,,BJ,,,, - - - - -r]ctgd}tgdgd}t|djdt|djdt|djddS)Nr7r8rkrrrj)rrrD)rur%r&s r[test_testSingleElementSubscriptz&TestMa.test_testSingleElementSubscriptsu )))   )))))) , , ,QqTZ$$$QqTZ$$$QqTZ$$$$$r]c~tgdgd}|dk}d||<t|dtudS)Nr)rjrrjrrkrrr)rrr*)rur%rs r[test_assignment_by_conditionz#TestMa.test_assignment_by_conditionsJ ,,,\\\ 2 2 2 F!!r]ctddgddg}tddgddg}|dk}||||<ddg}t|j|dS)NrrjFrkT)r+rrl)rur%r&rl expected_masks r[test_assignment_by_condition_2z%TestMa.test_assignment_by_condition_2sm !Quen 5 5 5 !QtTl 3 3 31uD'$ QV]+++++r]N)$__name__ __module__ __qualname__rrpytestmark parametrizerrrrrrrrrrrrrrrr rrrr+r2r5r?rCrKrMrOrRrr]r[r_r_$s  ; ; ;    [S66"23343&$=$=$=L222 HHH> + + +222666*0'0'0'd111f---,***4000c/c/c/J ( ( ((000d***###&8'8'8'tDDD 8 8 8888"   --- %%%   ,,,,,r]r_c,eZdZdZdZdZdZdZdS) TestUfuncsctdddtdz gdzddgdgdzztdddtdz gdzddgdgdzzf|_dS)Nrarrrrjrrk)rrqrsr1s r[rzTestUfuncs.setup_methodsraR!V,q01va7GHHHaR!V,q01va7GHHHKr]c gd}dD]?} tt|}n%#t$rtt|}YnwxYwttj|}|jd|j}t j5||vrt j d|dvrt j d||}||}dddn #1swxYwYtt| d| d|tt|j|jAdS)N) rIarctanhrrarccoshr]r'r(r true_divide floor_divide remainderfmod)*rIr'r(rrrErrMrrrrFrrNarcsinhr^r]r fabsnegativefloorceil logical_notr rJr9rr_r`rarbhypotrrr<r&r"r%r! logical_and logical_or logical_xorr)r)r]r'r(r4r)getattrumathAttributeError fromnumericrprrsninrseterrrr\reqmaskrl)ruf_invalid_ignorefufmfargsurmrs r[test_testUfuncRegressionz#TestUfuncs.test_testUfuncRegressions@@@> . .A  -UA&&! - - -[!,, -""B6'26'?D  (((Ih////333IX....RYRY                 Bryy||RYYq\\155 6 6 6 F27BG,, - - - -= . .s!!AA=CC C c&|jd}tt|d tt|dt t |ddddt t |dddS)Nrrr)rsrrrGrrKr?rBs r[ test_reducezTestUfuncs.test_reduces F1IGAA&&&&'''###$$$S2A2Q'''+++WQQ'''+++++r]ctdddd}t|dk|}t||t|dt |d|dkt |dgdkt |ddjt |ddjdS)Nrj rrrr)rrrr) rrr6rrrrr rl)rur%amasks r[ test_minmaxzTestUfuncs.test_minmaxs 1bMM ! !!Q ' 'QUA&&UYY[[!%%''***UYY[[!$$$1q)..001111-2244555 ! Q$%%% ! Q$%%%%%r]cdD]B}tgdgd}ttt|dgCdS)Nz?bhilqpBHILQPfdgFDGO)rjrrr)rrrjrjrkr)rrr\r;)rurrvs r[ test_nonzerozTestUfuncs.test_nonzerosX' ) )Alll666A BwqzzA3'' ( ( ( ( ) )r]N)rSrTrUrr{r}rrrr]r[rZrZscKKK#.#.#.J,,,&&&)))))r]rZc>eZdZdZdZdZdZdZdZdZ dZ d S) TestArrayMethodsctjgd}|dd}|dddd}tjgd}t||}t|||j}t|||j}|||||||f|_dS)N)$g @gGz.@g-!@g!@g&1?gʡE@g\( @gQ@g{G#@gPn@gh|?!@g!rh@gjt@gJ +@gzGa@g!rh@gʡE@g-@g)\(@g(\B#@g +@guV @g~j@g"~@gK7@g ףp=J#@gGz@gMb@g}?5^#@g/$@g!rhm@g-#@gJ +@gMb@gʡ@g-?rrr)$rrjrrjrrrjrrjrjrrjrrrrjrrjrrrrjrjrjrjrrrjrrrrrjrrjr)datarl)rprrrDrs)rurvXXXrmxmXmXXs r[rzTestArrayMethods.setup_methods H@@@ A A IIaOO YYq!Q " " H((( ) )  " " "  !' 2 2 3 3 3!))BH"5"5666QAr2s+r]c |j\}}}}}}}|}t||t t||t |j|zdz dS)Nrr) rsr>rtrace compressedrKrr\rl) rurvrrrrrrmXdiags r[ test_tracezTestArrayMethods.test_trace&s&*f#Ar1b"cRXXZZ!9!9!;!;!?!?!A!ABBB288::7799s6;+E01(3(3(3344 5 5 5 5 5r]c |j\}}}}}}}|dd}tt|j|jtt|j|ddtt|j|jdddS)Nrr)rscliprr\rlr) rurvrrrrrrclippeds r[ test_clipzTestArrayMethods.test_clip.s&*f#Ar1b"c''!Q--7<))***7=!&&A,,//0007="(--1"5"56677777r]c |j\}}}}}}}|j\}}t||t j|t j} t j|t j} t|D]5} |dd| f| | <6t|D]1} || | | <2tt|d| tt|d| dSNrrj) rsrDrptprrprQfloat_rrr\) rurvrrrrrrrrowscolsks r[test_ptpzTestArrayMethods.test_ptp5s?&*f#Ar1b"cARVVXXr}}2244555x29%%x29%%q 2 2AAh))++//11DGGq / /Ae&&((,,..DGG266!99d##$$$266!99d##$$$$$r]c|j\}}}}}}}|dd}tt|d|dddf|dd} t | jddS)Nrrjrr)rrrr)rsswapaxesrr\rrD) rurvrrrrrr mXswapped mXXswappeds r[ test_swapaxeszTestArrayMethods.test_swapaxesBs&*f#Ar1b"cKK1%% 9R="QQQU),,---\\!Q'' Z%|44444r]c|j\}}}}}}}|d}tt|j|dd|d}tt|j|dddSr)rscumprodrr\rr rurvrrrrrrmXcps r[ test_cumprodzTestArrayMethods.test_cumprodIs&*f#Ar1b"czz!}}4:ryy||33A6677888zz!}}4:ryy||33A667788888r]c|j\}}}}}}}|d}tt|j|dd|d}tt|j|dddSr)rscumsumrr\rrrs r[ test_cumsumzTestArrayMethods.test_cumsumPs&*f#Ar1b"cyy||4:ryy||2215566777yy||4:ryy||221556677777r]c .|j\}}}}}}}tt|d|tt|d|tt|dj|djtt|j|j|d|d} }tdD]} tt| | || tt|| |dd| ftttj || |dd| fdS)Nrrrrjr) rsrr\varrstdrDrrprI) rurvrrrrrrmXvar0mXvar1rs r[ test_varstdzTestArrayMethods.test_varstdWs&*f#Ar1b"c266t6$$bmmoo&9&9&;&;<<===266t6$$bmmoo&9&9&;&;<<===3777??("&&a&..*>??@@@26688>15577=11222FFFNNBFFFNNq = =A Bvay"Q%"2"2"4"4"8"8":":;; < < < Bvay"QQQT("5"5"7"7";";"="=>> ? ? ? Brwvay))!!!!Q$x224488::<< = = = = = =r]N) rSrTrUrrrrrrrrrr]r[rrs,,,,555888 % % %555999888 = = = = =r]rcz|tur |tuS|tur |tuS||kSr)r:r )ryrzs r[rsrses< V||V| V||V| "H>>  r])rS)b functoolsrrVnumpyrpnumpy.core.umathcorernnumpy.core.fromnumericrp numpy.testingrrrnumpy.marrr r r r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrLrMrNrOrPrQ numpy.compatrRrqr\r_rZrrsrr]r[rsi  ,,,,,,,,,                                                                                                                                                           Uf ,f ,f ,f ,f ,f ,f ,f ,R>)>)>)>)>)>)>)>)BT=T=T=T=T=T=T=T=nr]