bg*=ddlZddlmZddlZddlmZddlmcmZddl m Z ej Z GddZ e dkrdZd ezZd \ZZed d D]Zd ezZejeeeedzZejeZedezdzeedezededdded ddSdS)N)reduce)float_) build_err_msgc0eZdZdZ ddZddZejddZejdd Z ejdd Z ejdd Z ejdd Z ejdd Z ejddZejddZejddZejddZdS) ModuleTesterc||_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j |_ t|_ |j |_ |j |_ |j |_ |j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_ |j|_n!#t8$r|jj|_YnwxYwg|_dSN)moduleallequalarangearray concatenatecountequalfilledgetmask getmaskarrayidinner make_maskmasked masked_array masked_valuesmask_ornomaskonesouterrepeatresizesorttake transposezerosMaskTypeumathAttributeErrorcore testnames)selfr s l/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/ma/timer_comparison.py__init__zModuleTester.__init__sA  m \ !-\ \ m ~ "/\ )m "/#1~ m K \ m m K K )\   +DJJ + + +*DJJJ +s EE,+E,Tc T||}||}|||||} |||| |}|||| |}|jjdkrm|t}t|tj r#|j dkrd|tj |<ntj |rd}|jjdkrm|t}t|tj r#|j dkrd|tj |<ntj |rd} |j dkp|j dkp|j |j k} | s5t||g|d|j d|j dz|d } | s J| |||} | |jur|r|| | } t| t r| } dg} n<| } | } | } | sSd d | dzt+| z z }t||g|d |dz|d } | s J| dSdS#t,$r*}t||g||d } t-| |d}~wwxYw)zZ Assert that a comparison of two masked arrays is satisfied elementwise. maskOrz (shapes z, z mismatch)xy)headernamesdgY@z (mismatch z%)N)rrrrdtypecharastyper isinstancenpndarraysizeisnanshaperrboolravelalltolistrlen ValueError)r) comparisonr4r5err_msgr6 fill_valuexfyfmcondmsgvalreducedmatches r*assert_array_comparez!ModuleTester.assert_array_compare0s? [[^^ [[^^ LLa$,,q// : : KK))"1)55z B B KK))"1)55z B B GLC    A!RZ(( QVaZZ!""(1++!  GLC    A!RZ(( QVaZZ!""(1++!  )GrM2QW]Iqw!'7ID !#QF$+&Q17&Q&Qag&Q&Q&Q%R+1*4 666   S  t*Q""C ## #''!'44#t$$ +#))++{{}}!..** !E'--"2"223w<<??#QF$+G=BUU&D%E+1*4 666   S  t ! !!  ) ) )AjQQQCS//q ( )s8D7K33 L'=%L""L'cD||j|||ddS)zH Checks the elementwise equality of two masked arrays. zArrays are not equal)rIr6N)rTr)r)r4r5rIs r*assert_array_equalzModuleTester.assert_array_equalgs= !!$*aG)? " A A A A Aignore)rDctjddddtdz dddddddg }gd }||| }|d d S) ! Tests creation ?@@@$$@@ r1rrrrrr1rrrrrr.rN)r=r pir)r)r4rMxms r*test_0zModuleTester.test_0os` Hb"b#r#vr2tS"b"M N N 0 0 0   qq  ) ) 1rWctjddddtdz dddddddg }tjgd }gd }gd }||| }||| }tj|d |}|d ||z dsJ|j}|j td|ksJ| |t|td|z ksJdD]_}||_||_||_||_||_| |t|td|z ksJ`dS)rZr[r\r]r^r_r`rarb) r_rbr]ggrhr`rar[rhrbrc) rrr1rrr1r1rrrrr1r.g@xDrc ||zSr r2r3s r*z%ModuleTester.test_1..s QqSrWc ||zSr r2r3s r*rjz%ModuleTester.test_1..s acrW)))c ||zSr r2r3s r*rjz%ModuleTester.test_1..s !A#rWN) r=r rdrwhereset_fill_valueranyrAr?rrrF) r)r4r5m1m2reymrKss r*test_1zModuleTester.test_1zs Hb"b#r#vr2tS"b"M N N HJJJ K K 1 1 1 1 1 1   qr  * *   qr  * * Xb&! $ $ &!!!2~~a  $$&&''& Gw&!444444zz"~~R6//2+F+F!FFFFF! L LAAGAGBHBHBH::b>>SWWvoor/J/J%JJJJJJ  L LrWcZtjgd}||gd}||gd}||}t|t|t |dt |dusJ|d|dksJd|d<d|d<|||d|dd <d|dd <|j|d<|j|dd <||d d <|j|d<|gd gd |d d <|gd gd |d d <tjd dz}| |d}|gdt}tjgdt}|d|d|ddj dksJgd}| |}| |}||usJ| |d}||usJd S)z2 Tests conversions and indexing. )r1rorlrmr1rrrr.)rr1rr1r1 rocrmN)r1rormrl)rr1r1rr[rb)r1hellororm)r)rrr1rr)copy) r=r strreprtyperVrrr robjectrAr) r)x1x2x3x4nrMrum3s r*test_2zModuleTester.test_2sJ Xlll # # ZZZ . . ZZZ . . ZZ^^ B RBqE{{d2a5kk))))!u1~~~~11 B'''1Q31Q3 1+1Q3111 1!!,,, ==111!!,,, ==111 Yq\\#    C ( ( ZZ***F 3 3 X(((& 1 1 1 1!A#w}$$$$ OO NN1   ^^A  Bwwww ^^AA^ & &{{{{{{rWc|d}|j|d<||d}||||g|sJ|||gdsJ||dd}||gd||dd}|||sJ|dd}|||sJ|dd}|||sJd S) z& Tests resize/repeat rlro))rrr1rrrr1r)rororororaxis)rrr1r1rorormrmN)r rrr rrrrV)r)ry4y5y6y7y8s r*test_3zModuleTester.test_3sO [[^^ 1 [[T " "}}T--r2h77<<<<<}}T\\"--/G/G/GHHHHH [[\[ 2 2 $<$<$<=== [[QQ[ ' '}}R$$$$$ YY|!Y , ,}}R$$$$$ YYq!__}}R$$$$$$$rWc :|d}tjd}|j|dd<|ddd}|ddd}|tj|d||dsJ|tj|dd||ddsJ|tj||d ||d |||sJ|tj ||d ||d | ||sJ| gd t}|j|d<||gd }|d d ksJ|ddksJ|ddksJd S)zB Test of take, transpose, inner, outer products. r}rnrormrl)rorr1r1r)abcr1defrorm)rrmrlrN) r r=rreshaper r"r!rrrr r)r)r4r5ts r*test_4zModuleTester.test_4s KKOO IbMM!A# IIaA   IIaA  }}R\!Y7799U9UVVVVV}}RWQ 155tyyIq7Q7QRRRRR}}RXdkk!Q&7&7Q9J9JKK JJq!,,.. . ..}}RXdkk!Q&7&7Q9J9JKK JJq!,,.. . .. JJ... 7 7{! IIa # #tu}}}}tqyyyytqyyyyyyrWc|d}|d}|d}|j|d<|dz }|||dzsJ|dz }|||dzsJ|d}|d}|j|d<|dz}|||dz sJ|dz}|||dz sJ|ddz}|ddz}|j|d<|dz}|||dzsJ|dz}|||dzsJ|ddz}|ddz}|j|d<|dz}|||sJ|dz}|||sJ|ddz}|ddz}|j|d<|dz}|||dz sJ||dz}|||d|dt }|d}|j|d<|dz }|||dzsJdS)z* Tests inplace w/ scalar ror1r[r])rN)r rr rVrr;r)r)r4r5res r*test_5zModuleTester.test_5s KKOO KKOO [[__ 1 Q}}Q!$$$$$ a}}R1%%%%% KKOO [[__ 1 Q}}Q!$$$$$ a}}R1%%%%% KKOOC  [[__S  1 S}}Q!$$$$$ c }}R1%%%%% KKOOA  [[__Q  1 Q}}Q""""" a}}R##### KKOOC  [[__S  1 S}}Q#&&&&& dkk"oo DIIe$4$4555 KKOO " "6 * * [[__ 1 R}}QB'''''''rWc|dt}|d}|dt}|j|d<|j}|dt}|j|d<||z }||z }||||zsJ||||zsJ||j|||jsJ|dt}|dt}|j|d<|j}|dt}|j|d<||z}||z}||||z sJ||||z sJ||j|||jsJ|dt}|dt}|j|d<|j}|dt}|j|d<||z}||z}||||zsJ||||zsJ||j|||jsJ|dt}|dt}|j|d<|j}|dt}|j|d<||z}||z}dS)z) Tests inplace w/ array r)r9roN)r rrr/r r)r)r4r5rerMas r*test_6zModuleTester.test_6"s KK&K ) ) KKOO [[6[ * * 1 G KK&K ) ) " Q a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) ) [[6[ * * 1 G KK&K ) ) " Q a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) ) [[6[ * * 1 G KK&K ) ) " Q a}}Q!$$$$$}}R1%%%%%}}RWdll1af&=&=>>>>> KK&K ) ) [[6[ * * 1 G KK&K ) ) " Q arWcZ|dddtdz gdzddgdgdzz|dddtdz gdzddgdgdzzf}dD]} t|j|}n%#t$rtt |}YnwxYwt|j|}|d |j}||}||}|| d| d|||j |j d S) z Tests ufuncr[rrror1rnr.)sqrtloglog10exp conjugateN) r rdgetattrr%r& fromnumericr ninrVr_mask)r)dfufmfargsurmrs r*test_7zModuleTester.test_7VsRZZaRT*1,Aq61#a%%>??? T\\#?O?O?O\%P%PQQQll30@0@0@1lMM  V,,,czzzzAAA||Ca|((DK7777jj))) j==kk!QKAAAqD   Sq 9 9C:FFF||Ca|((+t{:::: R$,,s,*C*CDDDll3Ql;;  b"X...      "4"4"4 5 KKNN  QQ 7 7===  QQ C CSIII JJ ADKKNN(:; < <  Q 5 5rv}}RYq\\7R7RSU7UVY7YZZZ  QQ 7 719J29MNNN  QQ 7 7$,,qq,:Q:QSWS_S_`ahiS_SjSjmpSp9qrrr  Qb A A6JJJ  QQ C CE^E^E^___  QQ 7 7$,,qq,:Q:QSWS_S_`ahiS_SjSjmpSp9qrrr ZZ]]      "4"4"4 5 YYq\\     T->->q"-E-EA N NPSTTT  T->->q"-E-EA N NPSTTT  T->->q"-E-EA N NPSTTT  4<<8I8I!R8P8PWX<+Y+Y Z Z\]^^^   a $ $  Q 5 5v>>>  QQ 7 79U9U9UVVV  QQ 7 7#sDDD  QQ C CEbEbEbcccccrWcv|d}|j|dd<|ddd}dS)Nrr}rnrormrl)r rr)r)r4s r*test_AzModuleTester.test_As9 KKOO!A# IIaA  rWN)r,r,T)r,)__name__ __module__ __qualname__r+rTrVr=errstaterfrxrrrrrrrrr2rWr*rr s   DIK$(5)5)5)5)nAAAAR[XR[XLLL6R[X(((TR[X%%%&R[X.R[X2(2(2(hR[X111fR[X 8 8 8DR[X+d+d+dZR[XrWr__main__zNfrom __main__ import ModuleTester import numpy tester = ModuleTester(module) zimport numpy.ma.core as module )rrr1rztester.test_%i()rz#%iz2..................................................zModuleTester.test_%i.__doc__zcore_current : z.3fz - )timeit functoolsrnumpyr=rnumpy.core.fromnumericr'r numpy.testingrrdrr setup_base setup_curnrepeatnlooprangeifuncTimerrcurr printevalr2rWr*rs ,,,,,,,,,''''''U^^^^^^^^B  z4J3Z?IWe U1a[[==!A%fl4++227E"HEEbgcll eai& !!! dd1A566777 ;A;;;s1v;;;<<<<==rW