ó
    �œ‘h1: ã                   óP  • S r SSKrSSKrSSKrSSKrSSK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  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/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7  SSK8J9r9J:r:J;r;J<r<  SSK=J>r>J?r?   " S S	5      r@ " S
 S5      rA " S S5      rB " S S5      rC " S S5      rD " S S5      rE " S S5      rF " S S5      rG " S S5      rH " S S5      rI " S S5      rJ " S S5      rK " S  S!5      rL " S" S#5      rM " S$ S%5      rNg)&z�Tests suite for MaskedArray.
Adapted from the original test_ma by Pierre Gerard-Marchant

:author: Pierre Gerard-Marchant
:contact: pierregm_at_uga_dot_edu

é    N)Únormalize_axis_tuple)ÚMaskedArrayÚarangeÚarrayÚcountÚgetmaskarrayÚmaskedÚmasked_arrayÚnomaskÚonesÚshapeÚzeros)#Ú
_covhelperÚapply_along_axisÚapply_over_axesÚ
atleast_1dÚ
atleast_2dÚ
atleast_3dÚaverageÚclump_maskedÚclump_unmaskedÚcompress_ndÚcompress_rowcolsÚcorrcoefÚcovÚdiagflatÚdotÚediff1dÚflatnotmasked_contiguousÚin1dÚintersect1dÚisinÚmask_rowcolsÚ
masked_allÚmasked_all_likeÚmedianÚmr_ÚndenumerateÚnotmasked_contiguousÚnotmasked_edgesÚpolyfitÚ	setdiff1dÚsetxor1dÚstackÚunion1dÚuniqueÚvstack)Úassert_Úassert_almost_equalÚassert_array_equalÚassert_equal)Úassert_warnsÚsuppress_warningsc                   ó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)ÚTestGenericéJ   c                 óF  • [        S[        S9n[        SS/SS/[        S9n[        X5        [        R
                  " SS/SS/S.5      n[        SUS9n[        S	S	/S
S
/US9n[        X5        [        SUS9n[        S	S	/S	S	//S
S
/S
S
//US9n[        X5        [        R
                  " SSSS/4/5      n[        SUS9n[        SS/SS/US9n[        X5        [        SUS9n[        SS/SS/US9n[        X5        [        S
US9n[        S//S//US9n[        X5        g )N©é   ©Údtypeé   ©Úmaskr?   ÚaÚbÚf©ÚnamesÚformats©r   r   ©r@   r@   ©r=   r=   ©rC   rE   ©ÚbarE   ©ÚbbrE   ©r@   rJ   )r$   Úfloatr   r5   Únpr?   )ÚselfÚtestÚcontrolÚdts       Ú^C:\Users\psant\Downloads\Noiseback\school\venv\Lib\site-packages\numpy/ma/tests/test_extras.pyÚtest_masked_allÚTestGeneric.test_masked_allL   sb  € ô ˜$¤eÑ,ˆÜ˜˜A˜ a¨ V´5Ñ9ˆÜ�TÔ#ä�XŠX  c 
¸¸S°zÑBÓCˆÜ˜$ bÑ)ˆÜ˜ Ð(°¸Ð/?ÀrÑJˆÜ�TÔ#Ü˜&¨Ñ+ˆÜ˜& &Ð)¨F°FÐ+;Ð<Ø% vÐ.°¸Ð0@ÐAØ ñ"ˆô 	�TÔ#ä�XŠX�z C¨+°{Ð)CÐ#DÐEÓFˆÜ˜$ bÑ)ˆÜ˜ kÐ2Ø)¨;Ð7¸rñCˆä�TÔ#Ü˜$ bÑ)ˆÜ˜ kÐ2Ø)¨;Ð7¸rñCˆä�TÔ#Ü˜&¨Ñ+ˆÜ˜+˜˜°¨}¨oÀRÑHˆÜ�TÕ#ó    c                 ó0  • [         R                  " SS[        4/S44/5      n[         R                  R	                  SU5      n[        [        US   5      [         R                  R                  R                  5        [        [        US   S   5      [         R                  R                  R                  5        [        [        US   S   5      S5        [        US   S   R                  S5        [        US   S   R                  R                  S5        g )NrD   Úc©r@   r@   rJ   © ©rS   r?   ÚobjectÚmar$   r5   ÚtypeÚcorer   Úlenr   Ú_fill_value©rT   Úmy_dtypeÚ
masked_arrs      rX   Ú"test_masked_all_with_object_nestedÚ.TestGeneric.test_masked_all_with_object_nestedj   sÒ   € ô —8’8˜c c¬6 ] O°TÐ#:Ð;Ð<Ó=ˆÜ—U‘U×%Ñ% d¨HÓ5ˆ
ä”T˜* S™/Ó*¬B¯E©E¯J©J×,BÑ,BÔCÜ”T˜* S™/¨#Ñ.Ó/´·±·±×1GÑ1GÔHÜ”S˜ C™¨Ñ-Ó.°Ô2Ü�Z ‘_ SÑ)×/Ñ/°Ô8Ü�Z ‘_ SÑ)×5Ñ5×;Ñ;¸RÕ@r[   c                 óž  • [         R                  " S[        S44/5      n[         R                  R	                  SU5      n[        [        US   5      [         R                  R                  R                  5        [        [        US   5      S5        [        US   R                  S5        [        US   R                  R                  S5        g )NrD   r^   r@   rJ   r_   r`   rg   s      rX   Útest_masked_all_with_objectÚ'TestGeneric.test_masked_all_with_objectv   s”   € ä—8’8˜c¤F¨D >Ð2Ð3Ó4ˆÜ—U‘U×%Ñ% d¨HÓ5ˆ
ä”T˜* S™/Ó*¬B¯E©E¯J©J×,BÑ,BÔCÜ”S˜ C™Ó)¨1Ô-Ü�Z ‘_×*Ñ*¨FÔ3Ü�Z ‘_×0Ñ0×6Ñ6¸Õ;r[   c                 óš  • [        SS/[        S9n[        U5      n[        SS/SS/[        S9n[        X#5        [        R
                  " SS/SS/S.5      n[        S	S	/S
S
/US9n[        U5      n[        SS/S
S
/US9n[        X#5        [        R
                  " SSSS/4/5      n[        SS/SS/US9n[        U5      n[        X#5        g )Nr@   r=   r>   rA   rC   rD   rE   rF   rI   rJ   )é
   rp   rL   rM   rO   rQ   )r   rR   r%   r5   rS   r?   )rT   ÚbaserU   rV   rW   s        rX   Útest_masked_all_likeÚ TestGeneric.test_masked_all_like€   sä   € ô �a˜�V¤5Ñ)ˆÜ˜tÓ$ˆÜ˜˜A˜ a¨ V´5Ñ9ˆÜ�TÔ#ä�XŠX  c 
¸¸S°zÑBÓCˆÜ�f˜fÐ%¨V°VÐ,<ÀBÑGˆÜ˜tÓ$ˆÜ˜ 8Ð,°F¸FÐ3CÈ2ÑNˆÜ�TÔ#ä�XŠX�z C¨+°{Ð)CÐ#DÐEÓFˆÜ˜ kÐ2Ø)¨;Ð7¸rñCˆä˜wÓ'ˆÜ�TÕ#r[   c                 ó4  • [        SS5       GH  n[        SU-  5       Hñ  n[        R                  " U[        S9n[        R                  " X#[        S9n[        SU-  5      nUSU-  -  S:g  Ul        SnU" U5       H"  nXvR                  U   R                  5       -  nM$     U[        :X  a*  [        UR                  5       R                  5       U5        M·  UR                  ) Ul        [        UR                  5       R                  5       U5        Mó     GM
     g )Nr@   é   r=   r>   r   )ÚrangerS   r   ÚintÚfullr
   rB   ÚdataÚsumr   r5   Ú
compressed)	rT   rE   ÚiÚjÚkÚjarC   ÚsÚsls	            rX   Úcheck_clumpÚTestGeneric.check_clump”   s×   € Ü�q˜!—ˆAÜ˜1˜a™4–[�Ü—I’I˜a¤sÑ+�Ü—W’W˜Q¬Ñ-�Ü   A¡Ó&�Ø  1¡™+¨!Ñ+�”Ø�Ù˜Až$�BØŸ™ ™Ÿ™Ó)Ñ)’Añ àœÓ&Ü  §¡£×!3Ñ!3Ó!5°qÖ9àŸf™f˜W�A”FÜ  §¡£×!3Ñ!3Ó!5°qÖ9ô !ò r[   c                 óô   • [        [        R                  " S5      5      n[        U/ SQ'   [	        U5      n[        SS5      [        SS5      [        SS5      /n[        X#5        U R                  [        5        g )Nrp   ©r   r@   r=   é   é   é	   r   é   r†   ru   r‡   )r
   rS   r   r	   r   Úslicer5   r‚   ©rT   rC   rU   rV   s       rX   Útest_clump_maskedÚTestGeneric.test_clump_masked¤   s\   € äœŸš 2›Ó'ˆÜ &ˆÒ
Ñä˜A‹ˆÜ˜˜A“;¤ a¨£¬U°1°b«\Ð:ˆÜ�TÔ#à×ÑœÕ&r[   c                 óÞ   • [        [        R                  " S5      5      n[        U/ SQ'   [	        U5      n[        SS5      [        SS5      /n[        X#5        U R                  [        5        g )Nrp   r…   r‰   r†   ru   r‡   )r
   rS   r   r	   r   rŠ   r5   r‚   r‹   s       rX   Útest_clump_unmaskedÚTestGeneric.test_clump_unmasked¯   sU   € äœŸš 2›Ó'ˆÜ &ˆÒ
ÑÜ˜aÓ ˆÜ˜˜A“;¤ a¨£Ð.ˆÜ�TÔ#à×ÑœÕ(r[   c                 ó¸  • [        S5      n[        U5      n[        U[        SUR                  5      /5        [
        R                  " S[        S9Ul        [        U[        SUR                  5      /5        [        XS:  US:„  -  US:H  -  '   [        U5      n[        U[        SS5      [        SS5      /5        [        US S & [        U5      n[        U/ 5        g )	Nrp   r   r>   r‰   r‡   é   r†   rˆ   )
r   r   r5   rŠ   ÚsizerS   r   ÚboolrB   r	   )rT   rC   rU   s      rX   Útest_flatnotmasked_contiguousÚ)TestGeneric.test_flatnotmasked_contiguous¹   s·   € ä�2‹Jˆä'¨Ó*ˆÜ�TœE ! Q§V¡VÓ,Ð-Ô.ä—’˜"¤DÑ)ˆŒÜ�TœE ! Q§V¡VÓ,Ð-Ô.ä*0ˆˆq‰5�Q˜‘UÑ
˜q A™vÑ
&Ñ'Ü'¨Ó*ˆÜ�TœE ! Q›K¬¨q°!«Ð5Ô6äˆ‰!ˆÜ'¨Ó*ˆÜ�T˜2Õr[   r_   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__rY   rj   rm   rr   r‚   rŒ   r�   r•   Ú__static_attributes__r_   r[   rX   r9   r9   J   s+   † ò$ò<
Aò<ò$ò(:ò 	'ò)õr[   r9   c                   óÔ   • \ rS rSrS rS rS rS rS rS r	S r
\R                  R                  S	/ S
QSS// SQS/S/4/ SQ/ SQ/S/ SQ/SS// SQ// SQ/4/5      S 5       rS rSrg)ÚTestAverageéÌ   c                 óB  • [        / SQ/ SQS9n[        S[        USS95        [        S[        U/ SQS95        [        U/ SQS	S
9u  p#[        SU5        [        US:H  5        [        US S & [        [        USS9R
                  S	/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        [        [        USS9R
                  S   S	/5        [        SS/[        USS95        [        USS	S9u  p#[        USS/5        g )N)ç        ç      ð?ç       @ç      @)TFFF©rB   r¢   r   ©Úaxis)r¡   r¡   r¢   r¡   ©ÚweightsT©r¨   Úreturnedç      @r=   r@   r    ©r¦   rª   r¡   )r   r5   r   r2   r	   rB   Úreshape)rT   ÚottÚresultÚwtss       rX   Útest_testAverage1ÚTestAverage.test_testAverage1Î   s  € äÒ$Ò+FÑGˆÜ�Sœ' #¨AÑ.Ô/Ü�Sœ' #Ò/?Ñ@ÔAÜ˜cÒ+;ÀdÑK‰ˆÜ�S˜&Ô!Ü��s‘
ÔÜˆ‰AˆÜ”W˜S qÑ)×.Ñ.°°Ô7ÜÒ$Ò+FÑGˆØ�k‰k˜!˜QÓˆÜˆŠAˆqˆD‰	Ü”W˜S qÑ)¨C°¨:Ô6Ü”W˜S qÑ)×.Ñ.¨qÑ1°D°6Ô:Ü�b˜"�Xœw s°Ñ3Ô4Ü˜c¨°DÑ9‰ˆÜ�S˜2˜r˜(Õ#r[   c           
      óL  • / SQn/ SQ/ SQ/n[        S[        R                  S9n[        [	        USS9S5        [        [	        USUS9S5        [        [        S[        R                  S9S	[        S5      -  /5      n[        [	        US 5      [        R                  R                  [        R                   " S5      5      S
-  S-  5        [        [	        USS9[        R                   " S5      S
-  S	-  5        [        [	        USS9[	        USS9[	        USS9S	-  /5        [        [	        US US9S5        [        [	        USUS9/ SQ5        [        [	        USS9[	        USS9[	        USS9S	-  /5        [        S5      n/ SQn/ SQ/ SQ/n[        S5      n/ SQn	[        [	        [        X55      SS9S5        [        [	        [        X65      SS9S5        [        [	        [        X85      SS9R                  S/5        [        [	        [        X95      SS9S5        [        [        [	        [        X85      SS95      S5        [        XG5      n
[        [	        U
S 5      S5        [        [	        U
SS9/ SQ5        [        [	        U
SS9SS/5        [        [	        U
SUS9/ SQ5        g )N©r   r@   r@   r@   r@   r   ©r@   r   r   r   r   r@   r†   r>   r   r¥   ç      @©r¦   r¨   r¢   r£   ç      (@r@   r§   g«ªªªªª
@)r    r¡   r¢   r£   r«   ç      $@)r   r   r@   r@   r   r   )r   r@   r@   r@   r@   r@   Tr    )r    r¡   ç     ÀX@rº   r«   g      @ç      @)r    r¡   rº   rº   r«   r¹   )r   rS   Úfloat64r5   r   r   ÚaddÚreducer   r   r
   rB   r   )rT   Úw1Úw2ÚxÚyÚm1Úm2Úm3Úm4Úm5Úzs              rX   Útest_testAverage2ÚTestAverage.test_testAverage2á   s8  € âˆÚ Ò"4Ð5ˆÜ�1œBŸJ™JÑ'ˆÜ”W˜Q QÑ'¨Ô-Ü”W˜Q Q°Ñ3°SÔ9Ü”6˜!¤2§:¡:Ñ.°´f¸Q³i±Ð@ÓAˆÜ”W˜Q Ó%¤r§v¡v§}¡}´R·Y²Y¸q³\Ó'BÀRÑ'GÈ#Ñ'MÔNÜ”W˜Q QÑ'¬¯ª°1«¸Ñ):¸RÑ)?Ô@Ü”W˜Q QÑ'Ü˜a aÑ(¬'°!¸!Ñ*<¸sÑ*BÐCô	Eä”W˜Q ¨bÑ1°8Ô<Ü”W˜Q Q°Ñ3Ú.ô	0ä”W˜Q QÑ'Ü˜a aÑ(¬'°!¸!Ñ*<¸sÑ*BÐCô	Eä�1‹XˆÚˆÚ Ò"4Ð5ˆÜ�!‹WˆÚˆÜ”Wœ\¨!Ó0°qÑ9¸3Ô?Ü”Wœ\¨!Ó0°qÑ9¸3Ô?Ü”Wœ\¨!Ó0°qÑ9×>Ñ>ÀÀÔGÜ”Wœ\¨!Ó0°qÑ9¸3Ô?Ü”Uœ7¤<°Ó#6¸QÑ?Ó@À!ÔDÜ˜ÓˆÜ”W˜Q Ó% xÔ0Ü”W˜Q QÑ'Ò)EÔFÜ”W˜Q QÑ'¨#¨s¨Ô4Ü”W˜Q Q°Ñ3Ú2õ	4r[   c                 ó.  • [        S5      n[        S5      S-  n[        X/X!//SSS9u  p4[        [        U5      [        U5      5        [        UR                  UR                  5        [        [	        S5      SSS/SS9u  pV[        [        U5      [        U5      5        [        [	        S5      SS	9u  pV[        [        U5      [        U5      5        [        [	        S5      [	        S5      SS
9u  pV[        [        U5      [        U5      5        [        SS/SS//[        5      n[        USS/SS//5      n[        USS9n	[        U	SS/5        [        USS9n
[        U
SS/5        [        US S9n
[        U
S5        [        USS9n
[        U
SS/5        g )Nr†   r‰   r@   Tr¬   ©r=   r=   r‰   r   ©r¦   r¨   rª   )rª   r©   r=   é   Fr¥   ç      à?r£   r¡   g«ªªªªª@ç      ø?r«   )r   r   r5   r   r   r   rR   r
   )rT   rC   rD   Úr1r¿   Úr2rÀ   Úa2dÚa2dmÚa2daÚa2dmas              rX   Útest_testAverage3ÚTestAverage.test_testAverage3  se  € ä�1‹IˆÜ�1‹I˜‰MˆÜ˜1˜& 1 &Ð)°¸DÑA‰ˆÜ”U˜2“Y¤ b£	Ô*Ü�R—X‘X˜rŸx™xÔ(Üœ˜i›¨q¸1¸a¸&È4ÑP‰ˆÜ”U˜2“Y¤ b£	Ô*Üœ˜i›°4Ñ8‰ˆÜ”U˜2“Y¤ b£	Ô*Üœ˜i›´$°y³/ÈDÑQ‰ˆÜ”U˜2“Y¤ b£	Ô*Ü�a˜�V˜a ˜VÐ$¤eÓ,ˆÜ˜C 5¨% .°4¸°-Ð!@ÓAˆÜ�s Ñ#ˆÜ�T˜C ˜:Ô&Ü˜ 1Ñ%ˆÜ�U˜S #˜JÔ'Ü˜ 4Ñ(ˆÜ�U˜GÔ$Ü˜ 1Ñ%ˆÜ�U˜S #˜JÕ'r[   c                 óH  • [         R                  " / SQ5      R                  SS5      n[         R                  R                  US/S/S//S9n[         R                  " / SQ5      R                  SS5      n[	        X#SSS9n[        S	/S
/S//S/S/S//5      n[        XE5        g )N©r=   r‰   rÎ   r‰   r@   FTr¤   )rÎ   r’   r†   )r¨   r¦   Úkeepdimsr¢   r£   r«   )rS   r   r­   rb   r   r
   r5   )rT   rÁ   rD   ÚwÚactualÚdesireds         rX   Útest_testAverage4ÚTestAverage.test_testAverage4  s—   € ä�HŠH’YÓ×'Ñ'¨¨1Ó-ˆÜ�E‰E�K‰K˜ % ¨5¨'°D°6Ð :ˆKÐ;ˆÜ�HŠH’YÓ×'Ñ'¨¨1Ó-ˆÜ˜¨A¸Ñ=ˆÜ   r d¨R¨DÐ1°U°G¸e¸WÀtÀfÐ3MÓNˆÜ�VÕ%r[   c                 ór  • [         R                  " S5      R                  SSS5      n[         R                  " / SQ5      R                  SSS5      n[         R                  " SS5      n[         R
                  R                  XS9nUS S 2S S 2S4   n[        US	US
9n[        / SQ/ SQS9n[        Xg5        [         R                  " SS5      nSUS S 2S S 2S4'   SUS'   [         R
                  R                  XS9n[        US	US
9n[        [         R                  SS// SQS9n[        Xg5        [         R                  " SS5      n[         R
                  R                  XS9nUSS S 2S S 24   n[        USUS
9n[        SS/SS/S9n[        Xg5        [        R                  " [        SS9   [        USUS
9  S S S 5        [        R                  " [        SS9   [        US	US
9  S S S 5        [        USUS
9n[        US	UR                  S
9n[        Xg5        g ! , (       d  f       Ng= f! , (       d  f       NK= f)Né   r=   r‰   )r    r    r¡   rÏ   rÏ   r    r    rÏ   rÏ   r¡   r    r    rÌ   Fr¤   r   ©r   r@   r·   )ç      @ç       @ç      "@)FFFT©r   r   r@   rå   ræ   )TFFr@   ©r@   r=   g      @g     € @zHShape of weights must be consistent with shape of a along specified axis)Úmatch©r   r@   r=   )r@   r   )rS   r   r­   r   rx   rb   r   r
   r3   ÚnanÚpytestÚraisesÚ
ValueErrorÚT)	rT   rÂ   rÜ   ÚmÚymaÚsubw0rÝ   rÞ   Úsubw1s	            rX   Ú$test_weight_and_input_dims_differentÚ0TestAverage.test_weight_and_input_dims_different$  sö  € ô �IŠI�b‹M×!Ñ! ! Q¨Ó*ˆÜ�HŠHÒEÓFß‰W�Q˜˜1Óð 	
ô �GŠG�I˜uÓ%ˆÜ�e‰e�k‰k˜!ˆkÐ$ˆØ’!’Q˜�'‘
ˆä˜ 6°5Ñ9ˆÜš|Ò2GÑHˆÜ˜FÔ,ä�GŠG�I˜uÓ%ˆØˆŠ!ŠQ�ˆ'‰
Øˆˆ'‰
Ü�e‰e�k‰k˜!ˆkÐ$ˆÜ˜ 6°5Ñ9ˆÜÜ�V‰V�R˜ÐÚ%ñ'ˆô 	˜FÔ,ä�GŠG�I˜uÓ%ˆÜ�e‰e�k‰k˜!ˆkÐ$ˆà�!’Qš�'‘
ˆÜ˜ 6°5Ñ9ˆÜ  d˜|°5¸%°.ÑAˆÜ˜FÔ,ô �]Š]Üð8ó9ô �C˜i°Ò7÷	9ô �]Š]Üð8ó9ô �C˜f¨eÒ4÷	9ô ˜ 6°5Ñ9ˆÜ˜# F°E·G±GÑ<ˆÜ˜FÕ,÷9õ 9ú÷9õ 9ús   Æ*HÇH(È
H%È(
H6c                 óŽ   • [        [        SS/5      5      n[        US5        [        [        / SQ/ SQS95      n[        US5        g )Nr@   r=   rÐ   )r@   r=   r‰   rÎ   ©FFTTr¤   )r   r   r5   ©rT   rC   s     rX   Útest_onintegers_with_maskÚ%TestAverage.test_onintegers_with_maskW  s:   € ä”E˜1˜a˜&“MÓ"ˆÜ�Q˜ÔÜ”Eš,Ò-GÑHÓIˆÜ�Q˜Õr[   c                 ó  • [         R                  " / SQ/ SQ/[        S9n[        / SQ/ SQ/US9n[	        U5      n[         R                  " UR                  5       5      n[        UR                  UR                  5        [        UR                  UR                  5        [	        USS9n[	        UR                  SS9[	        UR                  SS9S	-  -   n[        UR                  UR                  5        [        UR                  UR                  5        [	        US
S9n[	        UR                  S
S9[	        UR                  S
S9S	-  -   n[        UR                  UR                  5        [        UR                  UR                  5        [         R                  " / SQ/ SQ/5      n	[	        X)S9n
[         R                  " UR                  5       X‘)    S9n[        U
R                  UR                  5        [        U
R                  UR                  5        [	        X)SS9n[	        UR                  U	SS9[	        UR                  U	SS9S	-  -   n[        UR                  UR                  5        [        UR                  UR                  5        [	        X)S
S9n[	        UR                  U	S
S9[	        UR                  U	S
S9S	-  -   n[        UR                  UR                  5        [        UR                  UR                  5        g )N)r   r   r   r@   r   ©r   r@   r   r   r   r>   )r   y      ð?       @y      @      @y      @      @y      @       @)y              "@ù              ð?y       @      @y      @      @y      @      @r¤   r   r¥   rý   r@   )rÏ   r¡   r¢   r¡   rÏ   )r¡   r¡   r¡   r¡   r¡   r§   ©r¨   r¦   )	rS   r   r”   r
   r   r{   r3   ÚrealÚimag)rT   rB   rC   ÚavÚexpectedÚav0Ú	expected0Úav1Ú	expected1r°   ÚwavÚwav0Úwav1s                rX   Útest_complexÚTestAverage.test_complex^  s@  € ô �xŠxšÚ(ð*Ü15ñ7ˆäÒ=Ú>ð@à"ñ$ˆô �Q‹ZˆÜ—:’:˜aŸl™l›nÓ-ˆÜ˜BŸG™G X§]¡]Ô3Ü˜BŸG™G X§]¡]Ô3ä�a˜aÑ ˆÜ˜AŸF™F¨Ñ+¬g°a·f±fÀ1Ñ.EÈÑ.JÑJˆ	Ü˜CŸH™H i§n¡nÔ5Ü˜CŸH™H i§n¡nÔ5ä�a˜aÑ ˆÜ˜AŸF™F¨Ñ+¬g°a·f±fÀ1Ñ.EÈÑ.JÑJˆ	Ü˜CŸH™H i§n¡nÔ5Ü˜CŸH™H i§n¡nÔ5ô �hŠhÒ1Ú1ð3ó 4ˆä�aÑ%ˆÜ—:’:˜aŸl™l›n°c¸%±jÑAˆÜ˜CŸH™H h§m¡mÔ4Ü˜CŸH™H h§m¡mÔ4ä�q¨AÑ.ˆÜ˜QŸV™V¨S°qÑ9Ü˜QŸV™V¨S°qÑ9¸BÑ>ñ?ˆ	ä˜DŸI™I y§~¡~Ô6Ü˜DŸI™I y§~¡~Ô6ä�q¨AÑ.ˆÜ˜QŸV™V¨S°qÑ9Ü˜QŸV™V¨S°qÑ9¸BÑ>ñ?ˆ	ä˜DŸI™I y§~¡~Ô6Ü˜DŸI™I y§~¡~Õ6r[   z<x, axis, expected_avg, weights, expected_wavg, expected_wsum©r@   r=   r‰   Nr¢   )r‰   rÎ   r@   g      ü?rå   )r@   r=   r’   )r@   r†   é   r   )r¡   r«   rå   r@   r‰   )r¡   r»   g      #@)rÎ   rÎ   rÎ   c                 óJ  • [         R                  R                  XSS9nUR                  [         R                  " U5      :X  d   e[	        Xs5        [         R                  R                  XUSS9nUR                  [         R                  " U5      :X  d   e[	        X…5        [         R                  R                  XUSSS9u  p‰UR                  [         R                  " U5      :X  d   e[	        X…5        U	R                  [         R                  " U5      :X  d   e[	        X–5        g )NT)r¦   rÛ   )r¦   r¨   rÛ   )r¦   r¨   rª   rÛ   )rS   rb   r   r   r4   )
rT   rÁ   r¦   Úexpected_avgr¨   Úexpected_wavgÚexpected_wsumÚavgÚwavgÚwsums
             rX   Útest_basic_keepdimsÚTestAverage.test_basic_keepdimsŠ  sá   € ô �e‰e�m‰m˜A°4ˆmÐ8ˆØ�y‰yœBŸHšH \Ó2Ó2Ð2Ð2Ü˜3Ô-ä�u‰u�}‰}˜Q°7ÀTˆ}ÐJˆØ�z‰zœRŸXšX mÓ4Ó4Ð4Ð4Ü˜4Ô/ä—U‘U—]‘] 1¸Ø,0¸4ð #ð A‰
ˆà�z‰zœRŸXšX mÓ4Ó4Ð4Ð4Ü˜4Ô/Ø�z‰zœRŸXšX mÓ4Ó4Ð4Ð4Ü˜4Õ/r[   c                 ó´  • [         R                  R                  [         R                  " S5      R	                  SS5      / SQ/ SQ/ SQ/S9n[        / SQSS9n[        / SQ/ SQS9n[        USUSS	9n[         R                  " / S
Q5      n[        XE5        [        USUSS	9n[         R                  " / SQ5      n[        Xg5        [         R                  R                  / SQ/ SQS9n[        USSS[         R                  /S9n[        US5        [         R                  R                  / SQ/ SQ/ SQ// SQ/ SQ/ SQ/S9n[         R                  R                  US[         R                  S/SS9n[         R                  R                  S[         R                  [         R                  S// SQS9n[        Xh5        [        UR                  UR                  5        g )Nrˆ   r‰   ©r@   r   r   ©r   r   r   r¤   )r’   é   é   Fr   rÍ   )ç      @g     à@g     à@)r  g0�­N@g0�­N@)r¡   r¢   r£   r«   r÷   r@   r§   rÐ   )r»   r  rä   rå   )ræ   r¡   r¢   r£   )FTTF)TFTT)TFTFrþ   r¡   ç      @)rS   rb   r   r   r­   r
   r   r3   rë   r5   rB   )	rT   rC   Úweights_unmaskedÚweights_maskedÚavg_unmaskedÚexpected_unmaskedÚ
avg_maskedÚexpected_maskedÚavg_expecteds	            rX   Útest_masked_weightsÚTestAverage.test_masked_weights¡  s�  € ô �E‰E�K‰KœŸ	š	 !›×,Ñ,¨Q°Ó2Ú'ª²IÐ>ð ð @ˆä'ª¸%Ñ@ÐÜ%¢kº	ÑBˆä˜q qØ'7À%ñIˆäŸHšHÒ%<Ó=ÐÜ˜LÔ<ä˜Q Q°ÈÑOˆ
ÜŸ(š(Ò#NÓOˆÜ˜JÔ8ô
 �E‰E�K‰KÒ1Ú2ð ð 4ˆä˜q¨1¨a°´B·F±FÐ*;Ñ<ˆä˜L¨#Ô.ä�E‰E�K‰KÚ Ú Ú ð
ò
 'Ú%Ú&ð
ð	 ð ˆô —U‘U—]‘] 1¨q´"·&±&¸!¨nÀ1�]ÐEˆ
Ü—u‘u—{‘{ C¬¯©´·±¸Ð#=Ú#=ð #ð ?ˆô 	˜JÔ5Ü�Z—_‘_ l×&7Ñ&7Õ8r[   r_   )r—   r˜   r™   rš   r±   rÉ   r×   rß   rô   rù   r
  rì   ÚmarkÚparametrizer  r%  r›   r_   r[   rX   r�   r�   Ì   s’   † ò$ò& 4òD(ò0&ò1-òfò*7ðX ‡[�[×ÑØFÚ
�T˜C˜5¢)¨d¨V°c°UÐ	;Ú’jÐ
! 1¢Ð&7Øˆaˆ&’?Ð#¢i [ð
2ð	3óñ0óð0õ")9r[   r�   c                   ó&   • \ rS rSrS rS rS rSrg)ÚTestConcatenatoriÍ  c                 ó  • [        [        S   [        / SQ5      5        [        S5      n/ SQn[	        XS9n[        USSU4   n[        [        U[        5      5        [        U/ SQ5        [        UR                  [        USSU4   5        g )N©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@   )	r4   r'   r   r   r
   r2   Ú
isinstancer   rB   )rT   rD   rð   Údr]   s        rX   Útest_1dÚTestConcatenator.test_1dÐ  su   € äœ3Ð/Ñ0´%Ò8JÓ2KÔLÜ�‹GˆÚˆÜ˜Ñ#ˆÜ��1�a˜�
‰OˆÜ”
˜1œkÓ*Ô+Ü˜1ÒBÔCÜ˜1Ÿ6™6¤3 q¨!¨Q° z¡?Õ3r[   c                 óP  • [         R                  R                  SS5      n[         R                  R                  SS5      n[         R                  " [         R                  R                  SS5      S5      n[         R                  " [         R                  R                  SS5      S5      n[	        XS9n[	        X$S9n[
        SXV4   n[        UR                  S:H  5        [        US S 2S S24   U5        [        US S 2SS 24   U5        [        UR                  [         R                  SX44   5        [
        XV4   n[        UR                  S:H  5        [        US S2S S 24   U5        [        USS 2S S 24   U5        [        UR                  [         R                  X44   5        g )Nr’   r   r¤   Ú1)r’   rp   )rp   r’   )rS   ÚrandomÚrandÚroundr
   r'   r2   r   r4   rB   Úr_)rT   Úa_1Úa_2Úm_1Úm_2Úb_1Úb_2r.  s           rX   Útest_2dÚTestConcatenator.test_2dÛ  s>  € ä�i‰i�n‰n˜Q Ó"ˆÜ�i‰i�n‰n˜Q Ó"ˆÜ�hŠh”r—y‘y—~‘~ a¨Ó+¨QÓ/ˆÜ�hŠh”r—y‘y—~‘~ a¨Ó+¨QÓ/ˆÜ˜3Ñ)ˆÜ˜3Ñ)ˆä��S�ÑˆÜ�—‘˜7Ñ"Ô#Ü˜1šQ   ˜U™8 SÔ)Ü˜1šQ ¡˜U™8 SÔ)Ü˜1Ÿ6™6¤2§5¡5¨¨c¨Ñ#7Ô8Ü��‰MˆÜ�—‘˜7Ñ"Ô#Ü˜1˜R˜a˜R¢˜U™8 SÔ)Ü˜1˜Q™R¢˜U™8 SÔ)Ü˜1Ÿ6™6¤2§5¡5¨¨¡?Õ3r[   c                 ó\  • [         [        R                  R                  S4   n[	        UR
                  SS/5        [	        UR                  S   S5        [         SS/[        R                  R                  4   n[	        UR
                  / SQ5        [	        UR                  S S SS/5        g )Nr@   TFr=   )FFT)r'   rS   rb   r	   r5   rB   ry   )rT   rÝ   s     rX   Útest_masked_constantÚ%TestConcatenator.test_masked_constantï  s~   € Ü”R—U‘U—\‘\ 1�_Ñ%ˆÜ�V—[‘[ 4¨ -Ô0Ü�V—[‘[ ‘^ QÔ'ä�a˜�VœRŸU™UŸ\™\Ð)Ñ*ˆÜ�V—[‘[Ò"6Ô7Ü�V—[‘[  !�_ q¨! fÕ-r[   r_   N)r—   r˜   r™   rš   r/  r=  r@  r›   r_   r[   rX   r*  r*  Í  s   † ò	4ò4õ(.r[   r*  c                   ó    • \ rS rSrS rS rSrg)ÚTestNotMaskediù  c           	      ó\  • [        [        R                  " S5      R                  SS5      / SQ/ SQ/ SQ/ SQ/ SQ/S9n[	        US 5      n[        US	S
/5        [	        US	5      n[        US	   SS/5        [        US   SS/5        [	        US5      n[        US	   SS/5        [        US   SS/5        [	        UR                  S 5      n[        US	S
/5        [	        UR                  S	5      n[        US	   SS/5        [        US   SS/5        [	        UR                  S5      n[        US	   SS/5        [        US   SS/5        [        US'   [	        US	5      n[        US	   SS/5        [        US   SS/5        [	        US5      n[        US	   SS/5        [        US   SS/5        g )Né   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   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
   rS   r   r­   r*   r5   ry   r	   ©rT   ry   rU   s      rX   Ú
test_edgesÚTestNotMasked.test_edgesü  s®  € äœBŸIšI b›M×1Ñ1°!°QÓ7Ú"1Ú"1Ú"1Ú"1Ú"1ð	"3ñ5ˆô ˜t TÓ*ˆÜ�T˜A˜r˜7Ô#Ü˜t QÓ'ˆÜ�T˜!‘W˜°Ð@ÔAÜ�T˜!‘W˜°Ð@ÔAÜ˜t QÓ'ˆÜ�T˜!‘W˜°Ð@ÔAÜ�T˜!‘W˜°Ð@ÔAä˜tŸy™y¨$Ó/ˆÜ�T˜A˜r˜7Ô#Ü˜tŸy™y¨!Ó,ˆÜ�T˜!‘W˜°Ð@ÔAÜ�T˜!‘W˜°Ð@ÔAÜ˜tŸy™y¨"Ó-ˆÜ�T˜!‘W˜°Ð@ÔAÜ�T˜!‘W˜°Ð@ÔAäˆˆR‰Ü˜t QÓ'ˆÜ�T˜!‘W˜°Ð@ÔAÜ�T˜!‘W˜°Ð@ÔAÜ˜t RÓ(ˆÜ�T˜!‘W˜|¨\Ð:Ô;Ü�T˜!‘W˜|¨\Ð:Õ;r[   c                 ó°  • [        [        R                  " S5      R                  SS5      / SQ/ SQ/ SQ/S9n[	        US 5      n[        U[        SS	S 5      [        S
SS 5      [        SSS 5      /5        [	        US5      n[        U[        SSS 5      [        SSS 5      /[        SSS 5      [        SSS 5      /[        SSS 5      [        SSS 5      /[        SSS 5      [        SSS 5      /[        SSS 5      /[        SSS 5      // [        SSS 5      //5        [	        US5      n[        U[        SS	S 5      // [        SSS 5      [        SSS 5      //5        g )NrF  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†   ru   )r
   rS   r   r­   r)   r5   rŠ   )rT   rC   Útmps      rX   Útest_contiguousÚTestNotMasked.test_contiguous  sp  € äœŸš 2›×.Ñ.¨q°!Ó4Ú7Ú7Ú7ð9ñ:ˆô # 1 dÓ+ˆÜ�SÜ�!�Q˜ÓÜ�"�b˜$ÓÜ�"�b˜$Óð
ô 	ô # 1 aÓ(ˆÜ�SÜ�1�a˜Ó¤ a¨¨DÓ 1Ð2Ü�1�a˜Ó¤ a¨¨DÓ 1Ð2Ü�1�a˜Ó¤ a¨¨DÓ 1Ð2Ü�1�a˜Ó¤ a¨¨DÓ 1Ð2Ü�1�a˜ÓÐÜ�1�a˜ÓÐØÜ�1�a˜ÓÐð	
ô 		ô # 1 aÓ(ˆÜ�SÜ�1�a˜ÓÐØÜ�1�a˜Ó¤ a¨¨DÓ 1Ð2ð
õ 	r[   r_   N)r—   r˜   r™   rš   rK  rR  r›   r_   r[   rX   rC  rC  ù  s   † ò <õDr[   rC  c                   ó  • \ rS rSrS rS rS r\R                  R                  S/ SQ5      \R                  R                  SS/\
R                  R                  S	4\
R                  R                  S
4/5      S 5       5       rS rS rS rSrg)ÚTestCompressFunctionsi?  c           
      óª  • [         R                  " [        [        S5      5      5      R	                  SSS5      n[         R
                  " S5      R                  [        5      nSUS'   [        XS9n[        U5      n[        U/ S	Q/ S
Q/ SQ// SQ/ SQ/ SQ//5        [        US5      n[        U/ SQ/ SQ/ SQ/ SQ// SQ/ SQ/ SQ/ SQ//5        [        US5      n[        U/ SQ/ SQ/ SQ// SQ/ SQ/ SQ// SQ/ SQ/ SQ//5        [        US5      n[        US5      n[        US5      n[        X45        [        X55        [        X65        [        US5      n[        U/ S	Q/ S Q/ S
Q/ SQ// S!Q/ S"Q/ S#Q/ S$Q// SQ/ S%Q/ SQ/ SQ//5        [        US&5      n[        US'5      n[        US(5      n[        X45        [        X55        [        X65        [        US)5      n[        U/ SQ/ SQ/ SQ// SQ/ SQ/ SQ//5        [        US*5      n[        X45        [        US+5      n[        U/ S	Q/ S
Q/ SQ// S!Q/ S#Q/ S$Q// SQ/ SQ/ SQ//5        [        US,5      n[        US-5      n[        US.5      n[        X45        [        X55        [        X65        [        US/5      n[        U/ S	Q/ S Q/ S
Q/ SQ// SQ/ S%Q/ SQ/ SQ//5        [        US05      n[        X45        g )1Né<   r‰   rÎ   r’   ©r‰   rÎ   r’   T©r@   r@   r@   r¤   )r   r=   r‰   rÎ   )rp   râ   é   é   )é   é   é   é   )é(   é*   é+   é,   )é2   é4   é5   é6   )é7   é9   é:   é;   r   rG  )r’   r†   ru   r‡   rˆ   )rp   r  râ   rZ  r[  )r\  rN  r]  r^  r_  )r`  é)   ra  rb  rc  )é-   é.   é/   é0   é1   )rd  é3   re  rf  rg  )rh  é8   ri  rj  rk  r@   )é   é   rO  rP  rF  )é   r  é    é!   é"   )é#   é$   é%   é&   é'   r^   rI  )rI  r=   )r’   ru   r‡   rˆ   )rt  rO  rP  rF  )rE  é   r  é   )rv  rw  rx  ry  )rz  r|  r}  r~  )rm  ro  rp  rq  r<   rH  )rH  rã   )r   rI  rè   )rI  r=   )r@   rH  )rI  rH  ©r   r=   )r   rH  )
rS   r   Úlistrv   r­   r   Úastyper”   r   r5   )rT   rÁ   rð   rC   Úa2Úa3Úa4s          rX   Útest_compress_ndÚ&TestCompressFunctions.test_compress_ndA  s  € ä�HŠH”Tœ% 	Ó*Ó+Ó,×4Ñ4°Q¸¸1Ó=ˆÜ�HŠH�YÓ×&Ñ&¤tÓ,ˆØˆˆ'‰
Ü�!Ñˆô ˜‹NˆÜ�QÒ*Ú*Ú*ð,ò +Ú*Ú*ð,ð-ô 	.ô ˜˜1ÓˆÜ�QÒ.Ú.Ú.Ú.ð0ò /Ú.Ú.Ú.ð0ð	1ô 	2ô ˜˜1ÓˆÜ�QÒ.Ú.Ú.ð0ò /Ú.Ú.ð0ò /Ú.Ú.ð0ð1ô 	2ô ˜˜DÓ!ˆÜ˜˜BÓˆÜ˜˜EÓ"ˆÜ�QÔÜ�QÔÜ�QÔô ˜˜1ÓˆÜ�Qš/Ú)Ú*Ú*ð,ò +Ú*Ú*Ú*ð,ò +Ú*Ú*Ú*ð,ð-ô 	.ô ˜˜DÓ!ˆÜ˜˜BÓˆÜ˜˜EÓ"ˆÜ�QÔÜ�QÔÜ�QÔô ˜˜6Ó"ˆÜ�QÒ.Ú.Ú.ð0ò /Ú.Ú.ð0ð1ô 	2ô ˜˜GÓ$ˆÜ�QÔô ˜˜6Ó"ˆÜ�QÒ*Ú*Ú*ð,ò +Ú*Ú*ð,ò +Ú*Ú*ð,ð-ô 	.ô ˜˜GÓ$ˆÜ˜˜GÓ$ˆÜ˜˜HÓ%ˆÜ�QÔÜ�QÔÜ�QÔô ˜˜6Ó"ˆÜ�QÒ*Ú*Ú*Ú*ð,ò +Ú*Ú*Ú*ð,ð	-ô 	.ô ˜˜GÓ$ˆÜ�QÕr[   c                 óÎ  • [        [        R                  " S5      R                  SS5      / SQ/ SQ/ SQ/S9n[	        [        U5      SS/SS	//5        [	        [        US
5      / SQ/ SQ/5        [	        [        US5      SS/SS/SS	//5        [        UR                  / SQ/ SQ/ SQ/S9n[	        [        U5      S
S/SS	//5        [	        [        US
5      / SQ/ SQ/5        [	        [        US5      S
S/SS/SS	//5        [        UR                  / SQ/ SQ/ SQ/S9n[	        [        U5      S	//5        [	        [        US
5      / SQ/5        [	        [        US5      S/S/S	//5        [        UR                  / SQ/ SQ/ SQ/S9n[	        [        U5      R                  S
5        [	        [        US
5      R                  S
5        [	        [        US5      R                  S
5        g )Nrˆ   r‰   r  r  r¤   rÎ   r’   ru   r‡   r   rX  )r†   ru   r‡   r@   r=   ©r   r@   r   r†   rê   rç   )r   rS   r   r­   r5   r   Ú_datar“   ©rT   rÁ   s     rX   Útest_compress_rowcolsÚ+TestCompressFunctions.test_compress_rowcols±  s°  € ä”"—)’)˜A“,×&Ñ& q¨!Ó,Ú!¢9ªiÐ8ñ:ˆäÔ% aÓ(¨A¨q¨6°A°q°6Ð*:Ô;ÜÔ% a¨Ó+ªiºÐ-CÔDÜÔ% a¨Ó+¨q°!¨f°q¸!°f¸qÀ!¸fÐ-EÔFÜ�!—'‘'¢ªI²yÐ AÑBˆÜÔ% aÓ(¨A¨q¨6°A°q°6Ð*:Ô;ÜÔ% a¨Ó+ªiºÐ-CÔDÜÔ% a¨Ó+¨q°!¨f°q¸!°f¸qÀ!¸fÐ-EÔFÜ�!—'‘'¢ªI²yÐ AÑBˆÜÔ% aÓ(¨A¨3¨%Ô0ÜÔ% a¨Ó+ªi¨[Ô9ÜÔ% a¨Ó,°¨s°Q°C¸!¸¨oÔ>Ü�!—'‘'¢ªI²yÐ AÑBˆÜÔ% aÓ(×-Ñ-¨qÔ1ÜÔ% a¨Ó+×0Ñ0°!Ô4ÜÔ% a¨Ó+×0Ñ0°!Õ4r[   c                 óü  • [        [        R                  " S5      R                  SS5      / SQ/ SQ/ SQ/S9n[	        [        U5      R                  / SQ/ SQ/ SQ/5        [	        [        US5      R                  / SQ/ SQ/ SQ/5        [	        [        US5      R                  / SQ/ SQ/ SQ/5        [        UR                  / SQ/ S	Q/ SQ/S9n[	        [        U5      R                  / S	Q/ SQ/ S	Q/5        [	        [        US5      R                  / SQ/ SQ/ SQ/5        [	        [        US5      R                  / S	Q/ S	Q/ S	Q/5        [        UR                  / SQ/ S	Q/ SQ/S9n[	        [        U5      R                  / SQ/ SQ/ S
Q/5        [	        [        US5      R                  / SQ/ SQ/ SQ/5        [	        [        US5      R                  / S
Q/ S
Q/ S
Q/5        [        UR                  / SQ/ S	Q/ SQ/S9n[        [        U5      R                  5       [        L 5        [        [        US5      R                  5       [        L 5        [        [        US5      R                  5       [        L 5        [        [        U5      R                  R                  5       5        [        [        US5      R                  R                  5       5        [        [        US5      R                  R                  5       5        g )Nrˆ   r‰   r  r  r¤   rY  r   r@   rŠ  )r@   r@   r   rç   )r   rS   r   r­   r5   r#   rB   r‹  r2   Úallr	   rŒ  s     rX   Útest_mask_rowcolsÚ'TestCompressFunctions.test_mask_rowcolsÅ  s=  € ä”"—)’)˜A“,×&Ñ& q¨!Ó,Ú!¢9ªiÐ8ñ:ˆä”\ !“_×)Ñ)Ú¢ªIÐ6ô	8ä”\ ! QÓ'×,Ñ,Ú¢ªIÐ6ô	8ä”\ ! QÓ'×,Ñ,Ú¢ªIÐ6ô	8ä�!—'‘'¢ªI²yÐ AÑBˆÜ”\ !“_×)Ñ)Ú¢ªIÐ6ô	8ä”\ ! QÓ'×,Ñ,Ú¢ªIÐ6ô	8ä”\ ! QÓ'×,Ñ,Ú¢ªIÐ6ô	8ä�!—'‘'¢ªI²yÐ AÑBˆÜ”\ !“_×)Ñ)Ú¢ªIÐ6ô	8ä”\ ! QÓ'×,Ñ,Ú¢ªIÐ6ô	8ä”\ ! QÓ(×-Ñ-Ú¢ªIÐ6ô	8ä�!—'‘'¢ªI²yÐ AÑBˆÜ”˜Q“×#Ñ#Ó%¬Ð/Ô0Ü”˜Q Ó"×&Ñ&Ó(¬FÐ2Ô3Ü”˜Q Ó"×&Ñ&Ó(¬FÐ2Ô3Ü”˜Q“×$Ñ$×(Ñ(Ó*Ô+Ü”˜Q Ó"×'Ñ'×+Ñ+Ó-Ô.Ü”˜Q Ó"×'Ñ'×+Ñ+Ó-Õ.r[   r¦   )Nr   r@   ÚfuncÚrowcols_axisr   r@   c                 óú   • [        [        R                  " S5      R                  SS5      / SQ/ SQ/ SQ/S9n[	        [
        5         U" XAS9n[        U[        XC5      5        S S S 5        g ! , (       d  f       g = f)Nrˆ   r‰   r  r  r¤   r¥   )r   rS   r   r­   r6   ÚDeprecationWarningr5   r#   )rT   r¦   r“  r”  rÁ   Úress         rX   Ú#test_mask_row_cols_axis_deprecationÚ9TestCompressFunctions.test_mask_row_cols_axis_deprecationå  s_   € ô
 ”"—)’)˜A“,×&Ñ& q¨!Ó,Ú!¢9ªiÐ8ñ:ˆô Ô,Õ-Ù�qÑ$ˆCÜ˜œl¨1Ó;Ô<÷ .×-Ö-ús   ÁA,Á,
A:c           	      ót  • [         R                  " SS5      n/ SQn[        XS9R                  SS5      n[        XS9R                  SS5      n[	        X4SS9n[        UR                  SS/SS	//5        [	        XCSS9n[        UR                  / S
Q/ SQ/ SQ/5        [	        X4SS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [	        XCSS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        / SQn[        XS9R                  SS5      n[        XS9R                  SS5      n[	        X4SS9n[        UR                  S	S/SS//5        [	        XCSS9n[        UR                  / SQ/ SQ/ S
Q/5        [	        X4SS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [        U[	        X45      5        [	        XCSS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        / SQn[        XS9R                  SS5      n[        XS9R                  SS5      n[	        X45      n[        UR                  [        5        [	        XC5      n[        UR                  [        5        [        U/ SQS9R                  SS5      n[        U/ SQS9R                  SS5      n[	        X4SS9n[        UR                  SS/S	S	//5        [	        X4SS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [	        XCSS9n[        UR                  / SQ/ SQ/ SQ/5        [	        XCSS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [        U/ SQS9R                  SS5      n[        U/ SQS9R                  SS5      n[	        X4SS9n[        UR                  S	S	/SS//5        [	        X45      n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [	        XCSS9n[        UR                  / SQ/ SQ/ SQ/5        [	        XCSS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [        U/ SQS9R                  SS5      n[        U/ SQS9R                  SS5      n[	        X4SS9n[        UR                  SS	/SS//5        [	        X4SS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [	        XCSS9n[        UR                  / SQ/ S
Q/ SQ/5        [	        XCSS9n[        U[         R                  " UR                  S	5      UR                  S	5      5      5        [        [         R                  " S5      R                  SSS5      SS	/S	S	//S	S	/S	S	///S9n[        [         R                  " S5      R                  SSS5      S	S	/S	S	//S	S	/S	S///S9n[	        X4SS9n[        UR                  SS/SS//S	S	/S	S///S	S	/S	S//S	S	/S	S////5        [	        X4SS9n[        UR                  S	S	/S	S//S	S	/S	S	///S	S	/S	S	//S	S	/S	S	////5        [	        XCSS9n[        UR                  SS	/S	S	//SS	/S	S	///SS	/S	S	//SS/SS////5        [	        XCSS9n[        UR                  S	S	/S	S	//S	S	/S	S	///S	S	/S	S	//SS	/S	S	////5        [        [         R                  " S5      R                  SSS5      SS	/S	S	//S	S	/S	S	///S9nSn[	        X4SS9n[        UR                  SS	/S	S	//S	S	/S	S	///5        [	        X4SS9n[        UR                  SS	/S	S	//S	S	/S	S	///5        [	        XCSS9n[        UR                  SS	/S	S	//S	S	/S	S	///5        [	        XCSS9n[        UR                  SS	/S	S	//S	S	/S	S	///5        [        [         R                  " S5      R                  SSS5      SS	/S	S	//S	S	/S	S	///S9n[        [         R                  " S5      S	S/S9n[	        X4SS9n[        UR                  SS/SS//5        [	        X4SS9n[        UR                  SS	/S	S	//5        g )Nr@   ru   )r@   r   r   r   r   r   r¤   r=   r‰   T)Ústrictr   rY  r  F©r   r   r   r   r   r@   rç   )r   r   r   r   r   r   )r   r   r@   r   r   r   r‡   r»   )	rS   r   r
   r­   r   r5   rB   Úfilledr   )rT   Únrð   rC   rD   r]   s         rX   Útest_dotÚTestCompressFunctions.test_dotñ  sÌ  € ä�IŠI�a˜‹OˆâˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�˜TÑ"ˆÜ�Q—V‘Všiª²IÐ>Ô?Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9âˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�˜TÑ"ˆÜ�Q—V‘Všiª²IÐ>Ô?Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9Ü�Qœ˜A›	Ô"Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9âˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ˜Ñ#×+Ñ+¨A¨qÓ1ˆÜ�‹IˆÜ�Q—V‘VœVÔ$Ü�‹IˆÜ�Q—V‘VœVÔ$ä˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9Ü�˜TÑ"ˆÜ�Q—V‘Všiª²IÐ>Ô?Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9ä˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�‹IˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9Ü�˜TÑ"ˆÜ�Q—V‘Všiª²IÐ>Ô?Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9ä˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ˜Ò!3Ñ4×<Ñ<¸QÀÓBˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9Ü�˜TÑ"ˆÜ�Q—V‘Všiª²IÐ>Ô?Ü�˜UÑ#ˆÜ�QœŸš˜qŸx™x¨›{¨A¯H©H°Q«KÓ8Ô9äœŸš 1›×-Ñ-¨a°°AÓ6Ø!" A ¨¨A¨Ð/°1°a°&¸1¸a¸&Ð1AÐBñDˆäœŸš 1›×-Ñ-¨a°°AÓ6Ø!" A ¨¨A¨Ð/°1°a°&¸1¸a¸&Ð1AÐBñDˆä�˜TÑ"ˆÜ�Q—V‘VØ˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:Ø˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:ð<ô	=ô �˜UÑ#ˆÜ�Q—V‘VØ˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:Ø˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:ð<ô	=ô �˜TÑ"ˆÜ�Q—V‘VØ˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:Ø˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:ð<ô	=ô �˜UÑ#ˆÜ�Q—V‘VØ˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:Ø˜A˜  A Ð'¨1¨a¨&°1°a°&Ð)9Ð:ð<ô	=ô œŸš 1›×-Ñ-¨a°°AÓ6Ø!" A ¨¨A¨Ð/°1°a°&¸1¸a¸&Ð1AÐBñDˆàˆÜ�˜TÑ"ˆÜ�Q—V‘V  1˜v¨¨1 vÐ.°!°Q°¸!¸Q¸Ð0@ÐAÔBÜ�˜UÑ#ˆÜ�Q—V‘V  1˜v¨¨1 vÐ.°!°Q°¸!¸Q¸Ð0@ÐAÔBÜ�˜TÑ"ˆÜ�Q—V‘V  1˜v¨¨1 vÐ.°!°Q°¸!¸Q¸Ð0@ÐAÔBÜ�˜UÑ#ˆÜ�Q—V‘V  1˜v¨¨1 vÐ.°!°Q°¸!¸Q¸Ð0@ÐAÔBäœŸš 1›×-Ñ-¨a°°AÓ6Ø!" A ¨¨A¨Ð/°1°a°&¸1¸a¸&Ð1AÐBñDˆäœŸš 1›¨Q°¨FÑ3ˆÜ�˜TÑ"ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Ô.Ü�˜UÑ#ˆÜ�Q—V‘V˜q !˜f q¨! fÐ-Õ.r[   c                 ó^  • [         R                  " S5      n[        U5      n[        [	        [        X5      5      [        L 5        [        [	        [        X5      5      [        L 5        [        [	        [        X!5      5      [        L 5        [        [	        [        X"5      5      [        L 5        g )Nr‰   )rS   Úeyer   r2   rc   r   r   ©rT   rC   rD   s      rX   Útest_dot_returns_maskedarrayÚ2TestCompressFunctions.test_dot_returns_maskedarray`  sl   € ä�FŠF�1‹IˆÜ�!‹HˆÜ””S˜“Y“¤;Ð.Ô/Ü””S˜“Y“¤;Ð.Ô/Ü””S˜“Y“¤;Ð.Ô/Ü””S˜“Y“¤;Ð.Õ/r[   c                 óÂ   • [        [        R                  " S5      5      n[        [        R                  " S5      5      n[	        XUS9n[        X2L 5        [        X5        g )Nr‰   )r‰   r‰   ©Úout)r   rS   r¢  r   r   r2   r5   )rT   rC   r¨  r—  s       rX   Útest_dot_outÚ"TestCompressFunctions.test_dot_outi  sE   € Ü”"—&’&˜“)ÓˆÜ”B—H’H˜VÓ$Ó%ˆÜ�!˜CÑ ˆÜ��
ÔÜ�QÕr[   r_   N)r—   r˜   r™   rš   r‡  r�  r‘  rì   r'  r(  rS   rb   Ú	mask_rowsÚ	mask_colsr˜  rŸ  r¤  r©  r›   r_   r[   rX   rU  rU  ?  sŒ   † ònò`5ò(/ð@ ‡[�[×Ñ˜V¢\Ó2Ø‡[�[×Ñ˜f nÐ5Ø!Ÿu™uŸ™°Ð2°R·U±U·_±_ÀaÐ4HÐIóKñ=óKó 3ð=òm/ò^0õr[   rU  c                   ó    • \ rS rSrS rS rSrg)ÚTestApplyAlongAxisiq  c                 ó€   • [        S5      R                  SSS5      nS n[        USU5      n[        USS/SS//5        g )	Nr¸   r=   r‰   c                 ó   • U S   $ ©Nr@   r_   )rD   s    rX   ÚmyfuncÚ*TestApplyAlongAxis.test_3d.<locals>.myfuncv  s   € Ø�Q‘4ˆKr[   r@   rÎ   ru   rp   ©r   r­   r   r5   ©rT   rC   r²  Úxas       rX   Útest_3dÚTestApplyAlongAxis.test_3ds  sF   € Ü�3‹K×Ñ  1 aÓ(ˆò	ô ˜f a¨Ó+ˆÜ�R˜1˜a˜& 1 b 'Ð*Õ+r[   c                 ó‚   • [        S5      R                  SSS5      nS
S jn[        USUSS9n[        USS/SS	//5        g )Nrâ   r=   r‰   c                 ó   • U SU-      $ r±  r_   )rD   Úoffsets     rX   r²  Ú1TestApplyAlongAxis.test_3d_kwargs.<locals>.myfunc€  s   € Ø�Q˜‘Z‘=Ð r[   r@   )r»  r’   r‡   r  )r   r´  rµ  s       rX   Útest_3d_kwargsÚ!TestApplyAlongAxis.test_3d_kwargs}  sH   € Ü�2‹J×Ñ˜q ! QÓ'ˆô	!ô ˜f a¨°1Ñ5ˆÜ�R˜1˜a˜& 1 b 'Ð*Õ+r[   r_   N)r—   r˜   r™   rš   r·  r½  r›   r_   r[   rX   r®  r®  q  s   † ò,õ,r[   r®  c                   ó   • \ rS rSrS rSrg)ÚTestApplyOverAxesi‡  c                 óŒ  • [        S5      R                  SSS5      n[        [        R                  USS/5      n[        R
                  " S/S/S///5      n[        X#5        [        XS-  R                  [        5      '   [        [        R                  USS/5      n[        R
                  " S	/S
/S///5      n[        X#5        g )NrF  r=   r‰   rÎ   r   rW  é\   é|   r  rc  )
r   r­   r   rS   rz   r   r5   r	   rƒ  r”   )rT   rC   rU   Úctrls       rX   Ú
test_basicÚTestApplyOverAxes.test_basic‰  s§   € Ü�2‹J×Ñ˜q ! QÓ'ˆÜœrŸv™v q¨1¨a¨&Ó1ˆÜ�xŠx˜2˜$   s eÐ,Ð-Ó.ˆÜ�TÔ Ü"(ˆˆq‰5�.‰.œÓ
ÑÜœrŸv™v q¨1¨a¨&Ó1ˆÜ�xŠx˜2˜$   r dÐ+Ð,Ó-ˆÜ�TÕ r[   r_   N)r—   r˜   r™   rš   rÅ  r›   r_   r[   rX   rÀ  rÀ  ‡  s   † õ!r[   rÀ  c                   óÜ   • \ 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\R&                  R)                  S/ SQS9S 5       rS rS rS rS rS rS rS rS rSrg)Ú
TestMediani”  c                 óð   • [         R                  R                  [         R                  [         R                  /[         R                  [         R                  //SS9n[	        U[         R                  5        g )NrH  r¥   )rS   rb   r&   Úinfr5   ©rT   Úrs     rX   Útest_pytypeÚTestMedian.test_pytype•  sE   € Ü�E‰E�L‰Lœ2Ÿ6™6¤2§6¡6Ð*¬R¯V©V´R·V±VÐ,<Ð=ÀBˆLÐGˆÜ�QœŸ™Õr[   c                 óˆ  • [         R                  R                  [         R                  R                  [         R                  [         R                  /[         R                  [         R                  //5      SS9n[        U[         R                  5        [         R                  R                  [         R                  R                  [         R                  [         R                  /[         R                  [         R                  //5      S S9n[        U[         R                  5        [         R                  R                  [         R                  R                  [         R                  [         R                  /[         R                  [         R                  //SS9SS9n[        UR                  S5        [         R                  R                  [         R                  R                  [         R                  [         R                  /[         R                  [         R                  //SS9S S9n[        UR                  S5        g )NrH  r¥   Tr¤   )rS   rb   r&   r
   rÊ  r5   rB   rË  s     rX   Útest_infÚTestMedian.test_inf™  s�  € ä�E‰E�L‰LœŸ™×+Ñ+¬b¯f©f´b·f±fÐ-=Ü.0¯f©f´b·f±fÐ-=ð-?ó @ØFHð ð Jˆä�QœŸ™ÔÜ�E‰E�L‰LœŸ™×+Ñ+¬b¯f©f´b·f±fÐ-=Ü.0¯f©f´b·f±fÐ-=ð-?ó @ØFJð ð Lˆä�QœŸ™Ôä�E‰E�L‰LœŸ™×+Ñ+¬b¯f©f´b·f±fÐ-=Ü.0¯f©f´b·f±fÐ-=ð-?ØEIð ,ð Kà ð ð "ˆô 	�Q—V‘V˜TÔ"Ü�E‰E�L‰LœŸ™×+Ñ+¬b¯f©f´b·f±fÐ-=Ü.0¯f©f´b·f±fÐ-=ð-?ØEIð ,ð Kà"ð ð $ˆô 	�Q—V‘V˜TÕ"r[   c                 ó®  • [         R                  " S5      n[        [         R                  R	                  U5      S5        [        [        [         R                  R	                  U5      5      [        L5        [        S5      n[        [         R                  R	                  U5      S5        [        [        [         R                  R	                  U5      5      [        L5        Sn[        [         R                  R	                  U5      S5        [        [        [         R                  R	                  U5      5      [        L5        [         R                  " S5      R                  SS5      n[        [         R                  R	                  USS	9[         R                  " USS	95        [        [         R                  R	                  US
S	9[         R                  " US
S	95        [        [         R                  R	                  US
S	9[        L5        [         R                  " S5      R                  SS5      n[        [         R                  R	                  USS	9[         R                  " USS	95        [        [         R                  R	                  US
S	9[         R                  " US
S	95        [        [         R                  R	                  US
S	9[        L5        g )Nrˆ   r«   r‡   r  r’   r»   éH   r   r¥   r@   g      R@)
rS   r   r5   rb   r&   r2   rc   r   rv   r­   rŒ  s     rX   Útest_non_maskedÚTestMedian.test_non_masked«  sÃ  € Ü�IŠI�a‹LˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü�!‹HˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ØˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ä�IŠI�eÓ×$Ñ$ Q¨Ó*ˆÜ”R—U‘U—\‘\ !¨!�\Ð,¬b¯iªi¸ÀÑ.BÔCÜ”R—U‘U—\‘\ !¨!�\Ð,¬b¯iªi¸ÀÑ.BÔCÜ”—‘—‘˜Q Q�Ð'¬{Ð:Ô;ä�IŠI�fÓ×%Ñ% a¨Ó+ˆÜ”R—U‘U—\‘\ !¨!�\Ð,¬b¯iªi¸ÀÑ.BÔCÜ”R—U‘U—\‘\ !¨!�\Ð,¬b¯iªi¸ÀÑ.BÔCÜ”—‘—‘˜Q Q�Ð'¬{Ð:Õ;r[   c                 óÔ  • [        [        R                  " S5      S/S-  S/S-  -   S9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S	5      R                  S
S5      S/S-  S/S-  -   S9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        R                  R                  USSS9n[        USS/5        [        UR                  SS5        [        [        U5      [        L 5        g)z5test the examples given in the docstring of ma.medianr‡   r   rÎ   r@   r¤   rÐ   r_   úshape mismatchrp   r=   r’   r†   r¶   rH  T©r¦   Úoverwrite_inputr¢   r»   r<   N)r   rS   r   r5   rb   r&   r   r2   rc   r   r­   )rT   rÁ   Úma_xs      rX   Útest_docstring_examplesÚ"TestMedian.test_docstring_examplesÀ  sG  € ä”"—)’)˜A“, a S¨1¡W°¨s°Q©wÑ%6Ñ7ˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü”"—)’)˜B“-×'Ñ'¨¨1Ó-°Q°C¸!±G¸q¸cÀA¹gÑ4EÑFˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü�u‰u�|‰|˜A B¸ˆ|Ð=ˆÜ�T˜B ˜8Ô$Ü�T—Z‘Z Ð'7Ô8Ü”�T“
œkÐ)Õ*r[   c                 ó"  • Sn[        S5       H»  nS H²  n[        SX#S9n[        R                  " [        U* U5      SS/5      nU H$  u  pg [        R
                  R                  XFUS9  M&     [        R                  " US-   * U/SS/5      nU H2  u  pg [        R
                  R                  XFUS9  [        XX&U4-  5      e   M´     M½     g ! [         a    [        XX&U4-  5      ef = f! [        R                  R                   a     M~  f = f)	Nz5mask = %s, ndim = %s, axis = %s, overwrite_input = %sr’   )FTr@   )ÚndminrB   FTrØ  )rv   r   Ú	itertoolsÚproductrS   rb   r&   Ú	ExceptionÚAssertionErrorÚ
exceptionsÚ	AxisError)rT   ÚmsgrÞ  rB   rÁ   Úargsr¦   Úovers           rX   Útest_axis_argument_errorsÚ$TestMedian.test_axis_argument_errorsÏ  s  € ØEˆÜ˜1–XˆEÛ%�Ü˜! 5Ñ4�ô !×(Ò(¬°¨v°uÓ)=ÀÀt¸}ÓM�Û"&‘J�DðNÜŸ™Ÿ™ QÀ4˜ÓHñ #'ô !×(Ò(¨E°A©I¨,¸Ð)>ÀÈÀÓN�Û"&‘J�DðNÜŸ™Ÿ™ QÀ4˜ÑHô -¨S¸%ÀtÐ3LÑ-LÓMÐMó #'ó &ò øô %ó NÜ,¨S¸%ÀtÐ3LÑ-LÓMÐMðNûô Ÿ=™=×2Ñ2ó Úðús   ÁCÂC,ÃC)	Ã,D	ÄD	c                 ó   • [        SSS9n[        [        R                  R	                  U5      S5        [        SSS9n[        [        R                  R	                  U5      [        R                  R
                  5        g )Nr@   Fr¤   T)r   r5   rS   rb   r&   r	   rŒ  s     rX   Útest_masked_0dÚTestMedian.test_masked_0dç  sN   € ä�!˜%Ñ ˆÜ”R—U‘U—\‘\ !“_ aÔ(Ü�!˜$ÑˆÜ”R—U‘U—\‘\ !“_¤b§e¡e§l¡lÕ3r[   c                 ó¨  • [        [        R                  " S5      SS9n[        [        R                  R                  U5      [        R                  R                  5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        R                  R                  R                  L 5        [        [        R                  " S5      SS9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / SQS9n[        [        R                  R                  U5      S	5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / S
QS9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / SQS9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / SQS9n[        [        R                  R                  U5      S5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / SQS9n[        [        R                  R                  U5      S	5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        [        [        R                  " S5      / SQS9n[        [        R                  R                  U5      S	5        [        [        R                  R                  U5      R                  SS5        [        [        [        R                  R                  U5      5      [        L5        g )Nr’   Tr¤   r_   r×  Fr¢   rü   r¶   )r   r@   r@   r@   r@   r    )r   r@   r@   r   r   r£   r»   r†   r´   r  )r   rS   r   r5   rb   r&   r	   r   r2   rc   rd   ÚMaskedConstantr   rŒ  s     rX   Útest_masked_1dÚTestMedian.test_masked_1dî  sR  € Ü”"—)’)˜A“, TÑ*ˆÜ”R—U‘U—\‘\ !“_¤b§e¡e§l¡lÔ3Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬¯©¯©×)BÑ)BÐBÔCÜ”"—)’)˜A“, UÑ+ˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü”"—)’)˜A“,¢_Ñ5ˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü”"—)’)˜A“,¢_Ñ5ˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ä”"—)’)˜A“,¢_Ñ5ˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ä”"—)’)˜B“-¢oÑ6ˆÜ”R—U‘U—\‘\ !“_ bÔ)Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ä”"—)’)˜A“,Ò%7Ñ8ˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9ä”"—)’)˜B“-Ò&8Ñ9ˆÜ”R—U‘U—\‘\ !“_ cÔ*Ü”R—U‘U—\‘\ !“_×*Ñ*¨BÐ0@ÔAÜ””R—U‘U—\‘\ !“_Ó%¬[Ð8Õ9r[   c           
      óè   • [        [        R                  R                  [	        / SQ/ SQS95      R
                  [        R                  R                  [	        / SQ/ SQS95      R
                  5        g )Nr  r  r¤   rŠ  )r5   rS   rb   r&   r   r   ©rT   s    rX   Útest_1d_shape_consistencyÚ$TestMedian.test_1d_shape_consistency  sD   € Ü”R—U‘U—\‘\¤%ª	º	Ñ"BÓC×IÑIÜ—U‘U—\‘\¤%ª	º	Ñ"BÓC×IÑIõ	Kr[   c                 ó–  • Su  p[        [        R                  " SSU5      5      n[        =US S& USS & [        [        R                  " X4[
        S95      nUS S  US S 2S4'   [        R                  " [        U5      5      n[        SU5       H-  n[        R                  R                  U5        X5   US S 2U4'   M/     [        [        US S 2S4   5      S5        [        [        U5      S5        [        [        USS	9[        R                  " U5      5        [        [        UR                  SS	9[        R                  " U5      5        g )
N)ée   rv  ç      ð¿r¡   rp   éöÿÿÿr>   r   r@   r¥   )r
   rS   Úlinspacer	   ÚemptyrR   r   re   rv   r3  Úshuffler5   r&   r   rï   )rT   rž  ÚprÁ   rÈ   Úidxr|   s          rX   r=  ÚTestMedian.test_2d  sý   € à‰ˆÜœŸš S¨"¨aÓ0Ó2ˆÜ!Ð!ˆˆ#ˆ2ˆ��3�4�ÜœŸš 1 &´Ñ6Ó7ˆØ‘A�$ˆŠ!ˆQˆ$‰Ü�iŠiœ˜A›ÓˆÜ�q˜!–ˆAÜ�I‰I×Ñ˜cÔ"Ø‘fˆAŠa�ˆd‹Gñ ô 	”V˜Aša ˜d™G“_ aÔ(Ü”V˜A“Y Ô"Ü”V˜A AÑ&¬¯ª°«Ô4Ü”V˜AŸC™C aÑ(¬"¯(ª(°1«+Õ6r[   c           	      ó†  • [        [        R                  " S5      R                  SS5      5      n[        =US S& USS & [        [        U5      S5        [        [        [        R                  R                  U5      5      [        L5        [        [        USS9/ SQ5        [        [        [        R                  R                  USS95      [        L 5        [        [        US	S9/ S
Q5        [        [        [        R                  R                  US	S95      [        L 5        [        [        US	S9R                  / SQ5        g )Nrv  rp   r‰   éýÿÿÿç      -@r   r¥   )ç      +@r  g      /@r@   )
r   r   r   rp   rZ  rN  r_  r   r   r   )
r@   r@   r@   r   r   r   r   r@   r@   r@   )r
   rS   r   r­   r	   r5   r&   r2   rc   rb   r   rB   rŒ  s     rX   Útest_2d_waxisÚTestMedian.test_2d_waxis(  så   € äœŸš 2›×.Ñ.¨r°1Ó5Ó6ˆÜÐˆˆ"ˆ1ˆ��"�#�Ü”V˜A“Y Ô%Ü””R—U‘U—\‘\ !“_Ó%¬[Ð8Ô9Ü”V˜A AÑ&Ò(:Ô;Ü””R—U‘U—\‘\ !¨!�\Ð,Ó-´Ð<Ô=Ü”V˜A AÑ&Ò(JÔKÜ””R—U‘U—\‘\ !¨!�\Ð,Ó-´Ð<Ô=Ü”V˜A AÑ&×+Ñ+Ò-KÕLr[   c                 óÊ  • [         R                  R                  S5      R                  SSS5      n[        XS-  S:H  '   [        [        US5      SS/SS	/SS/S
S	//5        SUl        [        [        US5      SS/SS/SS//5        [         R                  R                  S5      R                  SSS5      n[        XS-  S:H  '   [        [        US5      SS/SS/SS//5        g )NrF  r‰   rÎ   r=   r   râ   rˆ   r†   r\  r^  )rÎ   r‰   r=   éc   rp   r  rZ  r[  r’   r‡   rN  r]  )rS   rb   r   r­   r	   r5   r&   r   rŒ  s     rX   r·  ÚTestMedian.test_3d4  sØ   € ä�E‰E�L‰L˜Ó×$Ñ$ Q¨¨1Ó-ˆÜˆˆa‰%�1‰*‰Ü”V˜A˜q“\ R¨ G¨a°¨W°r¸1°gÀÀB¸xÐ#HÔIØˆŒÜ”V˜A˜q“\ R¨ H¨r°2¨h¸¸R¸Ð#AÔBÜ�E‰E�L‰L˜Ó×$Ñ$ Q¨¨1Ó-ˆÜˆˆa‰%�1‰*‰Ü”V˜A˜q“\ R¨ H¨q°!¨f°r¸2°hÐ#?Õ@r[   c                 ó¶   • [        [        R                  " S5      R                  SS5      5      n[        =US S& USS & [        [        USS9[        USS95        g )Nrv  rp   r‰   r   rH  r¥   r@   )r
   rS   r   r­   r	   r5   r&   rŒ  s     rX   Útest_neg_axisÚTestMedian.test_neg_axis?  sO   € ÜœŸš 2›×.Ñ.¨r°1Ó5Ó6ˆÜÐˆˆ"ˆ1ˆ��"�#�Ü”V˜A BÑ'¬°¸Ñ):Õ;r[   c                 óJ  • S H�  n[        [        R                  " U5      5      n[        =US S& USS & [        [        R                  " S5      5      n[        X#S9nUS:X  a  [        US5        O[        US5        [        XCL 5        [        [        U5      [        L 5        MŸ     g )	N)rv  ç      >@r  g      ?@r‰   r   r_   r§  rv  r  g      .@)
r
   rS   r   r	   r   r&   r5   r2   rc   r   )rT   ÚvrÁ   r¨  rÌ  s        rX   Útest_out_1dÚTestMedian.test_out_1dD  s†   € ã#ˆAÜœRŸYšY q›\Ó*ˆAÜ#Ð#ˆAˆbˆqˆE�A�b�c�FÜœrŸwšw r›{Ó+ˆCÜ�qÑ"ˆAØ�B‹wÜ˜S $Õ'ä˜S #Ô&Ü�A�HÔÜ”D˜“Gœ{Ð*Ö+ò $r[   c                 ó  • S Hú  n[        [        R                  " U5      R                  SS5      5      n[        =US S& USS & [        [        R
                  " S5      5      n[        USUS9nUS:X  a-  [        S	/S-  / S
Q-   S	/S-  -   S/S-  S/S-  -   S/S-  -   S9nO,[        S	/S-  / SQ-   S	/S-  -   S/S-  S/S-  -   S/S-  -   S9n[        XE5        [        XCL 5        [        [        U5      [        L 5        Mü     g )N)r`  g      D@rv  r  rp   rH  r‰   r   r@   )r¦   r¨  rv  r    )rp   rZ  rN  r_  TFrÎ   r¤   )r  g     €1@g     €5@g     €9@)r
   rS   r   r­   r	   r   r&   r5   r2   rc   r   )rT   r  rÁ   r¨  rÌ  Úes         rX   Útest_outÚTestMedian.test_outR  s  € ã#ˆAÜœRŸYšY q›\×1Ñ1°"°bÓ9Ó:ˆAÜ#Ð#ˆAˆbˆqˆE�A�b�c�FÜœrŸwšw r›{Ó+ˆCÜ�q˜q cÑ*ˆAØ�B‹wÜ  " ¨¡Ò,<Ñ!<À¸tÀa¹xÑ!GØ'+ f¨q¡j°E°7¸Q±;Ñ&>À$ÀÈ!ÁÑ&KñM‘ô ! " ¨¡Ò,DÑ!DÈÀtÈaÁxÑ!OØ'+ f¨q¡j°E°7¸Q±;Ñ&>À$ÀÈ!ÁÑ&KñM�ä˜ÔÜ�A�HÔÜ”D˜“Gœ{Ð*Ö+ò $r[   r¦   )Nr@   r^   rã   )r   rH  )ÚargnamesÚ	argvaluesc                 óÀ  ^^• [         R                  " S[        S9n[         R                  R                  S5      [         R                  " UR
                  5      S S 2S 4   -  nUR                  [         R                  5      n[         R                  U[        U5      '   [        [         R                  " UR
                  5      US9mUc  STR                  -  nO?[        UTR                  5      m[        UU4S j[        TR                  5       5       5      n[        [         R                  " U5      5      n[!        TUSUS9nXeL d   e[#        UR
                  U5        g )	N)r‰   r’   ru   r  r>   )rÎ   éÈ   r¤   r^   c              3   óP   >#   • U  H  oT;   a  S OTR                   U   v •  M     g7f)r@   N)r   )Ú.0r|   Ú	axis_normr.  s     €€rX   Ú	<genexpr>Ú/TestMedian.test_keepdims_out.<locals>.<genexpr>x  s&   øé € ð LÚ=J¸˜)“^‘¨¯©°©Ô3º]ùs   ƒ#&T)r¦   rÛ   r¨  )rS   r   r”   r3  r   r   rƒ  Úintprë   Útupler
   r   Úndimr   rv   rú  r&   r5   )	rT   r¦   rB   rÜ   Ú	shape_outr¨  r¯   r  r.  s	          @@rX   Útest_keepdims_outÚTestMedian.test_keepdims_outc  sÿ   ù€ ô �xŠx˜¬TÑ2ˆä�I‰I×Ñ˜XÓ&¬¯ª°$·*±*Ó)=ºaÀ¸gÑ)FÑFˆØ�H‰H”R—W‘WÓˆÜŸ™ˆŒU�1‹X‰ÜœŸš §¡Ó,°4Ñ8ˆØ‰<Ø˜qŸv™v™‰Iä,¨T°1·6±6Ó:ˆIÜõ LÜ=BÀ1Ç6Á6¼]óLó LˆIäœ2Ÿ8š8 IÓ.Ó/ˆÜ˜ ¨t¸Ñ=ˆØŠ}Ðˆ}Ü�V—\‘\ 9Õ-r[   c                 ó²   • SS/SS/SS//n[         R                  R                  US5      nSS/n[        [         R                  R	                  USS9U5        g )Nr¡   r    r£   r   r¥   )rS   rb   Úmasked_equalr4   r&   )rT   ry   ri   r  s       rX   Ú$test_single_non_masked_value_on_axisÚ/TestMedian.test_single_non_masked_value_on_axis  s]   € Ø�R�Ø�R�Ø�R�ðˆô —U‘U×'Ñ'¨¨aÓ0ˆ
Ø˜�8ˆÜœ2Ÿ5™5Ÿ<™<¨
¸˜<Ð;Ø#õ	%r[   c                 ó$  • S[         R                  " S[        S94 GHô  n[         R                  R	                  S[         R
                  S// SQ/5      nXl        [         R                  R                  US S9n[        [         R                  " U5      5        [        U[         R
                  5        [         R                  R                  UR                  5       SS9n[        [         R                  " U5      5        [        U[         R
                  5        [         R                  R                  USS9n[        [        U5      [        5        [        US[         R
                  S/5        [         R                  R                  USS9n[        [        U5      [        5        [        U[         R
                  S	/5        [         R                  R                  US
S9n[        [        U5      [        5        [        U[         R
                  S	/5        GM÷     [         R                  R	                  S[         R
                  S// SQ/5      n[         R                  R                  US S 2S	4'   [        [         R                  R                  US S9[         R
                  5        [        [         R                  R                  USS9S[         R
                  S/5        [        [         R                  R                  USS9[         R
                  S/5        g )NFr†   r>   r@   r‰   r  r¥   r   r=   rH  rÐ   )rS   r   r”   rb   r   rë   rB   r&   r2   Úisscalarr4   Úravelr5   rc   r   r	   )rT   rB   ÚdmrÌ  s       rX   Útest_nanÚTestMedian.test_nanˆ  sõ  € ØœBŸHšH Q¬dÑ3Ô4ˆDÜ—‘—‘˜q¤"§&¡&¨!˜nªiÐ8Ó9ˆBØŒGô —‘—‘˜R d�Ð+ˆAÜ”B—K’K “NÔ#Ü˜q¤"§&¡&Ô)Ü—‘—‘˜RŸX™X›Z¨a�Ð0ˆAÜ”B—K’K “NÔ#Ü˜q¤"§&¡&Ô)ä—‘—‘˜R a�Ð(ˆAÜœ˜a›¤+Ô.Ü˜q 1¤b§f¡f¨a .Ô1Ü—‘—‘˜R a�Ð(ˆAÜœ˜a›¤+Ô.Ü˜q¤2§6¡6¨1 +Ô.Ü—‘—‘˜R b�Ð)ˆAÜœ˜a›¤+Ô.Ü˜q¤2§6¡6¨1 +×.ñ) 5ô, �U‰U�[‰[˜1œbŸf™f a˜.ª)Ð4Ó5ˆÜ—5‘5—<‘<ˆŠ1ˆaˆ4‰Üœ2Ÿ5™5Ÿ<™<¨°˜<Ð6¼¿¹Ô?Üœ2Ÿ5™5Ÿ<™<¨°˜<Ð3°a¼¿¹À°^ÔDÜœ2Ÿ5™5Ÿ<™<¨°˜<Ð3´b·f±f¸c°]ÕCr[   c                 óî  • [         R                  R                  [         R                  " S5      5      n[         R                  R                  [         R                  " S5      5      n[         R
                  US'   [         R                  R                  US'   [        [         R                  R                  USUS9U5        [         R                  R                  [         R                  " S5      5      n[        [         R                  R                  USUS9U5        [         R                  R                  [         R                  " S	5      5      n[        [         R                  R                  X!S9U5        g )
N)rÎ   ©r‰   rÎ   ©r=   r@   rK   r   r§  ©r‰   r@   r_   )	rS   rb   r
   r   r   rë   r	   r5   r&   )rT   Úor.  s      rX   Útest_out_nanÚTestMedian.test_out_nan¥  sã   € Ü�E‰E×ÑœrŸxšx¨›~Ó.ˆÜ�E‰E×ÑœrŸwšw v›Ó/ˆÜ—&‘&ˆˆ$‰Ü—%‘%—,‘,ˆˆ$‰Ü”R—U‘U—\‘\ ! Q¨A�\Ð.°Ô2Ü�E‰E×ÑœrŸxšx¨›~Ó.ˆÜ”R—U‘U—\‘\ ! Q¨A�\Ð.°Ô2Ü�E‰E×ÑœrŸxšx¨›|Ó,ˆÜ”R—U‘U—\‘\ !�\Ð+¨QÕ/r[   c                 óz  • [         R                  R                  [         R                  " S[        S95      n[         R                  R
                  US S S2'   [         R                  US'   [        [         R                  R                  U5      [         R                  5        [        [         R                  R                  USS9[         R                  5        [         R                  R                  [         R                  " S[        S9R                  SSS5      5      n[         R                  " UR                  5      S-  S:H  Ul        UR                  5       n[         R                  US	'   [         R                  US
'   [        [         R                  R                  U5      [         R                  5        [        [         R                  " [         R                  R                  U5      5      5        [         R                  R                  USS9n[         R                  US'   [         R                  US'   [        [         R                  R                  US5      U5        [         R                  R                  USS9n[         R                  US'   [         R                  US'   [        [         R                  R                  US5      U5        [         R                  R                  USS9n[         R                  US'   [         R                  US'   [        [         R                  R                  US5      U5        g )NrF  r>   r‰   r=   r   r¥   rÎ   r@   r  )r@   r@   r=   ©r=   r‰   rè   ©r@   r‰   r�  )rS   rb   r
   r   rR   r	   rë   r4   r&   r­   r“   rB   Úcopyr2   r(  r5   )rT   rC   ÚaorigrD   s       rX   Útest_nan_behaviorÚTestMedian.test_nan_behavior°  s  € Ü�E‰E×ÑœrŸyšy¨´5Ñ9Ó:ˆÜ—‘—‘ˆ‰#ˆAˆ#‰Ü�v‰vˆˆ!‰Üœ2Ÿ5™5Ÿ<™<¨›?¬B¯F©FÔ3Üœ2Ÿ5™5Ÿ<™<¨°˜<Ð2´B·F±FÔ;ä�E‰E×ÑœrŸyšy¨´5Ñ9×AÑAÀ!ÀQÈÓJÓKˆÜ—’˜1Ÿ6™6Ó" QÑ&¨!Ñ+ˆŒØ—‘“ˆÜ—V‘Vˆˆ'‰
Ü—V‘Vˆˆ'‰
ô 	œ2Ÿ5™5Ÿ<™<¨›?¬B¯F©FÔ3Ü”—’œBŸE™EŸL™L¨›OÓ,Ô-ô �E‰E�L‰L˜ QˆLÐ'ˆÜ—&‘&ˆˆ$‰Ü—&‘&ˆˆ$‰Ü”R—U‘U—\‘\ ! QÓ'¨Ô+ô �E‰E�L‰L˜ QˆLÐ'ˆÜ—&‘&ˆˆ$‰Ü—&‘&ˆˆ$‰Ü”R—U‘U—\‘\ ! QÓ'¨Ô+ô �E‰E�L‰L˜ VˆLÐ,ˆÜ�v‰vˆˆ!‰Ü�v‰vˆˆ!‰Ü”R—U‘U—\‘\ ! VÓ,¨aÕ0r[   c                 óò  • [         R                  " / SQ/ SQ/[         R                  S9n[         R                  R	                  XS:H  S9n[        [         R                  R                  USS9S5        [        [         R                  R                  USS9R                  S5        [        [         R                  R                  US	S9US	   5        [        [         R                  R                  U5      S5        g )
N)r‰   r‰   éÿ   r>   r‰   r¤   r@   r¥   r<  Fr   )rS   r   Úuint8rb   r
   r4   r&   rB   rø   s     rX   Útest_ambigous_fillÚTestMedian.test_ambigous_fillÓ  s§   € ä�HŠH’k¢;Ð/´r·x±xÑ@ˆÜ�E‰E×Ñ˜q¨A¡vÐÐ.ˆÜœ2Ÿ5™5Ÿ<™<¨°˜<Ð2°CÔ8Üœ2Ÿ5™5Ÿ<™<¨°˜<Ð2×7Ñ7¸Ô?Üœ2Ÿ5™5Ÿ<™<¨°˜<Ð2°A°a±DÔ9Üœ2Ÿ5™5Ÿ<™<¨›?¨CÕ0r[   c           
      ó>  • [         R                  [         R                  * 4 GHw  n[         R                  " U[         R                  /[         R                  [         R                  //5      n[         R                  R                  U[         R                  " U5      S9n[        [         R                  R                  USS9U[         R                  /5        [        [         R                  R                  USS9U[         R                  /5        [        [         R                  R                  U5      U5        [         R                  " [         R                  [         R                  U/[         R                  [         R                  U//5      n[         R                  R                  U[         R                  " U5      S9n[        [         R                  R                  USS9U5        [        [         R                  R                  USS9R                  S5        [        [         R                  R                  USS9US   5        [        [         R                  R                  U5      U5        [         R                  " X/X//5      n[        [         R                  R                  U5      U5        [        [         R                  R                  USS9U5        [        [         R                  R                  USS9U5        [         R                  " USU* S/S[         R                  [         R                  S	/S
[         R                  [         R                  U//[         R                  S9n[         R                  R                  U[         R                  " U5      S9nUS:”  aW  [        [         R                  R                  USS9SSU* S/5        [        [         R                  R                  U5      S5        OV[        [         R                  R                  USS9SSU* S/5        [        [         R                  R                  U5      S5        [        [         R                  R                  USS9SSU/5        [        S5       GH  n[        SS5       Hú  n[         R                  " [         R                  /U-  U/U-  -   /S-  5      n[         R                  R                  U[         R                  " U5      S9n[        [         R                  R                  U5      U5        [        [         R                  R                  USS9U5        [        [         R                  R                  USS9[         R                  /U-  U/U-  -   5        Mü     GM     GMz     g )Nr¤   r   r¥   r@   Fru   i÷ÿÿÿrø  r’   rÎ   r>   r«   rä   r»   g      @g      $Àg      "Àg      Àr÷  rp   r=   )rS   rÊ  r   rë   rb   r
   Úisnanr5   r&   r4   rB   Úfloat32rv   )rT   rÊ  rC   r|   r}   s        rX   Útest_specialÚTestMedian.test_specialÜ  sÌ  € Ü—F‘FœRŸV™V˜GÔ$ˆCÜ—’˜3¤§¡˜-¬"¯&©&´"·&±&Ð)9Ð:Ó;ˆAÜ—‘×"Ñ" 1¬2¯8ª8°A«;Ð"Ð7ˆAÜœŸ™Ÿ™ a¨a˜Ð0°3¼¿¹°-Ô@ÜœŸ™Ÿ™ a¨a˜Ð0°3¼¿¹°-Ô@ÜœŸ™Ÿ™ a›¨#Ô.ä—’œ2Ÿ6™6¤2§6¡6¨3Ð/´"·&±&¼"¿&¹&À#Ð1FÐGÓHˆAÜ—‘×"Ñ" 1¬2¯8ª8°A«;Ð"Ð7ˆAÜœrŸu™uŸ|™|¨A°A˜|Ð6¸Ô<ÜœrŸu™uŸ|™|¨A°A˜|Ð6×;Ñ;¸UÔCÜœrŸu™uŸ|™|¨A°A˜|Ð6¸¸!¹Ô=ÜœrŸu™uŸ|™|¨A›°Ô4ô —’˜3˜* s jÐ1Ó2ˆAÜœŸ™Ÿ™ a›¨#Ô.ÜœŸ™Ÿ™ a¨a˜Ð0°#Ô6ÜœŸ™Ÿ™ a¨a˜Ð0°#Ô6ä—’˜3  C 4¨Ð,Ø¤§¡¬¯©°Ð2ØœbŸf™f¤b§f¡f¨cÐ2ð4ô !#§
¡
ñ,ˆAô —‘×"Ñ" 1¬2¯8ª8°A«;Ð"Ð7ˆAØ�Q‹wÜœRŸU™UŸ\™\¨!°!˜\Ð4°r¸2À¸tÀRÐ6HÔIÜœRŸU™UŸ\™\¨!›_¨cÕ2äœRŸU™UŸ\™\¨!°!˜\Ð4°t¸RÀ#ÀÀsÐ6KÔLÜœRŸU™UŸ\™\¨!›_¨dÔ3ÜœŸ™Ÿ™ a¨a˜Ð0°3¸¸cÐ2BÔCä˜2—Y�Ü˜q "ž�AÜŸš¤B§F¡F 8¨a¡<°S°E¸A±IÑ">Ð!?À!Ñ!CÓD�AÜŸ™×*Ñ*¨1´2·8²8¸A³;Ð*Ð?�AÜ ¤§¡§¡¨a£°#Ô6Ü ¤§¡§¡¨a°a Ð!8¸#Ô>Ü ¤§¡§¡¨a°a Ð!8Ü#%§6¡6 (¨Q¡,°3°%¸!±)Ñ!;ö=ô &ô òC %r[   c                 ó2  • [         R                  R                  [         R                  " / [        S95      n[        5        nUR                  [        5        [        [         R                  R                  U5      [         R                  5        [        UR                  S   R                  [        L 5        S S S 5        [         R                  R                  [         R                  " / [        SS95      n[        5        nUR                  [        5        [        R                  " SS[        5        [        [         R                  R                  U5      [         R                  5        [        UR                  S   R                  [        L 5        S S S 5        [         R                  R                  [         R                  " / [        SS95      n[!        [         R                  R                  USS9U5        [!        [         R                  R                  US	S9U5        [         R                  R                  [         R                  " [         R                  [        SS95      n[        R"                  " S
S9 n[        R                  " SS[        5        [!        [         R                  R                  USS9U5        [        US   R                  [        L 5        S S S 5        g ! , (       d  f       GN5= f! , (       d  f       GNm= f! , (       d  f       g = f)Nr>   r   r‰   )r?   rÞ  ÚalwaysÚ r=   r¥   r@   T)Úrecord)rS   rb   r
   r   rR   r7   rH  ÚRuntimeWarningr4   r&   rë   r2   ÚlogÚcategoryÚwarningsÚfilterwarningsr5   Úcatch_warnings)rT   rC   rÜ   rD   s       rX   Ú
test_emptyÚTestMedian.test_empty  sç  € ä�E‰E×ÑœrŸxšx¨´%Ñ8Ó9ˆÜÔ  AØ�H‰H”^Ô$ÜœrŸu™uŸ|™|¨A›´·±Ô7Ü�A—E‘E˜!‘H×%Ñ%¬Ð7Ô8÷ !ô �E‰E×ÑœrŸxšx¨´%¸qÑAÓBˆäÔ  AØ�H‰H”^Ô$Ü×#Ò# H¨b´.ÔAÜœrŸu™uŸ|™|¨A›´·±Ô7Ü�A—E‘E˜!‘H×%Ñ%¬Ð7Ô8÷	 !ô �E‰E×ÑœrŸxšx¨´%¸qÑAÓBˆÜ”R—U‘U—\‘\ !¨!�\Ð,¨aÔ0Ü”R—U‘U—\‘\ !¨!�\Ð,¨aÔ0ô �E‰E×ÑœrŸxšx¬¯©´eÀ1ÑEÓFˆÜ×$Ò$¨DÒ1°QÜ×#Ò# H¨b´.ÔAÜœŸ™Ÿ™ a¨a˜Ð0°!Ô4Ü�A�a‘D—M‘M¤^Ð3Ô4÷ 2Ð1÷+ !Ö ú÷ !Ö ú÷ 2Õ1ús'   ÁA5K$Ã?BK6É8A#LË$
K3Ë6
LÌ
Lc           	      ó´  • [         R                  R                  [         R                  " S5      5      n[	        [        [         R                  R                  UR                  [        5      5      5      [        5        [         R                  US'   [	        [        [         R                  R                  UR                  [        5      5      5      [        5        g )Nrä   r=   )rS   rb   r
   r   r2   rc   r&   rƒ  ra   rR   rë   )rT   r1  s     rX   Útest_objectÚTestMedian.test_object$  su   € Ü�E‰E×ÑœrŸyšy¨›}Ó-ˆÜ””R—U‘U—\‘\ !§(¡(¬6Ó"2Ó3Ó4´eÔ<Ü�v‰vˆˆ!‰Ü””R—U‘U—\‘\ !§(¡(¬6Ó"2Ó3Ó4´eÕ<r[   r_   N)r—   r˜   r™   rš   rÍ  rÐ  rÔ  rÛ  rè  rë  rï  ró  r=  r  r·  r	  r  r  rì   r'  r(  r!  r%  r+  r2  r9  r>  rC  rO  rR  r›   r_   r[   rX   rÈ  rÈ  ”  s¨   † ò ò#ò$<ò*+òNò04ò$:òLKò7ò 
Mò	Aò<ò
,ò,ð" ‡[�[×ÑØò
ð ð 	ñ.ó	ð.ò$%òDò:	0ò!1òF1ò)=òV5õ:=r[   rÈ  c                   ó8   • \ rS rSrS rS rS rS rS rS r	Sr
g	)
ÚTestCovi+  c                 ó^   • [        [        R                  R                  S5      5      U l        g ©Nrâ   )r   rS   r3  r4  ry   rò  s    rX   Úsetup_methodÚTestCov.setup_method-  s   € Üœ"Ÿ)™)Ÿ.™.¨Ó,Ó-ˆ�	r[   c                 óR  • U R                   n[        [        USS9S   R                  [        R
                  5        [        [        XSS9S   R                  [        R
                  5        US:„  n[        [        [        R                  R                  X5      SS9S   R                  [        5      UR                  SS5      ) 5        [        [        [        R                  R                  X5      USS9S   R                  [        5      [        R                  " U) U) 45      5        g )NT©Úrowvarr@   F)rÂ   r\  rÏ   rH  )ry   r2   r   r?   rS   rB  r4   rb   r
   rƒ  r”   r­   r1   )rT   rÁ   rB   s      rX   Útest_covhelperÚTestCov.test_covhelper0  sç   € Ø�I‰Iˆä”
˜1 TÑ*¨1Ñ-×3Ñ3´R·Z±ZÔ@Ü”
˜1¨%Ñ0°Ñ3×9Ñ9¼2¿:¹:ÔFà�3‰wˆÜÜÜ—‘×"Ñ" 1Ó+°Dñàñç‘œ“Ø�\‰\˜!˜RÓ Ð ô		
ô 	ÜÜ—‘×"Ñ" 1Ó+¨q¸ñàñç‘œ“Ü�IŠI˜�u˜t˜e�nÓ%õ		
r[   c           	      ó  • U R                   n[        [        R                  " U5      [        U5      5        [        [        R                  " USS9[        USS95        [        [        R                  " USSS9[        USSS95        g )NFr[  T©r\  Úbias)ry   r3   rS   r   rŒ  s     rX   Útest_1d_without_missingÚTestCov.test_1d_without_missingD  s`   € à�I‰IˆÜœBŸFšF 1›I¤s¨1£vÔ.ÜœBŸFšF 1¨UÑ3´S¸À5Ñ5IÔJÜœBŸFšF 1¨U¸Ñ>Ü ¨%°dÑ;õ	=r[   c           	      ó.  • U R                   R                  SS5      n[        [        R                  " U5      [	        U5      5        [        [        R                  " USS9[	        USS95        [        [        R                  " USSS9[	        USSS95        g )Nr‰   rÎ   Fr[  Tr`  )ry   r­   r3   rS   r   rŒ  s     rX   Útest_2d_without_missingÚTestCov.test_2d_without_missingL  sm   € à�I‰I×Ñ˜a Ó#ˆÜœBŸFšF 1›I¤s¨1£vÔ.ÜœBŸFšF 1¨UÑ3´S¸À5Ñ5IÔJÜœBŸFšF 1¨U¸Ñ>Ü ¨%°dÑ;õ	=r[   c           
      óÞ  • U R                   n[        US'   XR                  5       -  nUR                  5       n[	        [
        R                  " U5      [        U5      5        [	        [
        R                  " USS9[        USS95        [	        [
        R                  " USSS9[        USSS95         [        USS9  USS n[	        [
        R                  " X"S S S2   5      [        XS S S2   5      5        [	        [
        R                  " X"S S S2   SS9[        XS S S2   SS95        [	        [
        R                  " X"S S S2   SSS9[        XS S S2   SSS95        g ! [         a     N¯f = f)NrH  Fr[  Tr`  ©Úallow_maskedr@   )ry   r	   Úmeanr{   r3   rS   r   rî   )rT   rÁ   Únxs      rX   Útest_1d_with_missingÚTestCov.test_1d_with_missingT  s;  € à�I‰IˆÜˆˆ"‰Ø	�V‰V‹X‰ˆØ�\‰\‹^ˆÜœBŸFšF 2›J¬¨A«Ô/ÜœBŸFšF 2¨eÑ4´c¸!ÀEÑ6JÔKÜœBŸFšF 2¨e¸$Ñ?Ü ¨%°dÑ;ô	=ð	Ü� Ò&ð
 ˆq�ˆWˆÜœBŸFšF 2©$¨B¨$¡xÓ0´#°a¹4¸R¸4¹³/ÔBÜœBŸFšF 2©$¨B¨$¡x¸Ñ>Ü ¡T r T¡7°5Ñ9ô	;äœBŸFšF 2©$¨B¨$¡x¸ÀDÑIÜ ¡T r T¡7°5¸tÑDõ	Føô ó 	Ùð	ús   Â2
E Å
E,Å+E,c           	      ó¾  • U R                   n[        US'   UR                  SS5      n[        R                  " [        U5      5      R                  [        5      n[        R                  " X"R                  5      nXR                  S5      S S 2S 4   -
  R                  S5      n[        [        U5      [        R                  " U5      UR                  S   S-
  -  US-
  -  5        [        [        USS9[        R                  " USS9UR                  S   -  U-  5        [        R                  " UR                  U5      nXR                  S5      -
  R                  S5      n[        [        US	S
9[        R                  " US	S
9UR                  S   S-
  -  US-
  -  5        [        [        US	SS9[        R                  " US	SS9UR                  S   -  U-  5        g )NrH  r‰   rÎ   r@   r   r¡   T©ra  Fr[  r`  )ry   r	   r­   rS   Úlogical_notr   rƒ  rw   r   rï   rj  r�  r3   r   r   )rT   rÁ   ÚvalidÚfracÚxfs        rX   Útest_2d_with_missingÚTestCov.test_2d_with_missingl  s†  € à�I‰IˆÜˆˆ"‰Ø�I‰I�a˜‹OˆÜ—’œ|¨A›Ó/×6Ñ6´sÓ;ˆÜ�vŠv�eŸW™WÓ%ˆØ—&‘&˜“)šA˜t˜GÑ$Ñ$×,Ñ,¨QÓ/ˆÜœC ›FÜŸFšF 2›J¨!¯'©'°!©*°q©.Ñ9¸TÀB¹YÑGô	IäœC ¨Ñ-ÜŸFšF 2¨DÑ1°A·G±G¸A±JÑ>ÀÑEô	Gä�vŠv�e—g‘g˜uÓ%ˆØ—&‘&˜“)‰m×#Ñ# AÓ&ˆÜœC ¨%Ñ0ÜŸVšV B¨uÑ5ØŸg™g a™j¨1™nñ.Ø15¸±ñ<ô	>ô 	œC ¨%°dÑ;ÜŸVšV B¨u¸4Ñ@ØŸW™W Q™Zñ(Ø*.ñ/õ	1r[   )ry   N)r—   r˜   r™   rš   rX  r]  rb  re  rl  rt  r›   r_   r[   rX   rU  rU  +  s!   † ò.ò
ò(=ò=òFõ01r[   rU  c                   ó>   • \ rS rSrS rS rS rS rS rS r	S r
S	rg
)ÚTestCorrcoefi‚  c                 ó¸   • [        [        R                  R                  S5      5      U l        [        [        R                  R                  S5      5      U l        g rW  )r   rS   r3  r4  ry   Údata2rò  s    rX   rX  ÚTestCorrcoef.setup_method„  s4   € Üœ"Ÿ)™)Ÿ.™.¨Ó,Ó-ˆŒ	Üœ2Ÿ9™9Ÿ>™>¨"Ó-Ó.ˆ�
r[   c           	      ó@  • U R                   U R                  p![        R                  " U5      n[        R                  " X5      n[	        5        n[
        R                  " S5        [        [        [        USS9  UR                  [        S5        [        [        R                  " USS9[        USS95        [        [        USS9U5        [        [        XSS9U5        [        [        USS9U5        [        [        XSS9U5        S S S 5        g ! , (       d  f       g = f)NrF  rH  ©Úddofúbias and ddof have no effectr   r‰   ©ry   ry  rS   r   r7   rL  Úsimplefilterr6   r–  Úfilterr3   )rT   rÁ   rÂ   r  Ú	expected2Úsups         rX   Ú	test_ddofÚTestCorrcoef.test_ddofˆ  sÍ   € à�y‰y˜$Ÿ*™*ˆ1Ü—;’;˜q“>ˆÜ—K’K Ó%ˆ	ÜÔ  CÜ×!Ò! (Ô+ÜÔ+¬X°q¸rÒBØ�J‰JÔ)Ð+IÔJä¤§¢¨A°AÑ 6¼ÀÈÑ8KÔLÜ¤¨°Ñ 4°hÔ?Ü¤¨°BÑ 7¸ÔCÜ¤¨°Ñ 3°XÔ>Ü¤¨°AÑ 6¸	ÔB÷ !× Ö ús   ÁB8DÄ
Dc           	      óª  • U R                   U R                  p![        R                  " U5      n[	        5        n[
        R                  " S5        [        [        [        XSS5        [        [        [        XSS5        [        [        [        USS9  UR                  [        S5        [        [        USS9U5        S S S 5        g ! , (       d  f       g = f)NrF  TFro  r~  r@   r  )rT   rÁ   rÂ   r  rƒ  s        rX   Ú	test_biasÚTestCorrcoef.test_bias˜  s’   € Ø�y‰y˜$Ÿ*™*ˆ1Ü—;’;˜q“>ˆäÔ  CÜ×!Ò! (Ô+ÜÔ+¬X°q¸TÀ5ÔIÜÔ+¬X°q¸TÀ4ÔHÜÔ+¬X°q¸uÒEØ�J‰JÔ)Ð+IÔJä¤¨°Ñ 3°XÔ>÷ !× Ö ús   ¸BCÃ
Cc           
      ó‚  • U R                   n[        [        R                  " U5      [        U5      5        [        [        R                  " USS9[        USS95        [	        5        nUR                  [        S5        [        [        R                  " USSS9[        USSS95        S S S 5        g ! , (       d  f       g = f)NFr[  r~  Tr`  )ry   r3   rS   r   r7   r�  r–  ©rT   rÁ   rƒ  s      rX   rb  Ú$TestCorrcoef.test_1d_without_missing¥  sˆ   € à�I‰IˆÜœBŸKšK¨›N¬H°Q«KÔ8ÜœBŸKšK¨°%Ñ8Ü$ Q¨uÑ5ô	7äÔ  CØ�J‰JÔ)Ð+IÔJÜ¤§¢¨A°eÀ$Ñ GÜ (¨°5¸tÑ DôF÷ !× Ö ús   Á'A B0Â0
B>c           
      ó¢  • U R                   R                  SS5      n[        [        R                  " U5      [	        U5      5        [        [        R                  " USS9[	        USS95        [        5        nUR                  [        S5        [        [        R                  " USSS9[	        USSS95        S S S 5        g ! , (       d  f       g = f)Nr‰   rÎ   Fr[  r~  Tr`  )ry   r­   r3   rS   r   r7   r�  r–  rŠ  s      rX   re  Ú$TestCorrcoef.test_2d_without_missing°  s•   € à�I‰I×Ñ˜a Ó#ˆÜœBŸKšK¨›N¬H°Q«KÔ8ÜœBŸKšK¨°%Ñ8Ü$ Q¨uÑ5ô	7äÔ  CØ�J‰JÔ)Ð+IÔJÜ¤§¢¨A°eÀ$Ñ GÜ (¨°5¸tÑ DôF÷ !× Ö ús   Á7A C Ã 
Cc                 ó0  • U R                   n[        US'   XR                  5       -  nUR                  5       n[	        [
        R                  " U5      [        U5      5        [	        [
        R                  " USS9[        USS95        [        5        nUR                  [        S5        [	        [
        R                  " USSS9[        USSS95        S S S 5         [        USS9  USS n[	        [
        R                  " X"S S S2   5      [        XS S S2   5      5        [	        [
        R                  " X"S S S2   SS9[        XS S S2   SS95        [        5        nUR                  [        S5        [	        [
        R                  " X"S S S2   5      [        XS S S2   SS	95        [	        [
        R                  " X"S S S2   5      [        XS S S2   S
S95        S S S 5        g ! , (       d  f       GN= f! [         a     GNf = f! , (       d  f       g = f)NrH  Fr[  r~  Tr`  rh  r@   ro  r=   r|  )ry   r	   rj  r{   r3   rS   r   r7   r�  r–  rî   )rT   rÁ   rk  rƒ  s       rX   rl  Ú!TestCorrcoef.test_1d_with_missing»  sµ  € à�I‰IˆÜˆˆ"‰Ø	�V‰V‹X‰ˆØ�\‰\‹^ˆÜœBŸKšK¨›O¬X°a«[Ô9ÜœBŸKšK¨°5Ñ9Ü$ Q¨uÑ5ô	7äÔ  CØ�J‰JÔ)Ð+IÔJÜ¤§¢¨B°uÀ4Ñ HÜ (¨°5¸tÑ DôF÷ !ð	Ü�Q UÒ+ð ˆq�ˆWˆÜœBŸKšK¨©t°¨t©HÓ5´xÀÁTÀrÀTÁ7Ó7KÔLÜœBŸKšK¨©t°¨t©H¸UÑCÜ$ Q©$¨B¨$©¸Ñ>ô	@äÔ  CØ�J‰JÔ)Ð+IÔJä¤§¢¨B±4°R°4±Ó 9Ü (¨©d°¨d©G¸!Ñ <ô>ä¤§¢¨B±4°R°4±Ó 9Ü (¨©d°¨d©G¸!Ñ <ô>÷ !Ð ÷ !Ö ûô ó 	Úð	ú÷ !Õ ús,   ÂA G$Ã
G6 ÅA?HÇ$
G3Ç6
HÈHÈ
Hc                 ó(  • U R                   n[        US'   UR                  SS5      n[        U5      n[        R                  " U5      n[        US S2S S24   US S2S S24   5        [        5        nUR                  [        S5        [        [        USS9S S2S S24   US S2S S24   5        [        [        USS9S S2S S24   US S2S S24   5        [        [        USS9S S2S S24   US S2S S24   5        S S S 5        g ! , (       d  f       g = f)	NrH  r‰   rÎ   r~  rI  r|  r@   ro  )	ry   r	   r­   r   rS   r3   r7   r�  r–  )rT   rÁ   rU   rV   rƒ  s        rX   rt  Ú!TestCorrcoef.test_2d_with_missingÙ  s*  € à�I‰IˆÜˆˆ"‰Ø�I‰I�a˜‹Oˆä˜‹{ˆÜ—+’+˜a“.ˆÜ˜D  "  c r c ™N¨G°C°R°C¸¸"¸°HÑ,=Ô>ÜÔ  CØ�J‰JÔ)Ð+IÔJä¤¨°Ñ 4°S°b°S¸#¸2¸#°XÑ >Ø '¨¨¨¨S¨b¨S¨Ñ 1ô3ä¤¨°Ñ 3°C°R°C¸¸"¸°HÑ =Ø '¨¨¨¨S¨b¨S¨Ñ 1ô3ä¤¨°Ñ 3°C°R°C¸¸"¸°HÑ =Ø '¨¨¨¨S¨b¨S¨Ñ 1ô3÷ !× Ö ús   Á1B	DÄ
D)ry   ry  N)r—   r˜   r™   rš   rX  r„  r‡  rb  re  rl  rt  r›   r_   r[   rX   rw  rw  ‚  s(   † ò/òCò ?ò	Fò	Fò>õ<3r[   rw  c                   ó    • \ rS rSrS rS rSrg)ÚTestPolynomialií  c                 ó  • [         R                  R                  S5      n[         R                  R                  S5      R                  SS5      n[	        [        XS5      [         R
                  " XS5      5        UR                  [        5      n[        US'   UR                  [        5      n[        =US'   US'   [        XS S 2S4   SS	S
9u  p4pVn[         R
                  " USS  USS 2S4   R                  5       SS	S
9u  p‰p«n[        X4XVU4X‰X«U45       H  u  pÞ[	        XÞ5        M     [        XS S 2S4   SS	S
9u  p4pVn[         R
                  " USS USS2S4   SS	S
9u  p‰p«n[        X4XVU4X‰X«U45       H  u  pÞ[	        XÞ5        M     [        XSS	S
9u  p4pVn[         R
                  " USS USS2S S 24   SS	S
9u  p‰p«n[        X4XVU4X‰X«U45       H  u  pÞ[	        XÞ5        M     [         R                  R                  S5      S-   nUR                  5       nUSS nUSS nUSS n[        XSS	US9u  p4pVn[         R
                  " UUSS	US9u  p‰p«n[        UU5        [        X4XVU4X‰X«U45       H  u  pÞ[	        XÞ5        M     g )Nrp   rt  rH  r=   r‰   r   rI   ©rH  rH  T©rx   r@   )rx   rÜ   )rS   r3  r4  r­   r3   r+   Úviewr   r	   r{   Úzipr7  r5   )rT   rÁ   rÂ   ÚCÚRÚKÚSÚDr]   rÌ  r~   r€   r.  rC   Úa_rÜ   ÚwoÚxsÚysÚwss                       rX   Útest_polyfitÚTestPolynomial.test_polyfitï  sƒ  € ô �I‰I�N‰N˜2ÓˆÜ�I‰I�N‰N˜2Ó×&Ñ& r¨1Ó-ˆÜœG A¨!Ó,¬b¯jªj¸¸qÓ.AÔBà�F‰F”;ÓˆÜˆˆ!‰Ø�F‰F”;ÓˆÜ$Ð$ˆˆ$‰�!�F‘)ä! !¢q¨! t¡W¨a°dÑ;‰ˆˆq�QÜŸ*š* Q q r U¨A¨a©b°!¨e©H×,?Ñ,?Ó,AÀ1Ø*.ñ0‰ˆˆq�Qä˜A !¨˜?¨Q°1¸¨OÖ<‰GˆQÜ Ö&ñ =ô " !¢q¨" u¡X¨q°tÑ<‰ˆˆq�QÜŸ*š* Q q¨ W¨a°°"°°b°©k¸1À4ÑH‰ˆˆq�QÜ˜A !¨˜?¨Q°1¸¨OÖ<‰GˆQÜ Ö&ñ =ô " !¨°Ñ5‰ˆˆq�QÜŸ*š* Q q¨ W¨a°°"°²a°©j¸!À$ÑG‰ˆˆq�QÜ˜A !¨˜?¨Q°1¸¨OÖ<‰GˆQÜ Ö&ñ =ô �I‰I�N‰N˜2Ó Ñ"ˆØ�V‰V‹XˆØˆq�ˆWˆØˆq�ˆWˆØˆq�ˆWˆÜ! !¨°¸Ñ:‰ˆˆq�QÜŸ*š* R¨¨Q°T¸RÑ@‰ˆˆq�QÜ�Q˜ÔÜ˜A !¨˜?¨Q°1¸¨OÖ<‰GˆQÜ Ö&ò =r[   c                 ó  • [         R                  R                  S5      n[         R                  R                  S5      R                  SS5      n[         R                  US'   [         R                  US'   UR                  [        5      nUR                  [        5      n[        US'   [        US'   [        XSSS	9u  p4pVn[         R                  " US
S US
S2S S 24   SSS	9u  p‰p«n[        X4XVU4X‰X«U45       H  u  pÞ[        XÞ5        M     g )Nrp   rt  rH  r=   r   r•  r‰   Tr–  r@   )rS   r3  r4  r­   rë   r—  r   r	   r+   r˜  r3   )rT   rÁ   rÂ   r™  rš  r›  rœ  r�  r]   rÌ  r~   r€   r.  rC   rž  s                  rX   Útest_polyfit_with_masked_NaNsÚ,TestPolynomial.test_polyfit_with_masked_NaNs  sâ   € Ü�I‰I�N‰N˜2ÓˆÜ�I‰I�N‰N˜2Ó×&Ñ& r¨1Ó-ˆä�v‰vˆˆ!‰Ü—F‘Fˆˆ&‰	Ø�F‰F”;ÓˆØ�F‰F”;ÓˆÜˆˆ!‰Üˆˆ&‰	ä! !¨°Ñ5‰ˆˆq�QÜŸ*š* Q q¨ W¨a°°"°²a°©j¸!À$ÑG‰ˆˆq�QÜ˜A !¨˜?¨Q°1¸¨OÖ<‰GˆQÜ Ö&ò =r[   r_   N)r—   r˜   r™   rš   r£  r¦  r›   r_   r[   rX   r“  r“  í  s   † ò%'õN'r[   r“  c                   óz   • \ 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)ÚTestArraySetOpsi'  c                 óÔ   • / SQn[        USSS9n[        [        US   [        5      5        [	        US   [        / SQ/ SQS95        [	        US   / S	Q5        [	        US
   / SQ5        g )N©r@   r@   r@   r=   r=   r‰   T©Úreturn_indexÚreturn_inverser   r  r  r¤   r@   )r   r‰   r’   r=   )r   r   r   r@   r@   r=   )r0   r2   r-  r   r5   r
   rJ  s      rX   Útest_unique_onlistÚ"TestArraySetOps.test_unique_onlist)  sZ   € â!ˆÜ�d¨¸dÑCˆÜ”
˜4 ™7¤KÓ0Ô1Ü�T˜!‘Wœlª9º9ÑEÔFÜ�T˜!‘WšiÔ(Ü�T˜!‘WÒ0Õ1r[   c                 ól  • [        / SQ/ SQS9n[        USSS9n[        US   [        / SQ/ SQS95        [        US	   / S
Q5        [        US   / SQ5        SUl        [        / SQ/ SQSS9n[        USSS9n[        US   [        / SQ/ SQS95        [        US	   / S
Q5        [        US   / SQ5        g )Nr«  )r   r   r@   r   r@   r   r¤   Tr¬  r   )r@   r=   r‰   rH  ©r   r   r   r@   r@   )r   r‰   r’   r=   r=   )r   r   r‰   r@   r‰   r=   r‰   )ry   rB   Ú
fill_value)r
   r0   r5   r³  rJ  s      rX   Útest_unique_onmaskedarrayÚ)TestArraySetOps.test_unique_onmaskedarray2  sª   € äÒ.Ò5GÑHˆÜ�d¨¸dÑCˆÜ�T˜!‘Wœlª=º|ÑLÔMÜ�T˜!‘WšlÔ+Ü�T˜!‘WÒ0Ô1àˆŒÜÒ!3Ú!3ÀñCˆä�d¨¸dÑCˆÜ�T˜!‘Wœlª=º|ÑLÔMÜ�T˜!‘WšlÔ+Ü�T˜!‘WÒ0Õ1r[   c                 ó@  • [        / SQSS9n[        USSS9n[        US   [        S/S/S95        [        US   S/5        [        US   / SQ5        [        n[        USSS9n[        US   [        [        5      5        [        US   S/5        [        US   S/5        g )	NrY  Tr¤   r¬  r   r@   r=   r  )r
   r0   r5   r	   rJ  s      rX   Útest_unique_allmaskedÚ%TestArraySetOps.test_unique_allmaskedB  s›   € äšI¨DÑ1ˆÜ�d¨¸dÑCˆÜ�T˜!‘Wœl¨A¨5¸°vÑ>Ô?Ü�T˜!‘W˜q˜cÔ"Ü�T˜!‘WšiÔ(ô ˆÜ�d¨¸dÑCˆÜ�T˜!‘Wœl¬6Ó2Ô3Ü�T˜!‘W˜q˜cÔ"Ü�T˜!‘W˜q˜cÕ"r[   c                 ó   • [        [        R                  " S5      / SQS9n[        / SQ/ SQS9n[	        U5      n[        X25        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        g )Nr’   ©r@   r   r   r   r@   r¤   )r@   r@   r@   rÎ   )r@   r   r   r@   r   )r
   rS   r   r   r   r5   r�  rB   )rT   rÁ   rV   rU   s       rX   Útest_ediff1dÚTestArraySetOps.test_ediff1dQ  s`   € äœŸš 1›ªOÑ<ˆÜšª<Ñ8ˆÜ�q‹zˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Õ-r[   c                 ó  • [        [        R                  " S5      / SQS9n[        U[        S9n[        / SQ/ SQS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        [        U/ SQS9n[        / S	Q/ S
QS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        g )Nr’   rº  r¤   )Úto_begin)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
   rS   r   r   r	   r   r5   r�  rB   ©rT   rÁ   rU   rV   s       rX   Útest_ediff1d_tobeginÚ$TestArraySetOps.test_ediff1d_tobeginZ  s²   € äœŸš 1›ªOÑ<ˆÜ�q¤6Ñ*ˆÜšªoÑ>ˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Ô-ä�q¢9Ñ-ˆÜÒ-Ò4IÑJˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Õ-r[   c                 ó  • [        [        R                  " S5      / SQS9n[        U[        S9n[        / SQ/ SQS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        [        U/ SQS9n[        / S	Q/ S
QS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        g )Nr’   rº  r¤   )Úto_end)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À  s       rX   Útest_ediff1d_toendÚ"TestArraySetOps.test_ediff1d_toendi  s²   € äœŸš 1›ªOÑ<ˆÜ�q¤Ñ(ˆÜšªoÑ>ˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Ô-ä�q¢Ñ+ˆÜÒ-Ò4IÑJˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Õ-r[   c                 ó  • [        [        R                  " S5      / SQS9n[        U[        [        S9n[        / SQ/ SQS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        [        U/ SQ[        S9n[        / S	Q/ S
QS9n[        X#5        [        UR                  S5      UR                  S5      5        [        UR                  UR                  5        g )Nr’   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À  s       rX   Útest_ediff1d_tobegin_toendÚ*TestArraySetOps.test_ediff1d_tobegin_toendx  sº   € äœŸš 1›ªOÑ<ˆÜ�q¤´&Ñ9ˆÜÒ*Ò1CÑDˆÜ�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Ô-ä�q¢´VÑ<ˆÜÒ0Ú5ñ7ˆä�TÔ#Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Õ-r[   c                 óF  • [         R                  " S5      n[        U5      n[        / SQ/ SQS9n[	        X#5        [        [        U[        5      5        [	        UR                  S5      UR                  S5      5        [	        UR                  UR                  5        [        U[        [        S9n[        / SQ/ SQS9n[        [        U[        5      5        [	        UR                  S5      UR                  S5      5        [	        UR                  UR                  5        g )	Nr’   ©r@   r@   r@   r@   ©r   r   r   r   r¤   r   rÈ  r´   rµ   )rS   r   r   r   r5   r2   r-  r   r�  rB   r	   rÀ  s       rX   Útest_ediff1d_ndarrayÚ$TestArraySetOps.test_ediff1d_ndarrayˆ  sÀ   € ä�IŠI�a‹LˆÜ�q‹zˆÜšª<Ñ8ˆÜ�TÔ#Ü”
˜4¤Ó-Ô.Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Ô-ä�q¤´&Ñ9ˆÜÒ*Ò1CÑDˆÜ”
˜4¤Ó-Ô.Ü�T—[‘[ “^ W§^¡^°AÓ%6Ô7Ü�T—Y‘Y §¡Õ-r[   c                 ó„   • [        / SQ/ SQS9n[        / SQ/ SQS9n[        X5      n[        / SQ/ SQS9n[        X45        g )N)r@   r‰   r‰   r‰   r²  r¤   )r‰   r@   r@   r@   )r@   r‰   rH  rç   )r   r!   r5   )rT   rÁ   rÂ   rU   rV   s        rX   Útest_intersect1dÚ TestArraySetOps.test_intersect1d˜  s8   € ä’,¢\Ñ2ˆÜ’,¢\Ñ2ˆÜ˜1Ó ˆÜš
ªÑ3ˆÜ�TÕ#r[   c                 óJ  • [        / SQ/ SQS9n[        / SQ/ SQS9n[        X5      n[        U[        / SQ5      5        [        / SQ/ SQS9n/ SQn[        X5      n[        U[        / SQ/ S	QS95        [        / S
Q5      n[        / SQ5      n[        X5      n[        [	        U[
        5      5        [        U/ SQ5        [        / SQ/ SQS9n[        / SQ/ S	QS9n[        X5      n[        [	        U[
        5      5        [        U/ SQ5        [        / [        / / 5      5        g )N©r@   r=   r’   ru   rH  ©r   r   r   r   r@   r¤   ©r@   r=   r‰   rÎ   r’   rH  rœ  )r‰   rÎ   ru   ©r@   r=   r‰   rÎ   r’   ©r‰   rÎ   ru   rH  r²  r  )r†   r’   rÎ   r,  ©r@   r‡   r=   r‰   ©r   r@   r   r   ©r†   r’   rÎ   r‡   )r   r-   r5   r2   r-  r   r4   ©rT   rC   rD   rU   s       rX   Útest_setxor1dÚTestArraySetOps.test_setxor1d   sä   € äÒ"ªÑ9ˆÜÒ%Ò,>Ñ?ˆÜ˜‹~ˆÜ�Tœ5¢Ó+Ô,äÒ"ªÑ9ˆÚˆÜ˜‹~ˆÜ�Tœ5¢²\ÑBÔCä’)ÓˆÜ’)ÓˆÜ˜‹~ˆÜ”
˜4¤Ó-Ô.Ü�TÒ-Ô.ä’,¢\Ñ2ˆÜ’,¢\Ñ2ˆÜ˜‹~ˆÜ”
˜4¤Ó-Ô.Ü�TÒ-Ô.ä˜2œx¨¨BÓ/Õ0r[   c                 ó°  • [        / SQ/ SQS9n/ SQn[        XSS9n[        U[        / SQ/ SQS95        [        / S	Q/ S
QS9n[        / SQ/ SQS9n[        XSS9n[        [	        U[
        5      5        [        U/ SQ5        [        S/S/S/S//5      n[        SS/SS//5      n[        XSS9n[        [	        U[
        5      5        [        U/ SQ5        g )NrÔ  rÕ  r¤   r×  T)Úassume_uniquerØ  r²  rÙ  rÚ  rÛ  r,  r@   r‡   r=   r‰   r†   r’   rÎ   )r   r-   r5   r2   r-  r   rÜ  s       rX   Útest_setxor1d_uniqueÚ$TestArraySetOps.test_setxor1d_uniqueº  sÇ   € äÒ"ªÑ9ˆÚˆÜ˜¨DÑ1ˆÜ�Tœ5¢²\ÑBÔCä’,¢\Ñ2ˆÜ’,¢\Ñ2ˆÜ˜¨DÑ1ˆÜ”
˜4¤Ó-Ô.Ü�TÒ-Ô.ä�A�3˜˜˜a˜S 1 #Ð&Ó'ˆÜ�A�q�6˜A˜q˜6Ð"Ó#ˆÜ˜¨DÑ1ˆÜ”
˜4¤Ó-Ô.Ü�TÒ-Õ.r[   c                 óÌ  • [         R                  " S5      R                  / SQ5      n[         R                  " / SQ5      nSUS'   [	        XS9n[	        / SQ/ SQS9n[        S[
        S	9nS
US'   S
US'   S
US'   [        X5      n[        [        U[        5      5        [        XT5        [         R                  " XUR                  )    5      UR                  ) -  n[        XV5        g )NrF  rÚ   r@   )r@   r=   r   r¤   )	r   rp   rt  rv  r@   r‰   r  rO  rx  )	r   r@   r   r@   r   r@   r   r@   r   )ry   rB   r>   Tr  rç   )r   r=   r‰   )rS   r   r­   r   r   r”   r"   r2   r-  r   r4   rB   )rT   rC   rB   rD   Úecr]   r.  s          rX   Ú	test_isinÚTestArraySetOps.test_isinÍ  s¿   € ô �IŠI�b‹M×!Ñ!¢)Ó,ˆÜ�xŠxš	Ó"ˆØˆˆW‰Ü�!ÑˆÜÒ:Ú:ñ<ˆä�9¤DÑ)ˆØˆˆ7‰Øˆˆ7‰Øˆˆ7‰Ü�‹JˆÜ”
˜1œkÓ*Ô+Ü˜1Ô!ä�GŠG�A˜!Ÿ&™&˜‘zÓ" a§f¡f WÑ,ˆÜ˜1Õ r[   c                 ó  • [        / SQ/ SQS9n[        / SQ/ SQS9n[        X5      n[        U/ SQ5        [        / SQ/ SQS9n[        / SQ/ S	QS9n[        X5      n[        U/ S
Q5        [        / [        / / 5      5        g )NrÔ  rÕ  r¤   rÖ  rœ  )TTTFT©r’   r’   r=   r@   rH  ©r@   r’   rH  rç   )TTFTT)r   r    r5   r4   rÜ  s       rX   Ú	test_in1dÚTestArraySetOps.test_in1dâ  sn   € äÒ"ªÑ9ˆÜÒ%Ò,>Ñ?ˆÜ�A‹zˆÜ�TÒ:Ô;äÒ"ªÑ9ˆÜ’*¢9Ñ-ˆÜ�A‹zˆÜ�TÒ:Ô;ä˜2œt B¨›|Õ,r[   c           	      ób  • [        / SQ/ SQS9n[        / SQ/ SQS9n[        [        R                  " [	        X5      5      [	        XSS95        [        / SQ/ SQS9n[        / S	Q/ S
QS9n[        [        R                  " [	        X5      5      [	        XSS95        [        / [	        / / SS95        g )NrÔ  rÕ  r¤   rÖ  rœ  T)Úinvertrè  ré  rç   )r   r5   rS   rí  r    r4   r£  s      rX   Útest_in1d_invertÚ TestArraySetOps.test_in1d_invertð  s…   € äÒ"ªÑ9ˆÜÒ%Ò,>Ñ?ˆÜ”R—Y’Yœt A›zÓ*¬D°¸dÑ,CÔDäÒ"ªÑ9ˆÜ’*¢9Ñ-ˆÜ”R—Y’Yœt A›zÓ*¬D°¸dÑ,CÔDä˜2œt B¨°4Ñ8Õ9r[   c                 ó@  • [        / SQ/ SQS9n[        / SQ/ SQS9n[        X5      n[        / SQ/ SQS9n[        X45        [        / SQ/ SQ// S	Q/ S
Q/S9n[        / SQ/ SQS9n[        / SQ/ SQS9n[        XV5      n[        X‡5        [        / [        / / 5      5        g )N)r@   r=   r’   ru   r’   rH  rœ  r¤   rÖ  )r@   r=   r‰   rÎ   r’   ru   rH  )r   r   r   r   r   r   r@   rê   rX  r  rç   rG  rÕ  )r   r@   r=   r‰   rÎ   r’   )r   r/   r5   r4   )	rT   rC   rD   rU   rV   rÁ   rÂ   ÚezrÈ   s	            rX   Útest_union1dÚTestArraySetOps.test_union1dü  s�   € äÒ%Ò,>Ñ?ˆÜÒ%Ò,>Ñ?ˆÜ�q‹}ˆÜÒ.Ò5JÑKˆÜ�TÔ#ô ’9šiÐ(²	º9Ð/EÑFˆÜ’/ªÑ8ˆÜÒ%Ò,>Ñ?ˆÜ�A‹MˆÜ�QÔä˜2œw r¨2›Õ/r[   c                 óz  • [        / SQ/ SQS9n[        / SQ5      n[        X5      n[        U[        / SQ/ SQS95        [        S5      n[        S5      n[        [        X5      [        SS	/5      5        [        / [        R
                  / S9n[        [        U/ 5      R                  [        R
                  5        g )
N)r†   r’   rÎ   ru   ru   r@   r=   r@   )r   r   r   r   r   r   r   r@   r¤   )r=   rÎ   r‰   r‰   r=   r@   r’   )r†   ru   rH  rç   rp   r‡   rˆ   )r   r,   r5   r   rS   Úuint32r?   rÜ  s       rX   Útest_setdiff1dÚTestArraySetOps.test_setdiff1d  sŠ   € äÒ*Ò1IÑJˆÜÒ'Ó(ˆÜ˜‹ˆÜ�Tœ5¢²)Ñ<Ô=ä�2‹JˆÜ�1‹IˆÜ”Y˜q“_¤e¨Q°¨F£mÔ4Ü�"”b—i‘i bÑ)ˆÜ”Y˜q "Ó%×+Ñ+¬R¯Y©YÕ7r[   c                 ó¸   • [         R                  " / SQ5      n[         R                  " / SQ5      n[        [        X5      [         R                  " S/5      5        g )N)rC   rD   r]   )rC   rD   r€   r]   )rS   r   r4   r,   r£  s      rX   Útest_setdiff1d_char_arrayÚ)TestArraySetOps.test_setdiff1d_char_array  s6   € ä�HŠH’_Ó%ˆÜ�HŠH’_Ó%ˆÜœ9 Q›?¬B¯HªH°c°U«OÕ<r[   r_   N)r—   r˜   r™   rš   r¯  r´  r·  r»  rÁ  rÅ  rÉ  rÎ  rÑ  rÝ  rá  rå  rê  rî  rò  rö  rù  r›   r_   r[   rX   r©  r©  '  sW   † ò2ò2ò #ò.ò.ò.ò.ò .ò $ò1ò4/ò&!ò*-ò
:ò0ò$8õ=r[   r©  c                   ó    • \ rS rSrS rS rSrg)ÚTestShapeBasei"  c                 óÆ  • [        / SQ/ SQS9n[        U5      n[        UR                  S5        [        UR                  R                  UR
                  R                  5        [        UR                  S5        [        UR                  R                  UR
                  R                  5        [        UR                  R                  UR
                  R                  5        g )Nrê   rŠ  r¤   r6  r0  )r
   r   r5   r   rB   ry   r£  s      rX   Útest_atleast_2dÚTestShapeBase.test_atleast_2d$  s�   € äšªÑ3ˆÜ�q‹MˆÜ�Q—W‘W˜fÔ%Ü�Q—V‘V—\‘\ 1§6¡6§<¡<Ô0Ü�Q—W‘W˜dÔ#Ü�Q—V‘V—\‘\ 1§6¡6§<¡<Ô0Ü�Q—V‘V—\‘\ 1§6¡6§<¡<Õ0r[   c                 óf  • [        S5      n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        [        SS5      nU Hm  n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        Mo     [        S5      n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        [        SS5      nU Hm  n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        Mo     [        S5      n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        [        SS5      nU Hm  n[        UR                  S5        [        UR                  R                  UR                  5        [        UR                  R                  UR                  5        Mo     [        S5      n[        UR                  S5        [        UR                  R                  UR                  R                  5        g )Nr¡   r^   r¢   rJ   rY  )r   r5   r   rB   ry   r   r   r   )rT   rD   rC   s      rX   Útest_shape_scalarÚTestShapeBase.test_shape_scalar.  sô  € ô
 �s‹OˆÜ�Q—W‘W˜dÔ#Ü�Q—V‘V—\‘\ 1§7¡7Ô+Ü�Q—V‘V—\‘\ 1§7¡7Ô+ä�s˜CÓ ˆÛˆAÜ˜Ÿ™ $Ô'Ü˜Ÿ™Ÿ™ q§w¡wÔ/Ü˜Ÿ™Ÿ™ q§w¡wÖ/ñ ô
 �s‹OˆÜ�Q—W‘W˜fÔ%Ü�Q—V‘V—\‘\ 1§7¡7Ô+Ü�Q—V‘V—\‘\ 1§7¡7Ô+ä�s˜CÓ ˆÛˆAÜ˜Ÿ™ &Ô)Ü˜Ÿ™Ÿ™ q§w¡wÔ/Ü˜Ÿ™Ÿ™ q§w¡wÖ/ñ ô
 �s‹OˆÜ�Q—W‘W˜iÔ(Ü�Q—V‘V—\‘\ 1§7¡7Ô+Ü�Q—V‘V—\‘\ 1§7¡7Ô+ä�s˜CÓ ˆÛˆAÜ˜Ÿ™ )Ô,Ü˜Ÿ™Ÿ™ q§w¡wÔ/Ü˜Ÿ™Ÿ™ q§w¡wÖ/ñ ô
 �S‹MˆÜ�Q—W‘W˜fÔ%Ü�Q—V‘V—\‘\ 1§6¡6§<¡<Õ0r[   r_   N)r—   r˜   r™   rš   rþ  r  r›   r_   r[   rX   rü  rü  "  s   † ò1õ(1r[   rü  c                   ó&   • \ rS rSrS rS rS rSrg)ÚTestNDEnumerateiY  c           
      óÆ  • [         R                  " S5      R                  S5      n[         R                  " U[        S9n[        XS9n[        [        [         R                  " U5      5      [        [        U5      5      5        [        [        [        U5      5      [        [        U5      5      5        [        [        [        U5      5      [        [        USS95      5        g )Nr  )r@   r‰   r=   r>   r¤   F©r{   )	rS   r   r­   Ú
zeros_liker”   r
   r5   r‚  r(   )rT   ÚordinaryÚ
empty_maskÚ	with_masks       rX   Útest_ndenumerate_nomaskedÚ)TestNDEnumerate.test_ndenumerate_nomasked[  sž   € Ü—9’9˜R“=×(Ñ(¨Ó3ˆÜ—]’] 8´4Ñ8ˆ
Ü  Ñ;ˆ	Ü”Tœ"Ÿ.š.¨Ó2Ó3Üœ+ hÓ/Ó0ô	2ä”Tœ+ hÓ/Ó0Üœ+ iÓ0Ó1ô	3ä”Tœ+ iÓ0Ó1Üœ+ i¸EÑBÓCõ	Er[   c           
      ó*  • [        S5      n[        S5      n[        S5      n[        [        [        U5      5      / 5        [        [        [        U5      5      / 5        [        [        [        USS95      [        [	        [
        R                  " S5      S[        /-  5      5      5        [        [        [        U5      5      / 5        [        [        [        USS95      [        [	        [
        R                  " S5      S[        /-  5      5      5        g )Nr_   )éd   rÚ   Fr  r  rF  )r$   r5   r‚  r(   r˜  rS   Úndindexr	   )rT   rC   rD   r]   s       rX   Útest_ndenumerate_allmaskedÚ*TestNDEnumerate.test_ndenumerate_allmaskedf  sÃ   € Ü�r‹NˆÜ�vÓˆÜ�yÓ!ˆÜ”Tœ+ a›.Ó)¨2Ô.Ü”Tœ+ a›.Ó)¨2Ô.Ü”Tœ+ a°EÑ:Ó;Üœ#œbŸjšj¨Ó0°#¼¸±.ÓAÓBô	Dä”Tœ+ a›.Ó)¨2Ô.Ü”Tœ+ a°EÑ:Ó;Üœ#œbŸjšj¨Ó3°YÄ&ÀÑ5IÓJÓKõ	Mr[   c                 óV  • [        [        R                  " S5      R                  S5      / SQ/ SQ/ SQ/S9n/ SQn[	        [        [        U5      5      U5        [	        [        [        [        USS	95      5      UR                  5        [        USS	9 H  u  p4[	        X   U5        M     g )
Nrâ   r.  rÌ  )r@   r@   r   r@   rÍ  r¤   ))rè   r†   ))r=   r   r‡   )r/  rˆ   )rK   rp   )r5  r  Fr  )	r
   rS   r   r­   r5   r‚  r(   re   r“   )rT   rC   ÚitemsÚ
coordinateÚvalues        rX   Útest_ndenumerate_mixedmaskedÚ,TestNDEnumerate.test_ndenumerate_mixedmaskedr  sŠ   € ÜœŸš 2›×.Ñ.¨vÓ6Ú+Ú+Ú+ð-ñ.ˆòGˆä”Tœ+ a›.Ó)¨5Ô1Ü”Sœœk¨!¸Ñ>Ó?Ó@À!Ç&Á&ÔIÜ!,¨Q¸5Ô!AÑˆJÜ˜™¨Ö.ò "Br[   r_   N)r—   r˜   r™   rš   r  r  r  r›   r_   r[   rX   r  r  Y  s   † ò	Eò
Mõ
/r[   r  c                   ó&   • \ rS rSrS rS rS rSrg)Ú	TestStacki  c                 ó€  • [        / SQ/ SQS9n[        / SQ/ SQS9n[        X/SS9n[        UR                  S5        [	        UR
                  US   R
                  5        [	        UR
                  US	   R
                  5        [        X/5      n[	        UR                  UR                  5        [	        UR
                  UR
                  5        [        X/S	S9n[        UR                  S
5        [	        UR
                  US S 2S4   R
                  5        [	        UR
                  US S 2S	4   R
                  5        g )Nrê   rŠ  r¤   ©rˆ   r‡   ru   r  r   r¥   r5  r@   ©r‰   r=   ©r
   r.   r5   r   r4   rB   r1   ry   ©rT   rC   rD   r]   r.  s        rX   Útest_stack_1dÚTestStack.test_stack_1d�  sá   € ÜšªÑ3ˆÜšªÑ3ˆä�1�&˜qÑ!ˆÜ�Q—W‘W˜fÔ%Ü˜1Ÿ6™6 1 Q¡4§9¡9Ô-Ü˜1Ÿ6™6 1 Q¡4§9¡9Ô-ä�A�6‹NˆÜ˜1Ÿ6™6 1§6¡6Ô*Ü˜1Ÿ6™6 1§6¡6Ô*ä�1�&˜qÑ!ˆÜ�Q—W‘W˜fÔ%Ü˜1Ÿ6™6 1¢Q¨ T¡7§<¡<Ô0Ü˜1Ÿ6™6 1¢Q¨ T¡7§<¡<Õ0r[   c                 óx  • [        / SQSS9n[        / SQSS9n[        X/SS9n[        UR                  S5        [	        UR
                  US   R
                  5        [	        UR
                  US	   R
                  5        [        X/5      n[	        UR                  UR                  5        [	        UR
                  UR
                  5        [        X/S	S9n[        UR                  S
5        [	        UR
                  US S 2S4   R
                  5        [	        UR
                  US S 2S	4   R
                  5        g )Nrê   Tr¤   r  Fr   r¥   r5  r@   r  r  r  s        rX   Útest_stack_masksÚTestStack.test_stack_masks“  sá   € Üš¨Ñ.ˆÜš¨Ñ/ˆä�1�&˜qÑ!ˆÜ�Q—W‘W˜fÔ%Ü˜1Ÿ6™6 1 Q¡4§9¡9Ô-Ü˜1Ÿ6™6 1 Q¡4§9¡9Ô-ä�A�6‹NˆÜ˜1Ÿ6™6 1§6¡6Ô*Ü˜1Ÿ6™6 1§6¡6Ô*ä�1�&˜qÑ!ˆÜ�Q—W‘W˜fÔ%Ü˜1Ÿ6™6 1¢Q¨ T¡7§<¡<Ô0Ü˜1Ÿ6™6 1¢Q¨ T¡7§<¡<Õ0r[   c                 ó\  • Sn[         R                  R                  SSU5      n[         R                  R                  SSU5      n[         R                  R                  SSU5      R                  [        5      n[         R                  R                  SSU5      R                  [        5      n[        X$S9n[        X5S9n[        Xg/SS9nSU-   n	[        UR                  U	5        [        UR                  US   R                  5        [        UR                  US   R                  5        [        Xg/S	S9nUS-   n	[        UR                  U	5        [        UR                  US
   R                  5        [        UR                  US   R                  5        Sn[         R                  R                  SSU5      n[         R                  R                  SSU5      n[         R                  R                  SSU5      R                  [        5      n[         R                  R                  SSU5      R                  [        5      n[        X$S9n[        X5S9n[        Xg/SS9nSU-   n	[        UR                  U	5        [        UR                  US   R                  5        [        UR                  US   R                  5        [        Xg/S	S9nUS-   n	[        UR                  U	5        [        UR                  US
   R                  5        [        UR                  US   R                  5        g )Nr  r   rp   r=   r¤   r¥   r<   r@   rH  ).r   ).r@   )r‰   r=   rÎ   r’   )rS   r3  Úrandintrƒ  r”   r
   r.   r5   r   r4   rB   )
rT   ÚshpÚd1Úd2rÃ   rÄ   Úa1r„  r]   Úc_shps
             rX   Útest_stack_ndÚTestStack.test_stack_nd¥  sf  € àˆÜ�Y‰Y×Ñ˜q " cÓ*ˆÜ�Y‰Y×Ñ˜q " cÓ*ˆÜ�Y‰Y×Ñ˜q ! SÓ)×0Ñ0´Ó6ˆÜ�Y‰Y×Ñ˜q ! SÓ)×0Ñ0´Ó6ˆÜ˜"Ñ&ˆÜ˜"Ñ&ˆä�2�( Ñ#ˆØ�s‘
ˆÜ�Q—W‘W˜eÔ$Ü˜2Ÿ7™7 A a¡D§I¡IÔ.Ü˜2Ÿ7™7 A a¡D§I¡IÔ.ä�2�( Ñ$ˆØ�d‘
ˆÜ�Q—W‘W˜eÔ$Ü˜2Ÿ7™7 A f¡I§N¡NÔ3Ü˜2Ÿ7™7 A f¡I§N¡NÔ3ð ˆÜ�Y‰Y×Ñ˜q " cÓ*ˆÜ�Y‰Y×Ñ˜q " cÓ*ˆÜ�Y‰Y×Ñ˜q ! SÓ)×0Ñ0´Ó6ˆÜ�Y‰Y×Ñ˜q ! SÓ)×0Ñ0´Ó6ˆÜ˜"Ñ&ˆÜ˜"Ñ&ˆä�2�( Ñ#ˆØ�s‘
ˆÜ�Q—W‘W˜eÔ$Ü˜2Ÿ7™7 A a¡D§I¡IÔ.Ü˜2Ÿ7™7 A a¡D§I¡IÔ.ä�2�( Ñ$ˆØ�d‘
ˆÜ�Q—W‘W˜eÔ$Ü˜2Ÿ7™7 A f¡I§N¡NÔ3Ü˜2Ÿ7™7 A f¡I§N¡NÕ3r[   r_   N)r—   r˜   r™   rš   r  r"  r+  r›   r_   r[   rX   r  r    s   † ò1ò$1õ$)4r[   r  )OÚ__doc__rß  rL  rì   ÚnumpyrS   Únumpy._core.numericr   Únumpy.ma.corer   r   r   r   r   r	   r
   r   r   r   r   Únumpy.ma.extrasr   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   Únumpy.ma.testutilsr2   r3   r4   r5   Únumpy.testingr6   r7   r9   r�   r*  rC  rU  r®  rÀ  rÈ  rU  rw  r“  r©  rü  r  r  r_   r[   rX   Ú<module>r4     s%  ðñó Û ã ã Ý 4÷÷ ÷ ñ ÷$÷ $÷ $÷ $÷ $÷ $÷ $÷ $÷ $ñ $÷Jó ÷ :÷ñ ÷D~9ñ ~9÷B).ñ ).÷XCñ C÷Loñ o÷d	,ñ ,÷,
!ñ 
!÷T=ñ T=÷nT1ñ T1÷nh3ñ h3÷V7'ñ 7'÷tx=ñ x=÷v41ñ 41÷n#/ñ #/÷LO4ò O4r[   