
    Kh                        S SK Jr  S SK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  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/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;J<r<J=r=J>r>J?r?J@r@JArAJBrBJCrCJDrDJErEJFrFJGrGJHrHJIrIJJrJJKrKJLrLJMrMJNrNJOrOJPrPJQrQJRrRJSrSJTrTJUrUJVrVJWrWJXrXJYrYJZrZ  \R                  r[SS jr\ " S S5      r] " S S	5      r^ " S
 S5      r_S r`g)    )reduceN)assert_assert_raisesassert_equal)KMaskTypeMaskedArray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zerosc                 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       L/var/www/html/env/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$)%TestMa$   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U l        g )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   pirP   rD   set_fill_valuedselfxya10m1m2xmymzzmxfss               rY   setup_methodTestMa.setup_method&   s    HHb"b#r#vr2tS"b"MNHHJK1111HH&'1<(XXb%#GG
% :r[   c                 B   U R                   u  pp4pVpxpn[        [        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        [        [        Xj5      5        [        [        [        US5      U
5      5        [        [        X5      5        g )Nc                 
    X-  $ N ru   rv   s     rY   <lambda>)TestMa.test_testBasic1d.<locals>.<lambda>=   s    r[   c                 
    X-   $ r   r   r   s     rY   r   r   >   s    QUr[   rn   )rr   r   r$   r   rD   dtypesizer   r   lenrZ   r   rs   s               rY   test_testBasic1dTestMa.test_testBasic1d5   s    48FF1srqM!$$%b!"U2Y"RXXq!RXXqww'RWWf%6:;U2YB&1BB*G GH2
6"e$b)*1	r[   r         )      c                 >   U R                   u  p#pEpgppn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        [        [        X{5      5        [        [        [        US5      U5      5        [        [        X'5      5        g )Nc                 
    X-  $ r   r   r   s     rY   r   )TestMa.test_testBasic2d.<locals>.<lambda>Q   s    !%r[   c                 
    X-   $ r   r   r   s     rY   r   r   R   s    aer[   rn   )rr   rD   r   r$   r   r   r   r   r   rZ   r   )rt   r   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   s               rY   test_testBasic2dTestMa.test_testBasic2dC   s     59FF1srqM!$$%b!"U2Y"RXXq!RWWf%7;<U2YB&1CR*H HI2
6"e$b)*1	r[   c                    U R                   u  pp4pVpxpn[        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-   Xg-   5      5        [        [	        X-
  Xg-
  5      5        [        [	        X-  Xg-  5      5        [        R                  " SSS9   [        [	        X-  Xg-  5      5        S S S 5        [        [	        X2-   X7-   5      5        [        [	        X2-
  X7-
  5      5        [        [	        X2-  X7-  5      5        [        R                  " SSS9   [        [	        X2-  X7-  5      5        S S S 5        [        [	        X-   Xc-   5      5        [        [	        X-
  Xc-
  5      5        [        [	        X-  Xc-  5      5        [        [	        X-  Xc-  5      5        [        [	        US-  US-  5      5        [        [	        [        U5      S-  [        U5      S-  5      5        [        [	        X-  Xg-  5      5        [        [	        [        R                  " X5      [        Xg5      5      5        [        [	        [        R                  " X5      [        Xg5      5      5        [        [	        [        R                  " X5      [        Xg5      5      5        [        R                  " SSS9   [        [	        [        R                  " X5      [        Xg5      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      @)rr   r   r+   r   rZ   reshapero   errstateabsr
   rJ   r9   r   )rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   a2da2dms                 rY   test_testArithmeticTestMa.test_testArithmeticW   s   48FF1srq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)N;(3O)
N8	;
O
	
O	c                     [         R                  " S/5      n[        S/5      n[        [        X-   [        5      5        [        [        X!-   [        5      5        g )Nri   )ro   r   r   
isinstancer   )rt   namas      rY   test_testMixedArithmeticTestMa.test_testMixedArithmetic}   s>    XXqc]A3Z
27K01
27K01r[   c                 
   U R                   u  pp4pVpxpn[        [        [        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      ['        Xg5      5      5        [        [        [        R(                  " U5      [)        U5      5      5        [        [        [        R*                  " X5      [+        Xg5      5      5        [        [        [        R,                  " X5      [-        Xg5      5      5        [        [        [        R.                  " X5      [/        Xg5      5      5        [        [        [        R0                  " X5      [1        Xg5      5      5        [        [        [        R2                  " X5      [3        Xg5      5      5        [        [        [        R4                  " X5      [5        Xg5      5      5        [        [        [        R6                  " U5      [7        U5      5      5        [        [        [        R8                  " X45      [9        Xg4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   )rr   r   rZ   ro   r   r   rE   rF   rM   rN   r   rI   r   r'   r(   r   r   r   r   r   r	   r   r<   r%   r!   r&   r"   r   r   rs   s               rY   test_testUfuncs1TestMa.test_testUfuncs1   s'   48FF1srq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   B2U,,
U;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   r   r   r   typero   intpr   rZ   r   r   ndarrayr    r:   )rt   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                 4   U R                   u  pp4pVpxpn[        R                  " U5      n[        U5      n[        [	        [        U5      [        R                  " U5      5      5        [        [	        [        U5      [        R                  " U5      5      5        g r   )
rr   ro   rA   r   rZ   maxr7   r   minr8   )rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   xrxmrs                 rY   test_testMinMaxTestMa.test_testMinMax   sh    48FF1srqXXa[Bi 	3r7GNN3/013r7GNN3/01r[   c                 >   U R                   u  pp4pVpxpn[        [        [        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      [        Xg4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   )rr   r   rZ   ro   r
   r   
accumulaterK   r   r   prodr?   r   r   rs   s               rY   test_testAddSumProdTestMa.test_testAddSumProd   s	   48FF1srq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reprr   rZ   rH   r   r*   r   r    r+   r   r5   r   rk   r   objectr   rD   )rt   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)   r   ro   r   r   _datar   _mask__array_interface__r*   rk   r   r   rC   rZ   r   r    rB   )rt   nmry   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   r   r*   r   rZ   r@   )rt   rr   r   r   ry   ru   s         rY   test_testPutTestMa.test_testPut5  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   r   r*   rZ   )rt   rr   ru   r|   rv   s        rY   test_testPut2TestMa.test_testPut2K  sH   1I!/*2r(!Q(!F"#!F"#!A!!F"#1'()1I!/*aF2r(!Q(!F"#!F"#!!!F"#1r2h !!F"#1'()r[   c                     U R                   u  pp4pVpxpn/ 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   )rr   ro   r;   r@   r   r   rL   )rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   is                 rY   test_testMaPutTestMa.test_testMaPute  sQ    48FF1srq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   flatr   rZ   realimagr   r*   r   r   rP   rD   r   r6   r   r!   r-   r"   r.   r%   r0   r&   r1   r<   r2   r   r,   r/   listranger3   rk   r=   ro   float32rQ   r   r   r   r    r:   )
rt   ru   r|   catestbtestctestrv   cmr}   s
             rY   test_testOddFeaturesTestMa.test_testOddFeaturesl  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   )
r   rZ   r8   r7   r   r*   rP   r%   r!   r   )rt   ru   rv   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   r   rZ   rO   rL   r#   r   r>   r   r   )rt   ru   rv   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*   r   rZ   r=   astypero   r	  )rt   rv   ru   rz   s       rY   test_testInplaceTestMa.test_testInplace  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loadsr   rZ   )rt   ru   protor   rv   s        rY   test_testPickleTestMa.test_testPickle$  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*   r   r   r   r   )rt   xxs     rY   test_testMaskedTestMa.test_testMasked.  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r.  returnedrb   r   ri   rg   r   r0  r`   )r   r   rZ   r   r*   r   )rt   r   rX   wtss       rY   test_testAverage1TestMa.test_testAverage16  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   r.  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@r6  rb   g      @rc   )rg   r`   r6  r6  rb   re   r   Tr1  )r   r   r   )r   r.  r0  )r0  r/  r   r   rl   r`   g@g      ?rb   )r   r   r   r   r   ro   r
   r   rQ   r=   r+   r*   r   r   rD   floatrZ   )rt   w1w2ru   rv   rx   ry   r   m4m5r|   abr1r2r   r   a2daa2dmas                      rY   test_testAverage2TestMa.test_testAverage2I  s    "451I+S12B7=>6!9cF1Io./D)!#ryy|!<r!AC!GI 	J+RYYq\B->-CDE+")!!"4gaa6H6L!MO 	PD"5x@AB7!:< 	=+")!!"4gaa6H6L!MO 	P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   )r   intr   r7  r   	TypeError
ValueErrorboolrt   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   r   rk   r7   r8   r   r   r   r   r   r4   )rt   rz   ru   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   r   rZ   anyr   r   argmaxargminr   compressconjr   diagonalrK   rL   rO   )rt   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 )NrQ  ri   )r   r   ndimrt   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 )N_)r   dirro   r   r   
startswith)rt   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 )NrQ  rR  rj   r   r   ri   )r   r   rD   )rt   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   r   r*   )rt   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+   r   rk   )rt   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[   rr   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*  r3  rB  rJ  rN  rY  r^  rd  rg  rj  rn  __static_attributes__r   r[   rY   r]   r]   $   s    ; [[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U l        g )Nr`   r   r   r   ri   r   rj   )r   rp   rr   rI  s    rY   r   TestUfuncs.setup_method  sb    aR!V,q01va7GHaR!V,q01va7GHKr[   c           	      ~   / SQnS GH   n [        [        U5      n[        [        R
                  U5      nU R                  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)rI   arctanhr   r   arccoshr~  r'   r(   r   true_dividefloor_divide	remainderfmod)*rI   r'   r(   r   r   rE   r   rM   r   r   r   rF   r   rN   arcsinhr  r~  r	   fabsnegativefloorceillogical_notr
   rJ   r9   r   r  r  r  r  hypotr   r   r<   r&   r"   r%   r!   logical_and
logical_orlogical_xorr   )r   )r~  r'   r(   rM  r   )getattrumathAttributeErrorfromnumericro   r   rr   ninr   seterrr   rZ   r   eqmaskrk   )rt   f_invalid_ignorefufmfargsurmrs           rY   test_testUfuncRegression#TestUfuncs.test_testUfuncRegression  s    @>A -UA& "B66'266?D(IIh/33IIX.YY  Bryy|RYYq\156F277BGG,-=>$ " -[!,- s   D$>D.D+*D+.
D<	c                     U R                   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   )rr   r   r   rG   r   rK   r?   r]  s     rY   test_reduceTestUfuncs.test_reduce  sX    FF1I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   r6   r   r   r   r   r   rk   )rt   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   r   rZ   r;   )rt   r  ru   s      rY   test_nonzeroTestUfuncs.test_nonzero  s-    'Al6ABwqzA3'( (r[   rp  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)TestArrayMethodsi  c                 `   [         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U l        g )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   rD   rr   )rt   ru   XXXr   mxmXmXXs           rY   r   TestArrayMethods.setup_method  s    HH @ A IIaOYYq!Q"HH ( ) "		!'' 23))BHH"56rs+r[   c                 |   U R                   u  pp4pVn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   )	rr   rX  r   trace
compressedrK   r   rZ   rk   )	rt   ru   r  r  r   r  r  r  mXdiags	            rY   
test_traceTestArrayMethods.test_trace&  s    &*ff#rbcRXXZ!9!9!;!?!?!AB288:779s6;;+E01(3 34 	5r[   c           	      l   U R                   u  pp4pVn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   )rr   clipr   rZ   rk   r   )	rt   ru   r  r  r   r  r  r  clippeds	            rY   	test_clipTestArrayMethods.test_clip.  sy    &*ff#rbc''!Q-7<<)*7==!&&A,/07=="((--1"567r[   c                    U R                   u  pp4pVnUR                  u  p[        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                  " Xk   R                  5       5      X'   M-     [        [        UR                  S5      U
5      5        [        [        UR                  S5      U	5      5        g Nr   ri   )rr   rD   r   ptpro   r  rQ   float64r  r   rZ   )rt   ru   r  r  r   r  r  r  r   rowscolsks               rY   test_ptpTestArrayMethods.test_ptp5  s    &*ff#rbc 	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                   u  pp4pVn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   )rr   swapaxesr   rZ   r   rD   )
rt   ru   r  r  r   r  r  r  	mXswapped
mXXswappeds
             rY   test_swapaxesTestArrayMethods.test_swapaxesD  s_    &*ff#rbcKK1%	9R="QU),-\\!Q'
Z%%|4r[   c                 ^   U R                   u  pp4pVn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  )rr   cumprodr   rZ   r   r   	rt   ru   r  r  r   r  r  r  mXcps	            rY   test_cumprodTestArrayMethods.test_cumprodK  sz    &*ff#rbczz!}4::ryy|33A678zz!}4::ryy|33A678r[   c                 ^   U R                   u  pp4pVn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  )rr   cumsumr   rZ   r   r   r  s	            rY   test_cumsumTestArrayMethods.test_cumsumR  sz    &*ff#rbcyy|4::ryy|221567yy|4::ryy|221567r[   c           	         U R                   u  pp4pVn[        [        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p[        S5       H  n
[        [        X   Xj   R	                  5       R                  5       5      5        [        [        X   US S 2U
4   R	                  5       R                  5       5      5        [        [        [        R                  " X   5      US S 2U
4   R	                  5       R                  5       5      5        M     g )Nr   r   r   ri   r   )
rr   r   rZ   varr  stdrD   r  ro   rI   )rt   ru   r  r  r   r  r  r  mXvar0mXvar1r  s              rY   test_varstdTestArrayMethods.test_varstdY  sj   &*ff#rbc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[   rp  N)rq  rr  rs  rt  r   r  r  r  r  r  r  r  rx  r   r[   rY   r  r    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   )r:   r   )rx   ry   s     rY   r  r  g  s3    	V|V|	V|V|H>>r[   ) )a	functoolsr   r!  ru  numpyro   numpy._core.umath_corer  numpy._core.fromnumericr  numpy.testingr   r   r   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/   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   rO   rP   rQ   rp   rZ   r]   rz  r  r  r   r[   rY   <module>r     s        ! ! - -                     UUf
, f
,R>) >)BV= V=rr[   