
    >>i3                        S SK r S SKJr  S SKrS SKrS SKJs  Jr  S SK	Js  J
r
  S SKJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;J<r<J=r=J>r>J?r?J@r@JArAJBrBJCrCJDrDJErEJFrFJGrGJHrHJIrIJJrJJKrKJLrLJMrMJNrNJOrOJPrPJQrQJRrRJSrSJTrTJUrUJVrV  S SKWJXrXJYrYJZrZ  \R                  r[SS jr\ " S S5      r] " S S	5      r^ " S
 S5      r_S r`g)    N)reduce)KMaskedArrayMaskTypeabsoluteaddallallcloseallequalalltruearangearccosarcsinarctanarctan2arrayaveragechooseconcatenate	conjugatecoscoshcountdivideequalexpfilledgetmaskgreatergreater_equalinnerisMaskedArrayless
less_equalloglog10	make_maskmaskedmasked_arraymasked_equalmasked_greatermasked_greater_equalmasked_insidemasked_lessmasked_less_equalmasked_not_equalmasked_outsidemasked_print_optionmasked_valuesmasked_wheremaximumminimummultiplynomasknonzero	not_equalonesouterproductputravelrepeatresizeshapesinsinhsometruesortsqrtsubtractsumtaketantanh	transposewherezeros)assert_assert_equalassert_raisesc                 R    [        X5      nU(       d  [        SU SU  SU 35        U$ )NzNot eq:
z
----)r	   print)vwmsgresults       Y/var/www/html/land-tabula/venv/lib/python3.13/site-packages/numpy/ma/tests/test_old_ma.pyeqrZ   [   s.    a^FuBqcs+,M    c                   
   \ rS rSrS rS r\R                  R                  SSS/5      S 5       r	S r
S	 rS
 rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS rS r S r!S  r"S! r#S" r$S#r%g$)%TestMab   c                 ~   [         R                  " SSSS[        S-  SSSSSSS/5      n[         R                  " / S	Q5      nSn/ S
Qn/ SQn[        XS9n[        X%S9n[         R                  " / SQ5      n[        U/ SQS9n	[         R                  " USU5      n
UR                  nUR                  S5        XX4XVXxXU4$ )N      ?g              @      @      @      $      $@      @)rc           rf   ra   g            rg   rd   re   r`   rg   rf   )   r   r   r   r   r   ri   r   r   r   r   r   )r   r   ri   r   r   ri   ri   r   r   r   r   ri   mask)g      rg         ?g?r   ri   r   r   @xD)npr   pirM   rA   set_fill_value)selfxya10m1m2xmymzzmxfss               rY   _create_dataTestMa._create_datad   s    HHb"b#rCxRsBBOPHHJK1111HH&'1<(XXb%#GG
% SbbRQ66r[   c                 N   U R                  5       u  n  p#p$    p%n[        [        U5      (       + 5        [        [        U5      5        [        [	        U5      U5        [        UR                  U5        [        UR
                  UR
                  5        [        UR                  [        S U5      5        [        [        U5      [        U5      [        S U5      -
  5        [        [        XE5      5        [        [        [        US5      U5      5        [        [        X5      5        g )Nc                 
    X-  $ N rs   rt   s     rY   <lambda>)TestMa.test_testBasic1d.<locals>.<lambda>{       !%r[   c                 
    X-   $ r   r   r   s     rY   r   r   |       aer[   rn   )r~   rO   r!   rP   rA   dtypesizer   r   lenrZ   r   )rr   rs   _rv   rx   r|   r}   s          rY   test_testBasic1dTestMa.test_testBasic1ds   s    -1->->-@*1aQAq!M!$$%b!"U2Y"RXXq!RXXqww'RWWf%7;<U2YB&1CR*H HI2
6"e$b)*1	r[   r}         )      c                 H   U R                  5       u  p#pEpFn  pHnXl        Xl        Xl        Xl        Xl        [        [        U5      (       + 5        [        [        U5      5        [	        [        U5      U5        [	        UR                  U5        [	        UR
                  [        S U5      5        [	        [        U5      [        U5      [        S U5      -
  5        [        [        Xh5      5        [        [        [        US5      U5      5        [        [        X&5      5        g )Nc                 
    X-  $ r   r   r   s     rY   r   )TestMa.test_testBasic2d.<locals>.<lambda>   r   r[   c                 
    X-   $ r   r   r   s     rY   r   r      r   r[   rn   )r~   rA   rO   r!   rP   r   r   r   r   rZ   r   )	rr   r}   rs   rt   r   rv   rx   ry   r|   s	            rY   test_testBasic2dTestMa.test_testBasic2d   s     /3.?.?.A+aQB1!M!$$%b!"U2Y"RXXq!RWWf%7;<U2YB&1CR*H HI2
6"e$b)*1	r[   c                    U R                  5       u  pn  pEn  pGn[        SS/SS//5      n	[        U	SS/SS//5      n
[        [	        X-  X-  5      5        [        [	        X-   X-   5      5        [        [	        X-
  X-
  5      5        S GH  nUR                  U5      nUR                  U5      nUR                  U5      nUR                  U5      nUR                  U5      n[        [	        U* U* 5      5        [        [	        X-   XV-   5      5        [        [	        X-
  XV-
  5      5        [        [	        X-  XV-  5      5        [        R                  " SSS9   [        [	        X-  XV-  5      5        S S S 5        [        [	        X2-   X6-   5      5        [        [	        X2-
  X6-
  5      5        [        [	        X2-  X6-  5      5        [        R                  " SSS9   [        [	        X2-  X6-  5      5        S S S 5        [        [	        X-   XS-   5      5        [        [	        X-
  XS-
  5      5        [        [	        X-  XS-  5      5        [        [	        X-  XS-  5      5        [        [	        US-  US-  5      5        [        [	        [        U5      S-  [        U5      S-  5      5        [        [	        X-  XV-  5      5        [        [	        [        R                  " X5      [        XV5      5      5        [        [	        [        R                  " X5      [        XV5      5      5        [        [	        [        R                  " X5      [        XV5      5      5        [        R                  " SSS9   [        [	        [        R                  " X5      [        XV5      5      5        S S S 5        GM"     g ! , (       d  f       GNB= f! , (       d  f       GN= f! , (       d  f       GMY  = f)	Nri   r   r   r   ))   r   )r   r   ignorer   invalid      @)r~   r   r(   rO   rZ   reshapero   errstateabsr   rG   r6   r   )rr   rs   rt   ru   r   rx   ry   r|   r}   a2da2dms              rY   test_testArithmeticTestMa.test_testArithmetic   s   /3/@/@/B,c1aRA1aVaV$%C1a&1a&!1239cj)*39cj)*39cj)*(A		!A		!AABABABBrB3K Bqubg&'Bqubg&'Bqubg&'Hh?15"'*+ @Bsw)*Bsw)*Bsw)*Hh?37CH-. @Bqw)*Bqw)*Bqw)*Bqw)*BqAvrQw'(Bs1v}c"gn56Bqvrx()Brvva|S[12Br{{1((2*:;<Br{{1((2*:;<Hh?299Q?F2N;< @?7 ) @?
 @? @?s$   N/O.3O/
N>	
O	
O#	c                     [         R                  " S/5      n[        S/5      n[        [        X-   [        5      5        [        [        X!-   [        5      5        g )Nri   )ro   r   rO   
isinstancer   )rr   namas      rY   test_testMixedArithmeticTestMa.test_testMixedArithmetic   s>    XXqc]A3Z
27K01
27K01r[   c                    U R                  5       u  p    p4pVn  n[        [        [        R                  " U5      [	        U5      5      5        [        [        [        R
                  " U5      [        U5      5      5        [        [        [        R                  " U5      [        U5      5      5        [        [        [        R                  " U5      [        U5      5      5        [        [        [        R                  " U5      [        U5      5      5        [        [        [        R                  " U5      [        U5      5      5        [        R                  " SSS9   [        [        [        R                  " [        U5      5      [        U5      5      5        [        [        [        R                  " [        U5      5      [        U5      5      5        [        [        [        R                  " [        U5      5      [        U5      5      5        S S S 5        [        [        [        R                  " U5      [        U5      5      5        [        [        [        R                   " U5      [!        U5      5      5        [        [        [        R"                  " U5      [#        U5      5      5        [        [        [        R$                  " U5      [%        U5      5      5        [        [        [        R&                  " X5      ['        XE5      5      5        [        [        [        R(                  " U5      [)        U5      5      5        [        [        [        R*                  " X5      [+        XE5      5      5        [        [        [        R,                  " X5      [-        XE5      5      5        [        [        [        R.                  " X5      [/        XE5      5      5        [        [        [        R0                  " X5      [1        XE5      5      5        [        [        [        R2                  " X5      [3        XE5      5      5        [        [        [        R4                  " X5      [5        XE5      5      5        [        [        [        R6                  " U5      [7        U5      5      5        [        [        [        R8                  " X45      [9        XE45      5      5        [        [        [        R8                  " X45      [9        X45      5      5        [        [        [        R8                  " X45      [9        XB45      5      5        [        [        [        R8                  " XU45      [9        XU45      5      5        g ! , (       d  f       GNl= f)Nr   r   )r~   rO   rZ   ro   r   r   rB   rC   rJ   rK   r   rF   r   r$   r%   r   r   r   r   r   r   r   r9   r"   r   r#   r   r   r   )rr   rs   rt   r   rx   ry   rz   r{   s           rY   test_testUfuncs1TestMa.test_testUfuncs1   s0   -1->->-@*aA2"a266!9c"g&'2771:tBx()266!9c"g&'2771:tBx()266!9c"g&'2771:tBx()[[(;Brwws1vR12Brvvc!f~s2w/0BrxxA'r34 < 	266!9c"g&'299Q<,-299Q<,-299Q<,-2::a#WR_562;;q>8B<01288A>5=122<<%y'89:2771=$r,/02::a#WR_562==&
2(:;<2##A)=+@AB2<<?IbM232>>1&);x+@AB2>>1&);v+>?@2>>1&);w+?@A2>>1),k1!*.EFG) <;s   B2U22
Vc           
      B   [        / SQ/ SQS9n[        [        U5      R                  R                  [
        R                  L 5        [        S[        U5      5        [        S[        S5      5        [        [        S[        SS/S95      5        UR                  S5      n[        [        U5      R                  R                  [
        R                  L 5        [        [        [        US5      [
        R                  5      5        [        [        U5      R                  R                  [
        R                  L 5        [        [        S[        U5      5      5        [        [        [        US5      5      [        L 5        [        [        SS/[        US5      5      5        g )	Nrg   r`   ra   rf   ri   r   r   r   rj   r   ri   r   )r   r   r   )r   rO   r   r   typero   intprP   rZ   r   r   ndarrayr   r7   )rr   otts     rY   test_xtestCountTestMa.test_xtestCount   s   $<8c
  %%01Qc
#Qa!1eAQC()*kk&!c
  %%01
5a="**56c
  %%011eCj!"c1&&01Aq65a=)*r[   c                 D   U R                  5       u  n      p#        n[        R                  " U5      n[        U5      n[        [	        [        U5      [        R                  " U5      5      5        [        [	        [        U5      [        R                  " U5      5      5        g r   )
r~   ro   r>   rO   rZ   maxr4   r   minr5   )rr   rs   r   rx   xrxmrs         rY   test_testMinMaxTestMa.test_testMinMax   su    +/+<+<+>(1aA1aAqXXa[Bi 	3r7GNN3/013r7GNN3/01r[   c                 J   U R                  5       u  p    p4n    p6[        [        [        R                  R                  U5      [        R
                  " U5      5      5        [        [        [        R                  R                  U5      [        R                  " U5      5      5        [        [        S[        [        S5      SS95      5        [        [        S[        [        S5      SS95      5        [        [        [        R                  " USS9[        USS95      5        [        [        [        R                  " [        US5      SS9[        USS95      5        [        [        [        R                  " US5      [        US5      5      5        [        [        [        R                  " USS9[        USS95      5        [        [        [        R                  " US5      [        US5      5      5        [        [        [        R                  " [        US5      SS9[        USS95      5        [        U5      S:  a  [        [        [        R                  " X4S5      [        XE4S5      5      5        [        [        [        R                  R                  US5      [        R
                  " US5      5      5        [        [        [        R                  " US5      [        US5      5      5        [        [        [        R                  " US5      [        US5      5      5        g g )Nr   r   axisri   )r~   rO   rZ   ro   r   r   
accumulaterH   r   r   prodr<   r   r   )rr   rs   rt   r   rx   ry   r}   s          rY   test_testAddSumProdTestMa.test_testAddSumProd   s   ,0,=,=,?)aA2q!Q266==#SZZ]34266$$Q'):;<1c%(+,-1c%(+,-266!!$c!!n56266&Q-a0#bq/BC266!Q<Q+,27711%wqq'9:;2771a='!Q-012776"a=q1"2A.0 	1q6A:Br~~qfa0*B8Q79 :Brvv}}Q*CJJq!,<=>Brvva|SAY/0Brwwq!}gam45 r[   c           
      2   [         R                  " / SQ5      n[        U/ SQS9n[        U/ SQS9n[        U5      n[        U5        [        U5        [	        [        [         R                  " U5      [        USS95      5        [	        [        US   5      [        US   5      L 5        [	        US   US   :H  5        [	        US   [        L 5        [	        [        US   US   5      5        [	        [        USS	 USS	 5      5        [	        [        US S  US S  5      5        [	        [        USS  USS  5      5        S
US'   S
US'   [	        [        X5      5        SUSS& SUSS& [	        [        X5      5        [        US'   [	        [        X5      5        [        USS& [	        [        X5      5        XS S & [        US'   [	        [        [        U5      [        / SQ5      5      5        [        / SQ/ SQ5      US S & [	        [        [        U5      [        / SQ5      5      5        [        / SQ/ SQ5      US S & [	        [        [        U5      [        / SQ5      5      5        [	        [        U[        / SQ5      5      5        [         R                  " S	5      S-  n[        US5      n[	        [        X5      5        [	        [        [        / SQ[        5      UR                  5      5        [	        [        SUR                   5      5        [        / SQ["        5      n[         R                  " / SQ["        5      nUS   nUS   n[%        [        U5      [        5        [%        [        U5      [        5        [%        XV5        [	        USS R&                  S:H  5        g )N)ri   r   r   r   r   rj   )r   ri   r   ri   r   )
fill_valueri   r      	   c   r   rm   ri   r   r   r   )r   ri   ri   r   r`   rf   )r   r   r   ri   r   )ri   hellor   r   )r   )ro   r   strreprrO   rZ   rE   r   r'   r
   r   r(   r   r2   r   rk   r   objectrP   rA   )rr   x1x2x3x4s1s2s          rY   test_testCITestMa.test_testCI  s   XXl#2L)2L)2YBR2772;RA 678RUtBqE{*+1A1 2a5"Q% !2a7BqG$%2a5"Q% !2ab62ab6"#112
1Q1Q2
12
1Q2
11eL&9:;\<81eL&9:;\<81eL&9:;U<012YYq\C2s#2
9277CD3&'%v.XX(&1UUT"Xs#T"Xs#R1Q%&r[   c                    / SQn[        U5      n[        U5      n[        X#L 5        [        USS9n[        X$L5        [        R                  " S5      n[	        XRS9n[        UR
                  UL5        [        [        XVR
                  5      5        [        UR                  UL 5        [	        USS9n[        UR                  R                  UR                  R                  :H  5        [	        XTSS9n[        UR                  UL 5        [        US   [        L 5        S	US'   [        US   [        L5        [        UR                  UL 5        [        [        UR                  S5      5        [	        XRS
S9n	[        U	R                  UL5        [        U	S   [        L 5        S	U	S'   [        U	S   [        L5        [        U	R                  UL5        [        [        U	R                  S5      5        [	        US-  US9n
[        [        U
5      R                  US-  R                  L 5        [        S5      n[        US'   [        US5      n[        [        [        X/5      U5      5        [        [        [!        U5      / SQ5      5        [#        USSS9n[        [        U/ SQ5      5        [#        USSS9n[        [        X5      5        g )N)r   r   ri   r   r   T)copyr   rj   r   rk   r   r   r   ri   r`   r   )   )r   r   ri   r   r   r   ri   r   )r   r   r   r   r   )r   r   ri   ri   r   r   r   r   )r&   rO   ro   r   r   _datar
   _mask__array_interface__r'   rk   r   r   r@   rZ   r   r   r?   )rr   nmrw   m3r   y1y1ay2y2ay3r   y4y5y6s                  rY   test_testCopySizeTestMa.test_testCopySize@  s>   aLq\qt$YYq\2"#XX&'ABQ 			--,,- 	. 2Q'B1 11V#$B!$%BQ'		"#A& !AAf$%		"#1%&28!$r
  R#X$4$445AY1B;x("-.72; 89:B1-2/01B"2
r[   c                 d   [        S5      n/ SQn[        U5      nUR                  5       n[        XS9n[	        US   [
        L 5        [	        US   [
        L 5        SS/USS/'   [	        UR                  UL 5        [	        US   [
        L 5        [	        US   [
        L5        [	        [        U/ S	Q5      5        [        XS
S9nUR                  / SQ/ SQ5        [	        UR                  UL5        [	        US   [
        L 5        [	        US   [
        L 5        [	        [        U/ SQ5      5        g )Nr   )r   r   r   ri   ri   rj   r   r   
   (   ri   )r   r   r   r   Tr   )r   ri   r   )r   d      )r   r   r   r   r   )	r   r&   r   r   rO   r'   r   rZ   r=   )rr   dr   r   rw   rs   s         rY   test_testPutTestMa.test_testPuts  s    1IaLVVX!!!H1a&	1!!F"#1()*!4(	i(r!"!!1*+,r[   c                    [        S5      n[        U/ SQS9n[        SS/SS/S9n[        US   [        L5        [        US	   [        L5        X2SS
& [        US   [        L 5        [        US	   [        L5        [        [	        U/ SQ5      5        [        S5      n[        U/ SQS9nUSS
 n[        SS/SS/S9n[        US   [        L5        [        US	   [        L5        X4S S & [        US   [        L 5        [        US   [        L5        [        [	        USS/5      5        [        US   [        L 5        [        US	   [        L5        [        [	        U/ SQ5      5        g )Nr   )r   r   r   r   r   rj   r   r   ri   r   r   r   r   )r   ri   r   r   r   )r   r   rO   r'   rZ   )rr   r   rs   rz   rt   s        rY   test_testPut2TestMa.test_testPut2  sH   1I!/*2r(!Q(!F"#!F"#!A!!F"#1'()1I!/*aF2r(!Q(!F"#!F"#!!!F"#1r2h !!F"#1'()r[   c                     U R                  5       u            pp  n/ SQn[        R                  " U5      S   n[        X%U5        [	        [        [        X%SS9U:H  5      5        g )N)ri   r   r   r   r   r   ri   r   r   ri   r   ri   r   r   )r~   ro   r8   r=   rO   r   rI   )rr   r   ry   r{   r   is         rY   test_testMaPutTestMa.test_testMaPut  s\    ,0,=,=,?)1aAqaQ0JJqM!B2DQ'2-./r[   c                 D   [        S5      nUR                  SS5      nSUR                  S'   [        US   S:H  5        USU-  -   n[        [	        UR
                  U5      5        [        [	        UR                  SU-  5      5        [        [	        U[        U5      -  R
                  SU-  U-  5      5        S	UR                  S
'   [        S5      n[        US'   [        [        US   5      [        [        5      :H  5        US:  n[        [        [        U[        [        5      5      S:H  5        [        [        [        U[        [        5      5      UR                  :H  5        [        X1[        5      n[        UR                  UR                  L 5        [        US   [        L 5        [        US   [        L 5        [        US   [        L 5        [        US   [        L5        [        US   [        L5        [        [	        X5      5        [        U[        U5      n[        UR                  UR                  L 5        [        US   [        L 5        [        US   [        L5        [        US   [        L5        [        US   [        L 5        [        US   [        L 5        [        X15      n[        UR                  UR                  L 5        [        US   [        L 5        [        US   [        L5        [        US   [        L5        [        US   [        L 5        [        US   [        L 5        [        [	        X5      5        [        / SQ5      n[        / SQ5      n[        US'   [        X1U* 5      n[        [	        U/ SQ5      5        [        US'   [        X1U* 5      n[        [	        U/ SQ5      5        [        US   [        L 5        [        US   [        L5        [        US   [        L 5        [        [	        [        [!        US5      U5      [#        US5      5      5        [        [	        [        [%        US5      U5      ['        US5      5      5        [        [	        [        [)        US5      U5      [+        US5      5      5        [        [	        [        [-        US5      U5      [/        US5      5      5        [        [	        [        [1        US5      U5      [3        US5      5      5        [        [	        [        [5        US5      U5      [7        US5      5      5        [        [	        [        [1        US5      U5      [3        US5      5      5        [        [	        [9        [;        [=        S5      5      SS5      / SQ5      5        [        [	        [?        [;        [=        S5      5      SS5      / SQ5      5        [        [	        [9        [        [;        [=        S5      5      / SQS9SS5      R@                  / SQ5      5        [        [	        [?        [        [;        [=        S5      5      / SQS9SS5      R@                  / SQ5      5        [        [	        [7        [        [;        [=        S5      5      / SQS9S5      R@                  / SQ5      5        [        [	        [3        [        / SQ/ SQS9S5      R@                  / SQ5      5        [        [	        [        / SQ/ S Q5      / S!Q5      5        [C        S"[D        RF                  S#9n[I        UR                  [J        5      n[        XT5      n[        [	        XF5      5        [M        X1* U45      n[        [	        U/ SQ5      5        [        US   [        L 5        [        US   [        L5        [        US   [        L 5        [        S$5      n[        US'   [        S$5      S-  n[        US'   [        / S%Q/ S&QS9nURO                  S5      n[        X1U5      n[        XU5      n	[        [	        X)5      5        [        [Q        U	5      [R        L 5        [        [	        U	/ S'Q5      5        [        U[        S5      n[        [	        U/ S(Q5      5        [        US[        5      n[        [	        U/ S)Q5      5        g )*N   r   r   r   )ri   r   y              $@r   e   rg   .r   r   r      r   )r`   ra   rf   rb   rc   )ri   ri   ri   r   r   r   )r`   ra   rg   rh   ri   )r      r  r  r   )r  ri   r   r   r  )ri   r   r   r   r   rj   )ri   ri   ri   ri   r   )r   ri   r   r   r   )ri   ri   r   r   ri   )ri   r   ri   r   r   )r   r   ri   r   ri   )ri   r   ri   r   ri   )ri   ri   r   r   r   )ri   r   r   r   r   )r   r   r   r   r   )r   r   r   )r   r   )ri   ri   ri   r   r   r   )ri   r   r   r   r   r   )r   ri   r      r   2   )r   r   r   ri   ri   ri   )r   ri   ri   r   r   r   )*r   r   flatrO   rZ   realimagr   r'   r   r   rM   rA   r   r3   r   r   r*   r   r+   r"   r-   r#   r.   r9   r/   r   r)   r,   listranger0   rk   r:   ro   float32rN   r   r   r   r   r7   )
rr   rs   rz   catestbtestctestrt   cmr{   s
             rY   test_testOddFeaturesTestMa.test_testOddFeatures  s   2JIIaOq	$2aK166116626"#A	!$**C!GaK89s2J!AaD	S[()FeAvv./145eAvv./177:;!177"#!!!!F"#!F"#1!VQ177"#!!F"#!F"#!!177"#!!F"#!F"#!!1&'/"!!O1+,-!!O1+,-!!F"#!<1q1>!Q3GHI<a 3Q7'1-/ 	0<Q
A.Aq0ABC<
1a 0!46G16MNO<	!Q35Ea5KLM<aQ/a1CDE<	!Q35Ea5KLM=eAhA68MNO>$uQx.!Q79LMN=tE!H~,;"=>?DDHD"$ 	% 	>%U1X-<#>?@!EEIT"$ 	% 	<d58n+:!<=>@@D"$ 	% 	#E//>%@ABDDHD"$ 	% 	<A$& 	'\4ekk8,U*5 !1r1g1+,-!!F"#!1I!1IN!$+=>XXa[!N2!_1	v%&2,-.!VQ1+,-!Q1,-.r[   c                 :   [        [        [        / SQ/ SQ5      / SQ5      5        [        [        [        / SQ/ SQ5      / SQ5      5        [	        S5      n[	        S5      S-
  n[
        US'   [
        US'   [        [        [        X5      [        [        X5      X5      5      5        [        [        [        X5      [        [        X5      X5      5      5        [        [        R                  " U5      S:H  5        [        [        R                  " U5      S	:H  5        g )
N)ri   r   r   )r   r   r   )ri   r   r   )r   r   r   r   r   r   r   r   )
rO   rZ   r5   r4   r   r'   rM   r"   r   r   )rr   rs   rt   s      rY   test_testMinMax2TestMa.test_testMinMax2  s    79i0)<=79i0)<=1I1IM!!71=%Q
A"9:;71=%q"<=>q!Q&'q!Q&'r[   c                 L   [        S5      n[        R                   " S5      n[        USS& UR                  SSS5      nUR                  SSS5      n[	        [        [        R                  " US5      [        US5      5      5        [	        [        [        R                  " USS5      [        USS5      5      5        [	        [        [        R                  " [        US	5      [        US	5      5      [        X5      5      5        [	        [        [        R                  " [        US	5      [        US	5      5      [        X5      5      5        [        / S
Q[        5      n[        US'   [        U/ SQ5      n[	        US	   S:H  5        [	        US   S:H  5        [	        US   S:H  5        g )N   r   r   r   r   r   )r   r   ri   ri   r   )abcri   defr   r   )r   r   r   r  )r   ro   r'   r   rO   rZ   rL   rI   r    r   r;   r   r   )rr   rs   rt   ts       rY    test_testTakeTransposeInnerOuter'TestMa.test_testTakeTransposeInnerOuter  s=   2JIIbM!AIIaAIIaA2<<9-yI/FGH2771i+T!Y-BCD288F1aL&A,7;  	!288F1aL&A,7;  	!)62!I!!	!	r[   c                 b   [        S5      n[        S5      n[        S5      n[        US'   US-  n[        [        X!S-   5      5        US-  n[        [        X!S-   5      5        [        S5      n[        S5      n[        US'   US-  n[        [        X!S-
  5      5        US-  n[        [        X1S-
  5      5        [        S5      S-  n[        S5      S-  n[        US'   US-  n[        [        X!S-  5      5        US-  n[        [        X1S-  5      5        [        S5      S-  n[        S5      n[        US'   US-  n[        [        X!5      5        US-  n[        [        X!5      5        [        S5      S-  n[        S5      S-  n[        US'   US-  n[        [        X!S-  5      5        U[        S5      -  n[        [        U[	        S5      5      5        [        S5      R                  [        R                  5      n[        S5      n[        US'   US-  n[        [        X!S-   5      5        g )Nr   r   ri   r`   ra   )r   )r   r'   rO   rZ   r:   astypero   r  )rr   rt   rs   rx   s       rY   test_testInplaceTestMa.test_testInplace0  s   2J2JBZ1	Q1!e
a1!e2JBZ1	Q1!e
a21u2JBZ#1	S1!e
c	21u2JNBZ1	a1
q12JBZ#1	S1#g
fRj2tE{#$2Jbjj)BZ1	R1"fr[   c                    [        S5      n[        USSS2'   UR                  SS5      n[        S[        R
                  S-   5       HA  n[        R                  " XS9n[        R                  " U5      n[        [        X5      5        MC     g )Nr   r   r   r   r   ri   )protocol)
r   r'   r   r
  pickleHIGHEST_PROTOCOLdumpsloadsrO   rZ   )rr   rs   protor}   rt   s        rY   test_testPickleTestMa.test_testPickleb  sl    2J!Bq&	IIaO1f559:EQ/AQABqH ;r[   c                     [        S5      n[        US'   [        [        [        5      S:H  5        [        US   [        L 5        [	        [        US   S5      S5        g )Nr   ri   z--r   )r   r'   rO   r   rP   r   )rr   xxs     rY   test_testMaskedTestMa.test_testMaskedl  sJ    AY1Ft#$1 VBqE1%q)r[   c           
         [        / SQ/ SQS9n[        [        S[        USS95      5        [        [        S[        U/ SQS95      5        [        U/ SQS	S
9u  p#[        [        SU5      5        [        US:H  5        [        US S & [        [        USS9[        L 5        [        / SQ/ SQS9nUR                  SS5      n[        US S 2S4'   [        [        [        USS9SS/5      5        [        [        USS9S   [        L 5        [        [        SS/[        USS95      5        [        USS	S9u  p#[        [        USS/5      5        g )Nr   r   rj   ra   r   r   )r`   r`   ra   r`   weightsTr0  returnedrb   r   ri   rg   r   r2  r`   )r   rO   rZ   r   r'   r   )rr   r   rX   wtss       rY   test_testAverage1TestMa.test_testAverage1t  s   $<83!,-.3-=>?@c+;dK3 s
A!$./$<8kk!QAqD	73Q'#s45!$Q'612B8WSq123cD93R!"r[   c                 ~	   / SQn/ SQ/ SQ/n[        S5      n[        [        [        USS9S5      5        [        [        [        USUS9S5      5        [	        [        S5      S[        S5      -  /5      n[        [        [        US 5      [
        R                  R                  [
        R                   " S5      5      S	-  S
-  5      5        [        [        [        USS9[
        R                   " S5      S	-  S-  5      5        [        [        [        USS9[        USS9[        USS9S-  /5      5        [        [        [        US US9S5      5        [        [        [        USUS9/ SQ5      5        [        [        [        USS9[        USS9[        USS9S-  /5      5        [        S5      n/ SQn/ SQ/ SQ/n[        S5      n/ SQn	[        [        [        [        X55      SS9S5      5        [        [        [        [        X65      SS9S5      5        [        [        [        X85      SS9[        L 5        [        [        [        X95      SS9S5        [        [        [        [        X85      SS95      S5        [        XG5      n
[        [        [        U
S 5      S5      5        [        [        [        U
SS9/ SQ5      5        [        [        [        U
SS9SS/5      5        [        [        [        U
SUS9/ SQ5      5        [        S5      n[        S5      S-  n[        X/X//SSS9u  p[        [        U5      [        U5      5        [        UR                  UR                  5        [        [        S5      SSS/SS9u  p[        [        U5      [        U5      5        [        [        S5      SS9u  p[        [        U5      [        U5      5        [        [        S5      [        S5      SS9u  p[        [        U5      [        U5      :H  5        [	        SS/SS//[        5      n[        USS/SS//5      n[        USS9n[        [!        USS	/5      5        [        USS9n[        [!        USS	/5      5        [        US S9n[        [!        US 5      5        [        USS9n[        [!        US!S"/5      5        g )#N)r   ri   ri   ri   ri   r   )ri   r   r   r   r   ri   r   r   r   r   )r   r0  ra   rf   g      (@ri   r/  g
@)rg   r`   ra   rf   rb   re   )r   r   ri   ri   r   r   )r   ri   ri   ri   ri   ri   rg   )rg   r`        X@r8  rb   g      @rc   )rg   r`   r8  r8  rb   re   r   Tr3  )r   r   r   )r   r0  r2  )r2  r1  r   r   rl   r`   g@g      ?rb   )r   rO   r	   r   r   ro   r   r   rN   r:   r(   r'   rP   r   rA   floatrZ   )rr   w1w2rs   rt   rv   rw   r   m4m5rz   abr1r2r   r   a2daa2dmas                      rY   test_testAverage2TestMa.test_testAverage2  s    "451I+S12B7=>6!9cF1Io./D)!#ryy|!<r!AC!GI 	J+RYYq\B->-CDE+")!!"4gaa6H36N!OQ 	RD"5x@AB7!:< 	=+")!!"4gaa6H36N!OQ 	R1X "45!Wa!41=sCDa!41=sCDQ+!4>?W\!0q93?U7<#6Q?@!DD)845+!=? 	@+c3Z89B7!>@ 	A 1I1IM1&1&)DAU2Yb	*RXXrxx(iq1a&4PU2Yb	*i48U2Yb	*i$y/DQb	U2Y&'aVaV$e,C1a&1a&!12s#4#s$%1%53*%&4(5'"#1%53*%&r[   c           	         [        S[        [        S5      5      5        [        S[        [        S5      5      5        [        S[        [        S///5      5      5        [        S[        [        S//5      5      5        [	        [
        [        [        SS/5      5        [	        [        [        [        SS/5      5        [	        [        [        [        SS/SS/S95        g )Nri   r`   r   rj   )rP   intr   r9  rQ   	TypeError
ValueErrorboolrr   s    rY   test_testToPythonTestMa.test_testToPython  s    QE!H&S%a/*QEQC5'N+,S%se-.iq!f6j$q!f6j$q!fAq6(BCr[   c                    [        SSS9n[        R                  " SS9   [        S[        S5      -  R                  5        S S S 5        [        SU-   R                  5        [        U* R                  5        [        U* R                  5        [        [        X5      R                  5        [        [        X5      R                  5        [        UR                  5       R                  UR                  R                  L 5        [        SSS9n[        UR                  5       UR                  :H  5        [        [        U5      [        [        5      5        g ! , (       d  f       GN= f)Nr   ri   rj   r   r   )r   ro   r   rO   rk   r4   r5   r   r   r   rP   r   r1   )rr   rx   rs   s      rY   test_testScalarArithmetic TestMa.test_testScalarArithmetic  s    11[[)Qq\''( *R"

"

$$%$$%		!!RXX^^34!!
agg%&SWc"567 *)s   "E
E$c                    [        / SQ5      n[        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  SSSSS5      UR                  R                  SSSSS5      5      5        [        [        UR                  / SQ5      UR                  R                  / SQ5      5      5        [        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  5       UR                  R                  5       5      5        [        SS/SS//5      n[        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  5       UR                  R                  5       5      5        [        [        UR                  SS/5      UR                  R                  SS/5      5      5        [        [        UR                  5       UR                  R                  5       5      5        g )Nri   r   r   r   ri   r   r   r   ri   r   ri   )r   rO   rZ   anyr   r   argmaxargminr   compressconjr   diagonalrH   rI   rL   )rr   r>  r   s      rY   test_testArrayMethodsTestMa.test_testArrayMethods  s   )1557AGGKKM*+1557AGGKKM*+188:qww~~/01188:qww~~/01188Aq!Q*77>>!Q1a8: 	;1::i(!''*:*:9*EFG1668QWW\\^,-1;;=!''"3"3"567Aq6Aq6"#1::<!1!1!3451557AGGKKM*+1661a&>177<<A#7891;;=!''"3"3"567r[   c                 J    [        / SQ5      n[        UR                  S5        g )NrS  ri   )r   rP   ndimrr   r>  s     rY   test_testArrayAttributesTestMa.test_testArrayAttributes  s    )QVVQr[   c           	          [        [        [        R                  5       Vs/ s H2  nU[        [        5      ;  d  M  UR                  S5      (       a  M0  UPM4     sn(       + 5        g s  snf )Nr   )rO   dirro   r   r   
startswith)rr   r   s     rY   test_testAPITestMa.test_testAPI  sR    BJJ ,1[!11 c*  , , 	- ,s   A'A'A'c                     [        / SQ5      n[        / SQ/ SQS9n[        US   R                  S5        [        US   R                  S5        [        US   R                  S5        g )NrS  rT  rj   r   r   ri   )r   rP   rA   )rr   r>  r?  s      rY   test_testSingleElementSubscript&TestMa.test_testSingleElementSubscript  sN    ))),QqTZZ$QqTZZ$QqTZZ$r[   c                 Z    [        / SQ/ SQS9nUS:  nSX'   [        US   [        L 5        g )Nr   )ri   r   ri   r   rj   r   r   r   )r   rO   r'   )rr   r>  r  s      rY   test_assignment_by_condition#TestMa.test_assignment_by_condition  s.    ,\2F!r[   c                     [        SS/SS/S9n[        SS/SS/S9nUS:  nX   X#'   SS/n[        UR                  U5        g )Nr   ri   Frj   T)r(   rP   rk   )rr   r>  r?  rk   expected_masks        rY   test_assignment_by_condition_2%TestMa.test_assignment_by_condition_2  sS    !Quen5!QtTl31u'QVV]+r[   r   N)&__name__
__module____qualname____firstlineno__r~   r   pytestmarkparametrizer   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r(  r,  r5  rD  rL  rP  r[  r`  re  rh  rk  ro  __static_attributes__r   r[   rY   r]   r]   b   s    7 [[S66"23 4&$=L2H>+26*0'd1f-,*40c/J((0d*#&8'tD88" -
% ,r[   r]   c                   2    \ rS rSrS rS rS rS rS rSr	g)	
TestUfuncsi  c                     [        SSS[        S-  /S-  SS/S/S-  -   S9[        SSS[        S-  /S-  SS/S/S-  -   S94$ )Nr`   r   r   r   ri   r   rj   )r   rp   rK  s    rY   r~   TestUfuncs._create_data  sd    sAr26*Q.aVqcAg5EFaR!V,q01va7GHK 	Kr[   c           	         / SQnS GH  n [        [        U5      n[        [        R
                  U5      nU R                  5       S UR                   n[        R                  " 5          X!;   a  [        R                  " SS9  US;   a  [        R                  " SS9  U" U6 nU" U6 nS S S 5        [        [        WR                  S5      WR                  S5      U5      5        [        [        UR                  UR                  5      5        GM     g ! [         a    [        [        U5      n GNf = f! , (       d  f       N= f)N)rF   arctanhr   r   arccoshr~  r$   r%   r   true_dividefloor_divide	remainderfmod)*rF   r$   r%   r   r   rB   r   rJ   r   r   r   rC   r   rK   arcsinhr  r~  r   fabsnegativefloorceillogical_notr   rG   r6   r   r  r  r  r  hypotr   r   r9   r#   r   r"   r   logical_and
logical_orlogical_xorr   )r   )r~  r$   r%   rO  r   )getattrumathAttributeErrorfromnumericro   r   r~   ninr   seterrrO   rZ   r   eqmaskrk   )rr   f_invalid_ignorefufmfargsurmrs           rY   test_testUfuncRegression#TestUfuncs.test_testUfuncRegression  s    @>A -UA& "B$$&w/D(IIh/33IIX.YY  Bryy|RYYq\156F277BGG,-=>$ " -[!,- s   D(>D2D/.D/2
E 	c                     U R                  5       S   n[        [        USS9(       + 5        [        [        USS95        [	        [        US S SS9S5        [	        [        USS9S5        g )Nr   r   r   )r~   rO   r   rD   rP   rH   r<   r_  s     rY   test_reduceTestUfuncs.test_reduce6  s^    "GAA&&'#$S2AQ'+WQQ'+r[   c                 H   [        SS5      R                  SS5      n[        US:  U5      n[        UR	                  5       UR	                  5       5        [        UR                  5       S5        [        UR	                  S5      UR	                  S5      :H  R                  5       5        [        UR                  S5      / SQ:H  R                  5       5        [        UR	                  S5      S   R                  5        [        UR                  S5      S   R                  5        g )Nri      r   r   r   r   )r   r   r  r   )	r   r   r3   rP   r   r   rO   r   rk   )rr   r>  amasks      rY   test_minmaxTestUfuncs.test_minmax=  s    1bM!!!Q'QUA&UYY[!%%'*UYY[!$1q)..011-2245		!Q$$%		!Q$$%r[   c                 p    S H0  n[        / SQ/ SQS9n[        [        [        U5      S/5      5        M2     g )Nz?bhilqpBHILQPfdgFDGO)ri   r   r   r   )r   r   ri   ri   rj   r   )r   rO   rZ   r8   )rr   r  rs   s      rY   test_nonzeroTestUfuncs.test_nonzeroG  s-    'Al6ABwqzA3'( (r[   r   N)
rq  rr  rs  rt  r~   r  r  r  r  rx  r   r[   rY   rz  rz    s    K#.J,&)r[   rz  c                   D    \ rS rSrS rS rS rS rS rS r	S r
S	 rS
rg)TestArrayMethodsiM  c                 T   [         R                  " / SQ5      nUR                  SS5      nUR                  SSSS5      n[         R                  " / SQ5      n[        XS9n[        X$R                  UR                  5      S9n[        X4R                  UR                  5      S9nXX4XVU4$ )N)$g      @gGz.@g-!@g      !@g&1?gʡE@g\( @gQ@g{G#@gPn@gh|?!@g!rh@gjt@gJ+@gzGa@g!rh@gʡE@g-@g)\(@g(\B#@g+@guV@g~j@g"~@gK7@g
ףp=J#@gGz@gMb@g}?5^#@g/$@g!rhm@g-#@gJ+@gMb@gʡ@g-?r   r   r   )$r   ri   r   ri   r   r   ri   r   ri   ri   r   ri   r   r   r   ri   r   ri   r   r   r   ri   ri   ri   ri   r   r   ri   r   r   r   r   ri   r   ri   r   )datark   )ro   r   r   rA   )rr   rs   XXXr   mxmXmXXs           rY   r~   TestArrayMethods._create_dataO  s    HH @ A IIaOYYq!Q"HH ( ) "		!'' 23))BHH"56RBC''r[   c                    U R                  5       u  p    pnUR                  5       n[        UR                  5       UR                  5       R	                  5       R                  5       5        [        [        UR                  5       UR                  5       [        UR                  UR                  5       -  SS9-
  5      5        g )Nr   r   )	r~   rZ  rP   trace
compressedrH   rO   rZ   rk   )rr   r   r  r  mXdiags        rY   
test_traceTestArrayMethods.test_tracee  s    #002aA1RXXZ!9!9!;!?!?!AB288:779s6;;+E01(3 34 	5r[   c           	      x   U R                  5       u  n    p#  nUR                  SS5      n[        [        UR                  UR                  5      5        [        [        UR
                  UR                  SS5      5      5        [        [        UR
                  UR
                  R                  SS5      5      5        g )Nr   r   )r~   cliprO   rZ   rk   r   )rr   rs   r   r  clippeds        rY   	test_clipTestArrayMethods.test_clipm  s    #0021a1''!Q-7<<)*7==!&&A,/07=="((--1"567r[   c                    U R                  5       u  pppEnUR                  u  pc[        UR                  5       [        R                  " UR                  5       5      5        [        R                  " U[        R                  5      n[        R                  " U[        R                  5      n[        U5       H0  n	[        R                  " US S 2U	4   R                  5       5      X'   M2     [        U5       H+  n	[        R                  " XY   R                  5       5      Xy'   M-     [        [        UR                  S5      U5      5        [        [        UR                  S5      U5      5        g )Nr   ri   )r~   rA   rP   ptpro   r  rN   float64r
  rO   rZ   )
rr   r   r  r   r  r  r   rowscolsks
             rY   test_ptpTestArrayMethods.test_ptpt  s     $ 1 1 3aBAww 	RVVXrvvbmmo67xx2::&xx2::&qAffR1X0023DG qAffRU--/0DG 266!9d#$266!9d#$r[   c                     U R                  5       u          pnUR                  SS5      n[        [        US   US S 2S4   5      5        UR                  SS5      n[	        UR
                  S5        g )Nr   ri   r   r   )r   r   r   r   )r~   swapaxesrO   rZ   rP   rA   )rr   r   r  r  	mXswapped
mXXswappeds         rY   test_swapaxesTestArrayMethods.test_swapaxes  sh    !%!2!2!41aA3KK1%	9R="QU),-\\!Q'
Z%%|4r[   c                 b   U R                  5       S   nUR                  S5      n[        [        UR                  UR                  S5      R                  S5      5      5        UR                  S5      n[        [        UR                  UR                  S5      R                  S5      5      5        g Nr   r   ri   )r~   cumprodrO   rZ   r   r   rr   r  mXcps      rY   test_cumprodTestArrayMethods.test_cumprod  s{     #zz!}4::ryy|33A678zz!}4::ryy|33A678r[   c                 b   U R                  5       S   nUR                  S5      n[        [        UR                  UR                  S5      R                  S5      5      5        UR                  S5      n[        [        UR                  UR                  S5      R                  S5      5      5        g r  )r~   cumsumrO   rZ   r   r   r  s      rY   test_cumsumTestArrayMethods.test_cumsum  s{     #yy|4::ryy|221567yy|4::ryy|221567r[   c           	         U R                  5       u  pn  pn[        [        UR                  S S9UR	                  5       R                  5       5      5        [        [        UR                  S S9UR	                  5       R                  5       5      5        [        [        UR                  SS9R                  UR                  SS9R                  5      5        [        [        UR                  5       R                  UR                  5       R                  5      5        UR                  SS9UR                  SS9pv[        S5       H  n[        [        Xx   XH   R	                  5       R                  5       5      5        [        [        Xh   US S 2U4   R	                  5       R                  5       5      5        [        [        [        R                  " Xh   5      US S 2U4   R	                  5       R                  5       5      5        M     g )Nr   r   r   ri   r   )
r~   rO   rZ   varr  stdrA   r
  ro   rF   )	rr   r   r  r  r  r  mXvar0mXvar1r  s	            rY   test_varstdTestArrayMethods.test_varstd  sq   "&"3"3"5b!QC266t6$bmmo&9&9&;<=266t6$bmmo&9&9&;<=3777?(("&&a&.*>*>?@2668>>1557==12FFFNBFFFNqABvy"%"2"2"4"8"8":;<Bvy"QT("5"5"7";";"=>?Brwwvy)!!Q$x22488:< = r[   r   N)rq  rr  rs  rt  r~   r  r  r  r  r  r  r  rx  r   r[   rY   r  r  M  s*    (,58%598=r[   r  c                 j    U [         L a  U[         L $ U[         L a  U [         L $ X:H  R                  5       $ r   )r7   r   )rv   rw   s     rY   r  r    s3    	V|V|	V|V|H>>r[   ) )ar#  	functoolsr   ru  numpyro   numpy._core.fromnumeric_corer  numpy._core.umathr  numpy.mar   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/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   numpy.testingrO   rP   rQ   rp   rZ   r]   rz  r  r  r   r[   rY   <module>r     s        - - ! !L L L L L L L L L L L L L L L L L L L LZ ? >UUf
, f
,R?) ?)DV= V=rr[   