Ë
    0Ò^jB  ã                   ó  — d dl mZ d dlZd dlmZmZmZ d dlZd dl	Z	d dl
mZmZmZ d ej                  ej                  «      j                   z  Z ej$                  ddd¬	«      Z ej(                  d
d dd¬«      Z ej,                  eddd…    d geg«      Zej.                  j1                  deeg«      ej.                  j1                  dej2                  ej                  ej4                  ej6                  g«      d„ «       «       Zej.                  j1                  deeg«      ej.                  j1                  dej:                  ej<                  ej>                  g«      d„ «       «       Z  G d„ d«      Z! G d„ d«      Z" G d„ d«      Z#y)é    )Ú	factorialN)Úassert_allcloseÚassert_equalÚassert_array_less)ÚAAAÚFloaterHormannInterpolatorÚBarycentricInterpolatorg     ˆÃ@éÿÿÿÿé   éè  ©Únuméñÿÿÿé
   éô  )Úbaser   ÚmethodÚdtypec                 ó  — t        j                  |«      j                  dz  dz  }| t        u r|dz  }t         j                  j                  d«      }t        j                  dd|¬«      } | |t        j                  |«      «      }|j                  ddd¬«      j                  |«      }t         ||«      t        j                  |«      |¬«        ||«      j                  |k(  sJ ‚| t        u re|j                  j                  |k(  sJ ‚|j                  j                  |k(  sJ ‚|j                  j                  |j                   j                  k(  sJ ‚|j"                  j                  |k(  sJ ‚|j%                  «       j                  t        j&                  |d	«      k(  sJ ‚|j)                  «       j                  t        j&                  |d	«      k(  sJ ‚|j+                  «       j                  t        j&                  |d	«      k(  sJ ‚y )
Ng      è?éd   l   ´d³�w¤r
   r   ©r   ©Úsize©Úrtolù              ð?)ÚnpÚfinfoÚepsr   ÚrandomÚdefault_rngÚlinspaceÚsinÚuniformÚastyper   r   r   Úsupport_pointsÚsupport_valuesÚerrorsÚrealÚweightsÚpolesÚresult_typeÚresiduesÚroots)r   r   r   ÚrngÚzÚrÚz2s          úu/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/scipy/interpolate/tests/test_bary_rational.pyÚtest_dtype_preservationr4   (   s•  € ô �8‰8�E‹?×Ñ $Ñ&¨Ñ,€DØÔ+Ñ+Ø�‰ˆÜ
�)‰)×
Ñ
Ð 1Ó
2€Cä
�‰�B˜ Ô'€AÙˆq”"—&‘&˜“)Ó€Aà	�‰�R˜ ˆÓ	%×	,Ñ	,¨UÓ	3€BÜ‘A�b“Eœ2Ÿ6™6 "›:¨DÕ1ÙˆR‹5�;‰;˜%ÒÐÐà”�}Ø×Ñ×%Ñ%¨Ò.Ð.Ð.Ø×Ñ×%Ñ%¨Ò.Ð.Ð.Ø�x‰x�~‰~ §¡§¡Ò-Ð-Ð-Ø�9‰9�?‰?˜eÒ#Ð#Ð#Ø�7‰7‹9�?‰?œbŸn™n¨U°BÓ7Ò7Ð7Ð7Ø�:‰:‹<×Ñ¤§¡°°rÓ!:Ò:Ð:Ð:Ø�7‰7‹9�?‰?œbŸn™n¨U°BÓ7Ò7Ð7Ñ7ó    c                 óT  — t        j                  d|¬«      } | ||«      }|j                  j                  t        j                  |d«      k(  sJ ‚| t
        u r�|j                  j                  t        j                  |d«      k(  sJ ‚|j                  j                  t        j                  |d«      k(  sJ ‚|j                  j                  t        j                  |d«      k(  sJ ‚|j                  «       j                  t        j                  |d«      k(  sJ ‚|j                  «       j                  t        j                  |d«      k(  sJ ‚|j                  «       j                  t        j                  |d«      k(  sJ ‚ ||«      j                  t        j                  |d«      k(  sJ ‚y )Nr   r   ç      ð?r   )r   Úaranger*   r   r,   r   r&   r'   r(   r+   r-   r.   )r   r   r0   r1   s       r3   Útest_integer_promotionr9   A   s>  € ô 	�	‰	�"˜EÔ"€AÙˆq�!‹€AØ�9‰9�?‰?œbŸn™n¨U°CÓ8Ò8Ð8Ð8Ø”�}Ø×Ñ×%Ñ%¬¯©¸¸sÓ)CÒCÐCÐCØ×Ñ×%Ñ%¬¯©¸¸sÓ)CÒCÐCÐCØ�x‰x�~‰~¤§¡°°sÓ!;Ò;Ð;Ð;Ø�7‰7‹9�?‰?œbŸn™n¨U°BÓ7Ò7Ð7Ð7Ø�:‰:‹<×Ñ¤§¡°°rÓ!:Ò:Ð:Ð:Ø�7‰7‹9�?‰?œbŸn™n¨U°BÓ7Ò7Ð7Ð7áˆQ‹4�:‰:œŸ™¨¨sÓ3Ò3Ð3Ñ3r5   c                   óR  — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zd
„ Zej                  j                  dd„ ddfd„ ddfd„ ddfd„ ddfd„ ddfd„ ddfd„ ddfg«      d„ «       Zd„ Zej                  j                  dd„ d„ d „ d!„ d"„ d#„ d$„ d%„ d&„ g	«      d'„ «       Zd(„ Zd)„ Zd*„ Zy+),ÚTestAAAc                 óž  — t        j                  t        d¬«      5  t        dgddg«       d d d «       t        j                  t        d¬«      5  t        dgdggdgdgg«       d d d «       t        j                  t        d¬«      5  t        t        j
                  gdg«       d d d «       t        j                  t        «      5  t        dgdgd¬«       d d d «       t        j                  t        d	¬«      5  t        dgdgd
¬«       d d d «       y # 1 sw Y   ŒèxY w# 1 sw Y   Œ¼xY w# 1 sw Y   ŒˆxY w# 1 sw Y   ŒbxY w# 1 sw Y   y xY w)Nú `x` be of size 2 but got size 1.©Úmatchr   r   z1-DÚfiniter7   ©Ú	max_termsÚgreaterr
   )ÚpytestÚraisesÚ
ValueErrorr   r   ÚinfÚ	TypeError©Úselfs    r3   Útest_input_validationzTestAAA.test_input_validationS   s  € Ü�]‰]œ:Ð-OÔPñ 	Ü���a˜�VÔ÷	ä�]‰]œ:¨UÔ3ñ 	(Ü�!��q�c�
˜a˜S 1 #˜JÔ'÷	(ä�]‰]œ:¨XÔ6ñ 	Ü”—‘�˜1˜#Ô÷	ä�]‰]œ9Ó%ñ 	)Ü���a�S CÕ(÷	)ä�]‰]œ:¨YÔ7ñ 	(Ü���a�S BÕ'÷	(ð 	(÷	ð 	ú÷	(ð 	(ú÷	ð 	ú÷	)ð 	)ú÷	(ð 	(ús;   œDÁDÂD+ÃD7Ã9EÄDÄD(Ä+D4Ä7E ÅEc                 ó¶   — t        j                  t        d¬«      5  t        t        t        j                  t        «      d¬«       d d d «       y # 1 sw Y   y xY w)Nz
AAA failedr>   r   rA   )rD   ÚwarnsÚRuntimeWarningr   ÚUNIT_INTERVALr   ÚexprI   s    r3   Útest_convergence_errorzTestAAA.test_convergence_error_   s>   € Ü�\‰\œ.°Ô=ñ 	DÜ”œrŸv™v¤mÓ4ÀÕC÷	D÷ 	Dñ 	Dús   œ*AÁAc                 óÆ  — t        j                  t        «      }t        t        |«      }t	         |t        «      |t
        ¬«       t         |t         j                  «      t         j                  «       t        j                   |t         j                  «      «      sJ ‚|j                  j                  }t        t        |d¬«      }|j                  j                  |k  sJ ‚y )N©Úatolgü©ñÒMbP?r   )r   rP   rO   r   r   ÚTOLr   ÚnanÚisfiniterG   r&   r   )rJ   Úfr1   Úm1s       r3   Útest_expzTestAAA.test_expe   s–   € Ü�F‰F”=Ó!ˆÜ”˜qÓ!ˆä™œ-Ó(¨!´#Õ6Ü‘Q”r—v‘v“Y¤§¡Ô'Ü�{‰{™1œRŸV™V›9Ô%Ð%Ð%à×Ñ×"Ñ"ˆÜ”˜q tÔ,ˆØ×Ñ×$Ñ$ rÒ)Ð)Ñ)r5   c                 óD  — t        j                  t         j                  t        z  «      }t	        t        |«      }t         |t        «      |dt        z  d¬«       t        t        j                  t        j                  |j                  «       «      «      dd¬«       t        t        j                  t        j                  |j                  «       dz
  «      «      dt        ¬«       t        j                  t        j                  |j                  «       «      «      dkD  sJ ‚y )	Nr   g³zê]Ê‚>©rT   r   r   gLÎaã§�ô=rS   ç      à?ç‚vIhÂ%<=)r   ÚtanÚpirO   r   r   rU   ÚminÚabsr.   r+   r-   ©rJ   rX   r1   s      r3   Útest_tanzTestAAA.test_tanq   s®   € Ü�F‰F”2—5‘5œ=Ñ(Ó)ˆÜ”˜qÓ!ˆä™œ-Ó(¨!°"´s±(ÀÕHÜœŸ™œrŸv™v a§g¡g£iÓ0Ó1°1¸5ÕAÜœŸ™œrŸv™v a§g¡g£i°#¡oÓ6Ó7¸ÄÕEä�v‰v”b—f‘f˜QŸZ™Z›\Ó*Ó+¨eÒ3Ð3Ñ3r5   c                 óì  — t        j                  ddg«      }t        j                  ddg«      }t        ||d¬«      }t         ||«      |t        ¬«       t        |j                  «       d«       t        |j                  «       d«       t        |j                  «       d	«       t        |j                  |«       t        |j                  |«       t        |j                  d
d
g«       t        |j                  ddg«       t        j                  g d¢«      }t        j                  g d¢«      }t        ||d¬«      }t         ||«      |t        ¬«       t        t        j                  |j                  «       «      t        j                  ddg«      «       t        t        j                  |j                  «       «      t        j                  ddg«      «       t        t        j                  |j                  «       «      t        j                  ddg«      «       t        |j                  |«       t        |j                  |«       t        |j                  g d¢«       t        |j                  g d¢«       y )Nr   r   é   r^   r   rS   r]   g      Ð?gUUUUUUÕ?gÈ;fž æ?)r   r   rf   )r   r   r   g�È¢Ó<ù?gÃ™ßt±Û?g‚fêMv²¿g¯»ÚÁ7ÓÐ¿)ç3�E§yâ?rg   rg   )r   r   r   )r   Úarrayr   r   rU   r+   r-   r.   r   r&   r'   r*   r(   Úsort)rJ   r0   rX   r1   s       r3   Útest_short_caseszTestAAA.test_short_cases{   s¹  € ô �H‰H�a˜�VÓˆÜ�H‰H�a˜�VÓˆÜ��1˜5Ô!ˆÜ™˜!›˜a¤cÕ*Ü˜Ÿ™›	 3Ô'Ü˜Ÿ
™
› dÔ+Ü˜Ÿ™›	 3Ô'Ü�Q×%Ñ% qÔ)Ü�Q×%Ñ% qÔ)Ü˜Ÿ	™	Ð$5Ð7HÐ#IÔJÜ�Q—X‘X  1˜vÔ&ô �H‰H’YÓˆÜ�H‰H’YÓˆÜ��1˜5Ô!ˆÜ™˜!›˜a¤cÕ*ÜœŸ™ §¡£	Ó*ÜŸ™Ð!2Ð4EÐ FÓGô	IäœŸ™ §
¡
£Ó-ÜŸ™Ð!3Ð5GÐ HÓIô	KäœŸ™ §¡£	Ó*¬B¯G©G°Q¸°F«OÔ<Ü�Q×%Ñ% qÔ)Ü�Q×%Ñ% qÔ)Ü˜Ÿ	™	ò $7ô 	8ä�Q—X‘XšyÕ)r5   c                 ó   — t        j                  dd«      }t        j                  |«      dz  }t        ||«      }t        |ddz  |z  j	                  t         j
                  «      «      }t        |d|z  j	                  t         j
                  «      «      }t         |d«      d |d«      z  «       t         |d«      ddz   |d«      z  «       y )	Ng333333Ó?g      ø?ù      ð?      ð?rf   i7  g      €,y        š™™™™™É?gffffffö?)r   r"   rP   r   r%   Ú
complex128r   )rJ   r0   rX   Úr1Úr2Úr3s         r3   Útest_scale_invariancezTestAAA.test_scale_invarianceœ   s£   € Ü�K‰K˜˜SÓ!ˆÜ�F‰F�1‹I˜Ñ ˆÜ��A‹YˆÜ��Q˜‘V˜a‘Z×'Ñ'¬¯©Ó6Ó7ˆÜ��W˜q‘[×(Ñ(¬¯©Ó7Ó8ˆÜ‘R˜“X˜w©¨D«Ñ1Ô2Ü‘R˜“W˜a ™f¡r¨#£wÑ.Õ/r5   c                 óô   — t         j                  j                  d«      }|j                  d«      d|j                  d«      z  z   }d„ }t	        | ||«      «      }t         |d«       |d«      t        ¬«       y )Nl   (pg98B i'  y              @c                 óD   — t        j                  d| z
  «      d| dz  z   z  S )Né   r   rf   )r   Úlog©r0   s    r3   rX   z TestAAA.test_log_func.<locals>.f©   s!   € Ü—6‘6˜!˜a™%“= A¨¨1©¡HÑ-Ð-r5   r   rS   )r   r    r!   Ústandard_normalr   r   rU   )rJ   r/   r0   rX   r1   s        r3   Útest_log_funczTestAAA.test_log_func¥   sf   € Ü�i‰i×#Ñ#Ð$7Ó8ˆØ×Ñ Ó&¨¨c×.AÑ.AÀ%Ó.HÑ)HÑHˆò	.ô �‘1�Q“4‹LˆÜ™˜!›™a ›d¬Ö-r5   c                 óä   — t        j                  dd«      }t        |t        j                  j                  |«      «      }t         |d«      t        j                  j                  d«      d¬«       y )Nr
   r   g)\�Âõ(ä?çVçž¯Ò<rS   )r   r"   r   ÚscipyÚspecialÚgammar   ©rJ   r0   r1   s      r3   Útest_infinite_datazTestAAA.test_infinite_data¯   sL   € Ü�K‰K˜˜AÓˆÜ�”5—=‘=×&Ñ& qÓ)Ó*ˆÜ™˜$›¤§¡×!4Ñ!4°TÓ!:ÀÖGr5   c                 ó"  — t        j                  dd«      }t        j                  d¬«      5  t        j                  |«      |z  }d d d «       t	        |«      }t         |d«      t        j                  d«      dz  d¬«       y # 1 sw Y   Œ@xY w)Nr   é   Úignore)Úinvalidrf   rz   rS   )r   r"   Úerrstater#   r   r   )rJ   ÚxrX   r1   s       r3   Útest_nanzTestAAA.test_nan´   sk   € Ü�K‰K˜˜2ÓˆÜ�[‰[ Ô*ñ 	Ü—‘�q“	˜A‘ˆA÷	ä��1‹IˆÜ™˜!›œbŸf™f Q›i¨!™m°%Ö8÷	ð 	ús   ­BÂBc                 ó"  — t        j                  ddd¬«      }t        |t        j                  |«      |z  «      }t        j                  t        j
                  |j                  «       «      dk  «      }t        |j                  «       |   dd¬«       t        |d	t        j                  j                  |«      z  «      }t        j                  t        |j                  «       d
z
  «      dk  «      }t        |j                  «       |   dd¬«       y )Ng1¬Zdõ¿rf   i  r   ç:Œ0âŽyE>r   rz   rS   rl   r
   y      ð¿      ð¿)r   r"   r   rP   Úflatnonzerorb   r+   r   r-   r{   r|   r}   )rJ   r…   r1   Úiis       r3   Útest_residueszTestAAA.test_residues»   s¿   € Ü�K‰K˜  sÔ+ˆÜ�”2—6‘6˜!“9˜q‘=Ó!ˆÜ�^‰^œBŸF™F 1§7¡7£9Ó-°Ñ4Ó5ˆÜ˜Ÿ
™
› RÑ(¨!°%Õ8ä��FœeŸm™m×1Ñ1°!Ó4Ñ4Ó5ˆÜ�^‰^œC §¡£	¨RÑ 0Ó1°DÑ8Ó9ˆÜ˜Ÿ
™
› RÑ(¨'¸Ö>r5   zfunc,atol,rtolc                 ó8   — t        j                  | dz   dz   «      S )Nr]   y        {®Gáz„?©r   rb   ©r…   s    r3   ú<lambda>zTestAAA.<lambda>È   s   € ¬¯©°°C±¸%±Ó)@€ r5   gê-�™—a=gH¯¼šò×z>c                 ó8   — t        j                  dd| z
  z  «      S )Nr   gÍÌÌÌÌÌð?)r   r#   rŽ   s    r3   r�   zTestAAA.<lambda>É   s   € ¬¯©°°4¸!±8±Ó)=€ r5   ç‚vIhÂ%L=c                 ó8   — t        j                  d| dz  z  «      S )Nr
   rf   ©r   rP   rŽ   s    r3   r�   zTestAAA.<lambda>Ê   s   € ¬¯©°°A°q±D±	Ó):€ r5   g8JåÏ=Ã=r   c                 ó8   — t        j                  d| dz  z  «      S )Niœÿÿÿrf   r“   rŽ   s    r3   r�   zTestAAA.<lambda>Ë   s   € ¬¯©°°Q¸±T±	Ó):€ r5   gê-�™—�=c                 ó8   — t        j                  dd| z
  z  «      S )Néöÿÿÿg333333ó?r“   rŽ   s    r3   r�   zTestAAA.<lambda>Ì   s   € ¬¯©°°S¸1±W±Ó)>€ r5   ç›+¡†›„=c                 óD   — ddt        j                  d| dz   z  «      z   z  S )Nr   r   r]   r“   rŽ   s    r3   r�   zTestAAA.<lambda>Í   s!   € ¨¨A¬b¯f©f°S¸!¸c¹'±]Ó.CÑ,CÑ)D€ r5   c                 ó2   — t        j                  | dz
  «      S )Ngffffffî?r�   rŽ   s    r3   r�   zTestAAA.<lambda>Î   s   € ¬¯©°°D±Ó)9€ r5   ç�íµ ÷Æ°>c           	      óÔ   — t        j                  d¬«      5   |t        «      }d d d «       t         t	        t
         |t
        «      «      t        «      ||¬«       y # 1 sw Y   Œ;xY w)Nr‚   )Údivider\   )r   r„   ÚPTSr   r   rO   )rJ   ÚfuncrT   r   rX   s        r3   Útest_basic_functionszTestAAA.test_basic_functionsÇ   sS   € ô �[‰[ Ô)ñ 	Ù”S“	ˆA÷	äÐ?œœM©4´Ó+>Ó?ÄÓDØ ¨4ö	1÷	ð 	ús   —AÁA'c                 óì  — d„ }t        t         |t        «      «      }t        t        j                  |j                  «       |j                  «       z   «      dd¬«       d„ }t        t         |t        «      «      }t        |j                  «       j                  «       dd¬«       t        t        t        j                  dt        j                  z  t        z  «      «      }t        t        j                  t        j                  |j                  «       «      «      d   d	d¬«       d
„ }t        t         |t        «      «      }t        |j                  «       d   |j                  «       d   z  dd¬«       y )Nc                 ó0   — | dz   | dz   z  | dz   | dz   z  z  S )Nr   rf   é   é   © rv   s    r3   rX   z,TestAAA.test_poles_zeros_residues.<locals>.fÖ   s%   € Ø�a‘C˜A˜a™C‘= Q q¡S¨Q¨q©S¡MÑ2Ð2r5   r–   çê-�™—q=rS   c                 ó$   — dd| z   z  d| dz
  z  z   S )Nrf   r¢   rt   y               @r¤   rv   s    r3   rX   z,TestAAA.test_poles_zeros_residues.<locals>.fÛ   s   € Ø�a˜!‘e‘9˜q ! b¡&™zÑ)Ð)r5   r   rˆ   é   gÍÌÌÌÌÌì?c                 ó   — | dz
  | dz   z  S )Ny      @      @rf   r¤   rv   s    r3   rX   z,TestAAA.test_poles_zeros_residues.<locals>.fã   s   € Ø˜‘L 1 q¡5Ñ)Ð)r5   r   y      À      À)r   rO   r   r   Úsumr+   r.   r-   Úprodr#   r`   ri   rb   rc   s      r3   Útest_poles_zeros_residuesz!TestAAA.test_poles_zeros_residuesÕ   sõ   € ò	3ä”™q¤Ó/Ó0ˆÜœŸ™˜qŸw™w›y¨1¯7©7«9Ñ4Ó5°sÀÕGò	*ä”™q¤Ó/Ó0ˆÜ˜Ÿ
™
›×)Ñ)Ó+¨R°dÕ;ä”œrŸv™v b¬¯©¡h¬}Ñ&<Ó=Ó>ˆÜœŸ™¤§¡ q§w¡w£yÓ 1Ó2°2Ñ6¸À%ÕHò	*ä”™q¤Ó/Ó0ˆÜ˜Ÿ™›	 !™ Q§W¡W£Y¨q¡\Ñ1°EÀÖFr5   rž   c                 ó,   — t        j                  | «      S ©N)r   Ú
zeros_likerv   s    r3   r�   zTestAAA.<lambda>é   s   € ¬¯©°aÓ(8€ r5   c                 ó   — | S r­   r¤   rv   s    r3   r�   zTestAAA.<lambda>é   s   € ÀA€ r5   c                 ó   — d| z  S )Nr   r¤   rv   s    r3   r�   zTestAAA.<lambda>é   s   € ÐQSÐTUÑQU€ r5   c                 ó   — | dz  | z   S )Nrf   r¤   rv   s    r3   r�   zTestAAA.<lambda>ê   s   € ¨¨1©¨q©€ r5   c                 ó   — | dz  | z   S )Nr¢   r¤   rv   s    r3   r�   zTestAAA.<lambda>ê   s   € ¸A¸q¹DÀ1¹H€ r5   c                 ó   — dd| z   z  S )Nr   gš™™™™™ñ?r¤   rv   s    r3   r�   zTestAAA.<lambda>ë   s   € ¨¨3°©7©€ r5   c                 ó   — ddd| z  z   z  S )Nr   r   r¤   rv   s    r3   r�   zTestAAA.<lambda>ë   s   € ¸qÀ!ÀbÈÁdÁ(¹|€ r5   c                 ó   — dd| z   | dz  z   z  S )Nr   r¢   rf   r¤   rv   s    r3   r�   zTestAAA.<lambda>ì   s   € ¨¨1¨q©5°1°a±4©<Ñ(8€ r5   c                 ó   — dd| dz  z   z  S )Nr   g)\�Âõ(ð?r¢   r¤   rv   s    r3   r�   zTestAAA.<lambda>ì   s   € ÀAÀtÈaÐQRÉdÁ{ÁO€ r5   c           	      óx   — t         t        t         |t        «      «      t        «       |t        «      d¬«       y )Nr‘   rS   )r   r   rO   r�   )rJ   rž   s     r3   Ú test_polynomials_and_reciprocalsz(TestAAA.test_polynomials_and_reciprocalsè   s+   € ô 	Ð?œœM©4´Ó+>Ó?ÄÓDÙœS›	¨ö	/r5   c                 óv  — t        j                  t        j                  dddt         j                  z  z   d¬«      «      }t	        |t        j
                  t         j                  |z  dz  «      «      }t        t        j                  t        j                  |j                  «       «      «      d d g d¢d	¬
«       y )Nç      à¿r]   ù              .@r   r   rf   r£   )r   r   r¢   r¢   g1Eîð2®>r   )
r   rP   r"   r`   r   r_   r   ri   rb   r+   r~   s      r3   Útest_spiralzTestAAA.test_spiraló   su   € Ü�F‰F”2—;‘;˜t S¨3¬r¯u©u©9¡_¸$Ô?Ó@ˆÜ�”2—6‘6œ"Ÿ%™% ™' !™)Ó$Ó%ˆÜœŸ™¤§¡ q§w¡w£yÓ 1Ó2°2°AÐ6ºÈ4ÖPr5   c                 ó:  — t        j                  t        j                  dddt         j                  z  z   d¬«      «      }t	        j
                  t        «      5  t        |t        j                  t         j                  |z  dz  «      ddd	¬
«      }d d d «       t        j                  t        j                  j                  «       «      dk  «      }t	        j
                  t        «      5  |j                  «       dk\  sJ ‚	 d d d «       t        j                  t        j                  |j                  «       «      dk  «      |k  sJ ‚t         ||«      t        j                  t         j                  |z  dz  «      dd¬«       y # 1 sw Y   ŒõxY w# 1 sw Y   Œ�xY w)Nrº   r]   r»   r   r   rf   r   é<   F)r   rB   Úclean_upr—   r   gßÄAfcš=gßÄAfcŠ=r\   )r   rP   r"   r`   rD   rM   rN   r   r_   r©   rb   r-   r¿   r   )rJ   r0   r1   Ú
n_spuriouss       r3   Útest_spiral_cleanupzTestAAA.test_spiral_cleanupø   s"  € Ü�F‰F”2—;‘;˜t S¨3¬r¯u©u©9¡_¸$Ô?Ó@ˆô �\‰\œ.Ó)ñ 	PÜ�A”r—v‘vœbŸe™e A™g a™iÓ(¨q¸BÈÔOˆA÷	Pä—V‘VœBŸF™F 1§:¡:£<Ó0°5Ñ8Ó9ˆ
Ü�\‰\œ.Ó)ñ 	%Ø—:‘:“< 1Ò$Ð$Ñ$÷	%ô �v‰v”b—f‘f˜QŸZ™Z›\Ó*¨UÑ2Ó3°jÒ@Ð@Ð@ä™˜!›œbŸf™f¤R§U¡U¨1¡W¨Q¡YÓ/°eÀ%ÖH÷	Pð 	Pú÷	%ð 	%ús   Á8FÃ+FÆFÆFc                 óð   — t        j                  ddd«      }t        j                  |«      }t        ||«      }t        j                  ddd«      }t	         ||«      t        j                  |«      d¬«       y )Nr   r   i,  r   çñhãˆµøä>r   )r   ÚlogspaceÚsqrtr   r   )rJ   r0   rX   r1   Úzzs        r3   Útest_diag_scalingzTestAAA.test_diag_scaling  sV   € ä�K‰K˜˜Q Ó$ˆÜ�G‰G�A‹JˆÜ��1‹Iˆä�[‰[˜˜a Ó%ˆÜ™˜"›œrŸw™w r›{°Ö6r5   N)Ú__name__Ú
__module__Ú__qualname__rK   rQ   rZ   rd   rj   rq   rx   r   r†   r‹   rD   ÚmarkÚparametrizerŸ   r«   r¸   r¼   rÁ   rÇ   r¤   r5   r3   r;   r;   R   s  „ ò
(òDò
*ò4ò*òB0ò.òHò
9ò?ð ‡[�[×ÑÐ-Ù@À%ÈÐNÙ=¸uÀdÐKÙ:¸GÀQÐGÙ:¸EÀ1ÐEÙ>ÀÀqÐIÙDÀeÈTÐRÙ9¸4ÀÐFðHóIñ1óIð1òGð& ‡[�[×Ñ˜VÙ8¹+Á~Ù0Ñ2DÙ3Ñ5KÙ8Ñ:SðUóVñ
/óVð
/òQò
Ió7r5   r;   c                   ó   — e Zd Zd„ Zd„ Zy)ÚBatchFloaterHormannc                 óê   — t        j                  ||d«      }|j                  d d | _        |j	                  d|j                  d   «      D �cg c]  }t        ||«      ‘Œ c}| _        || _        y c c}w ©Nr
   )r   ÚmoveaxisÚshapeÚ_batch_shapeÚreshaper   Ú_interpsÚ_axis)rJ   r…   ÚyÚaxisÚyis        r3   Ú__init__zBatchFloaterHormann.__init__  sf   € Ü�K‰K˜˜4 Ó$ˆØŸG™G C R˜LˆÔà#$§9¡9¨R°·±¸±Ó#=ö?Øô 4°A°rÕ;ò ?ˆŒàˆ�
ùò?s   ÁA0c                 ó  — | j                   D �cg c]
  } ||«      ‘Œ }}t        j                  || j                  |j                  z   «      }|j                  r!t        j
                  |d| j                  «      S |S c c}w rÐ   )rÕ   r   rÔ   rÓ   rÒ   rÑ   rÖ   )rJ   r…   Úinterpr×   s       r3   Ú__call__zBatchFloaterHormann.__call__  sc   € Ø%)§]¡]Ö3˜6‰V�A�YÐ3ˆÐ3Ü�J‰J�q˜$×+Ñ+¨a¯g©gÑ5Ó6ˆØ12·²Œr�{‰{˜1˜b $§*¡*Ó-Ð?¸aÐ?ùò 4s   �A<N)rÈ   rÉ   rÊ   rÚ   rÝ   r¤   r5   r3   rÎ   rÎ     s   „ òó@r5   rÎ   c            
       ó  — e Zd Zd„ Zd„ Zd„ Zej                  j                  ddg d¢fdg d¢fd	g d
¢fdg d¢fdg d¢fg«      d„ «       Z	ej                  j                  d e
d«      «      d„ «       Zd„ Zd„ Zej                  j                  dg d¢«      ej                  j                  dddg«      d„ «       «       Zd„ Zd„ Zej                  j                  dg d¢«      ej                  j                  dg d ¢«      d!„ «       «       Zy")#ÚTestFloaterHormannc                 ó   — dd|dz  z   z  S )Nr   rf   r¤   )rJ   r0   s     r3   ÚrungezTestFloaterHormann.runge   s   € Ø�!�a˜‘d‘(‰|Ðr5   c                 óP   — dt        j                  |«      |z   z  t        |«      z  S rÐ   )r   r8   r   )rJ   ÚnÚds      r3   ÚscalezTestFloaterHormann.scale#  s#   € Ø”b—i‘i “l QÑ&Ñ'¬)°A«,Ñ6Ð6r5   c                 ó¬  — t        j                  t        d¬«      5  t        dggdgd¬«       d d d «       t        j                  t        d¬«      5  t        dgdd¬«       d d d «       t        j                  t        d¬«      5  t        dgddgddggd¬«       d d d «       t        j                  t        d¬«      5  t        t        j
                  gdgd¬«       d d d «       t        j                  t        d	¬«      5  t        dgdgd
¬«       d d d «       t        j                  t        d	¬«      5  t        dgdgd¬«       d d d «       t        j                  t        «      5  t        dgdgd¬«       d d d «       y # 1 sw Y   �ŒSxY w# 1 sw Y   �Œ-xY w# 1 sw Y   �ŒxY w# 1 sw Y   ŒËxY w# 1 sw Y   Œ£xY w# 1 sw Y   Œ{xY w# 1 sw Y   y xY w)Nz`x`r>   r   ©rä   z`y`r=   r   r@   z`d`r
   r   ç        )rD   rE   rF   r   r   rG   rH   rI   s    r3   Útest_ivzTestFloaterHormann.test_iv&  s~  € Ü�]‰]œ:¨UÔ3ñ 	8Ü&¨¨ u¨q¨c°QÕ7÷	8ä�]‰]œ:¨UÔ3ñ 	4Ü&¨ s¨A°Õ3÷	4ä�]‰]œ:Ð-OÔPñ 	CÜ&¨ s¨a°¨V°a¸°VÐ,<ÀÕB÷	Cä�]‰]œ:¨XÔ6ñ 	;Ü&¬¯© x°!°¸Õ:÷	;ä�]‰]œ:¨UÔ3ñ 	7Ü&¨ s¨Q¨C°2Õ6÷	7ä�]‰]œ:¨UÔ3ñ 	7Ü&¨ s¨Q¨C°2Õ6÷	7ä�]‰]œ9Ó%ñ 	8Ü&¨ s¨Q¨C°3Õ7÷	8ð 	8÷	8ñ 	8ú÷	4ñ 	4ú÷	Cñ 	Cú÷	;ð 	;ú÷	7ð 	7ú÷	7ð 	7ú÷	8ð 	8úsS   œE?ÁFÂFÂ=F&Ã?F2Ä3F>Å%G
Å?F	ÆFÆF#Æ&F/Æ2F;Æ>GÇ
Gz
d,expectedr   )r   r   r   r   r   r   r   r   r   r   r   r   )r   rf   rf   rf   rf   rf   rf   rf   rf   rf   r   rf   )r   r¢   r£   r£   r£   r£   r£   r£   r£   r¢   r   r¢   )r   r£   é   é   rë   rë   rë   rë   rê   r£   r   r£   )r   rt   é   é   é   rî   rî   rí   rì   rt   r   c                 óØ   — t        j                  d«      }t        |d|z  |¬«      }t        |j                  j                  «       | j                  |j                  |«      z  |dd¬«       y )Nrì   rè   rç   rz   ©r   rT   )r   r8   r   r   r*   Úravelrå   r   )rJ   rä   Úexpectedr…   r1   s        r3   Útest_uniform_gridz$TestFloaterHormann.test_uniform_grid7  sS   € ô �I‰I�b‹MˆÜ& q¨#¨a©%°1Ô5ˆÜ˜Ÿ	™	Ÿ™Ó)¨$¯*©*°Q·V±V¸QÓ*?Ñ?ÀØ"¨ö	0r5   rä   r   c                 óŒ  — t        j                  ddd«      }t         j                  j                  d«      }|j	                  ddd¬«      }| j                  |«      }|d   |d   z
  }t        |||¬«      }d||dz   z  z  }t         ||«      | j                  |«      d	|¬
«       t         ||«      | j                  |«      «       y )Nr   r   é3   l   Mû+çRYr   r   rç   r   g»½×Ùß|Û=r\   )	r   r"   r    r!   r$   rá   r   r   r   )	rJ   rä   r…   r/   Úxxr×   Úhr1   Útols	            r3   Ú
test_rungezTestFloaterHormann.test_rungeE  sª   € ä�K‰K˜˜1˜bÓ!ˆÜ�i‰i×#Ñ#Ð$6Ó7ˆØ�[‰[˜˜A Dˆ[Ó)ˆØ�J‰J�q‹MˆØˆa‰D�1�Q‘4‰Kˆä& q¨!¨qÔ1ˆà��Q�q‘S‘‰kˆÜ™˜"›˜tŸz™z¨"›~°EÀÕDä‘Q�q“T˜4Ÿ:™: a›=Õ)r5   c                 ó  — t        j                  dd«      }||dz  z   }t        |t        j                  |«      d¬«      }t        j                  ddd¬«      }||dz  z   }t	         ||«      t        j                  |«      d¬	«       y )
Nr
   r   r   é   rç   r   r   r¥   r   )r   r"   r   r#   r   )rJ   r…   r0   r1   rö   rÆ   s         r3   Útest_complexzTestFloaterHormann.test_complexT  sl   € Ü�K‰K˜˜AÓˆØ��"‘‰HˆÜ& q¬"¯&©&°«)°rÔ:ˆÜ�[‰[˜˜Q DÔ)ˆØ�"�R‘%‰ZˆÜ™˜"›œrŸv™v b›z°Ö6r5   c                 ó  — t        j                  ddd«      }t        j                  ddd«      }t        j                  |«      }t        |||j                  dz
  ¬«      }t        ||«      }t         ||«       ||«      dd¬«       y )Nr   r   rì   ié  rç   r¥   rð   )r   r"   r#   r   r   r	   r   )rJ   r…   rö   r×   r1   Úps         r3   Útest_polyinterpz"TestFloaterHormann.test_polyinterp\  sk   € ä�K‰K˜˜1˜bÓ!ˆÜ�[‰[˜˜A˜tÓ$ˆÜ�F‰F�1‹IˆÜ& q¨!¨q¯v©v°a©xÔ8ˆÜ# A qÓ)ˆÜ™˜"›™q ›u¨5°uÖ=r5   Úy_shape))rf   )rf   r¢   r   )r   rt   é   r£   Úxx_shaper   )r   r   c                 óÜ  — t        j                  dd«      }t        j                  t        j                  t        j                  |«      t        t        dt        |«      dz   «      «      «      |j                  |z   «      }t        ||«      }t         j                  j                  d«      }|j                  |«      }t        j                  t        j                  t        j                  |«      t        t        |j                  t        |«      |j                  z   «      «      «      |j                  |z   «      } ||«      }	|	j                  |j                  |z   k(  sJ ‚t        |	|d¬«       y )Nr   r   l   �‹wcïBMþYÔrš   r   )r   r"   Úbroadcast_toÚexpand_dimsr#   ÚtupleÚrangeÚlenrÒ   r   r    r!   Úndimr   )
rJ   r   r  r…   r×   r1   r/   rö   ÚyyÚrrs
             r3   Útest_trailing_dimz$TestFloaterHormann.test_trailing_dime  s	  € ô �K‰K˜˜1ÓˆÜ�O‰OÜ�N‰Nœ2Ÿ6™6 !›9¤e¬E°!´S¸³\ÀAÑ5EÓ,FÓ&GÓHØ�G‰G�gÑó
ˆô
 ' q¨!Ó,ˆä�i‰i×#Ñ#Ð$BÓCˆØ�Z‰Z˜Ó!ˆÜ�_‰_Ü�N‰Nœ2Ÿ6™6 "›:¤u¬U°2·7±7¼CÀ»LÈ2Ï7É7Ñ<RÓ-SÓ'TÓUØ�H‰H�wÑó
ˆñ ˆr‹UˆØ�x‰x˜2Ÿ8™8 gÑ-Ò-Ð-Ð-Ü˜˜B TÖ*r5   c                 ót  — t        j                  ddd¬«      }t        |t        j                  t         j                  |z  «      «      }t        j
                  t         j                  j                  |j                  «       t        j                  d«      «      «      j                  d¬«      }t        |d«       y )Nr   r   r   r   rì   ©rØ   rÃ   )r   r"   r   r#   r`   rb   ÚsubtractÚouterr.   r8   ra   r   )rJ   r…   r1   Úerrs       r3   Ú
test_zeroszTestFloaterHormann.test_zeros{  su   € Ü�K‰K˜˜2 3Ô'ˆÜ& q¬"¯&©&´·±°q±«/Ó:ˆä�f‰f”R—[‘[×&Ñ& q§w¡w£y´"·)±)¸B³-Ó@ÓA×EÑEÈ1ÐEÓMˆÜ˜#˜tÕ$r5   c                 ó,  — t        j                  dd«      }t        |d|dz  z  «      }|j                  «       }|j                  dk\  |j                  dk  z  t        j
                  |j                  «      dk  z  }t        j                  |«      dk(  sJ ‚y )Nr
   r   rf   r¥   r   )r   r"   r   r+   r)   rb   Úimagr©   )rJ   r…   r1   rþ   Úmasks        r3   Útest_no_polesz TestFloaterHormann.test_no_poles‚  sw   € Ü�K‰K˜˜AÓˆÜ& q¨!¨A¨q©D©&Ó1ˆØ�G‰G‹IˆØ—‘˜"‘ §¡¨1¡Ñ-´·±¸¿¹³À&Ñ1HÑIˆÜ�v‰v�d‹|˜qÒ Ð Ñ r5   Ú
eval_shape)r¤   )r   )r¢   rØ   )r
   r   r   c                 óì  — t         j                  j                  d«      }d}ddd|f}d}t        j                  g |¢|‘­Ž }t        j                  |j                  |«      d|«      }t        |||¬«      }	t        |||¬«      }
 |j                  |d	|iŽ}t         |	|«       |
|«      «       t        j                  t        |	j                  «       t        j                  t        |	j                  «       y )
Nl   ®UUû{ r   rf   r¢   r£   )r   r   r
   r  r   )r   r    r!   r"   rÑ   r   rÎ   r$   r   rD   rE   ÚNotImplementedErrorr.   r-   )rJ   r  rØ   r/   rã   rÒ   Údomainr…   r×   ÚresÚrefs              r3   Ú
test_batchzTestFloaterHormann.test_batch‰  sÊ   € ô �i‰i×#Ñ#Ð$4Ó5ˆØˆØ�A�q˜!�ˆØˆä�K‰KÐ#˜Ð# Ò#ˆÜ�K‰K˜Ÿ
™
 5Ó)¨2¨tÓ4ˆä(¨¨A°DÔ9ˆÜ! ! Q¨TÔ2ˆàˆC�K‰K˜Ð1 jÑ1ˆÜ™˜A›¡ A£Ô'ä�‰Ô)¨3¯9©9Ô5Ü�‰Ô)¨3¯<©<Õ8r5   N)rÈ   rÉ   rÊ   rá   rå   ré   rD   rË   rÌ   ró   r  rù   rü   rÿ   r  r  r  r  r¤   r5   r3   rß   rß     s)  „ òò7ò8ð" ‡[�[×Ñ˜\Ø	
Ò-Ð.Ø	
Ò-Ð.Ø	
Ò-Ð.Ø	
Ò-Ð.Ø	
Ò4Ð5ð,ó ñ0óð0ð ‡[�[×Ñ˜S¡%¨£)Ó,ñ*ó -ð*ò7ò>ð ‡[�[×Ñ˜YÒ(GÓHØ‡[�[×Ñ˜Z¨3°Ð):Ó;ñ+ó <ó Ið+ò(%ò!ð ‡[�[×Ñ˜\Ò+;Ó<Ø‡[�[×Ñ˜V¢ZÓ0ñ9ó 1ó =ñ9r5   rß   )$Úmathr   Únumpyr   Únumpy.testingr   r   r   rD   r{   Úscipy.interpolater   r   r	   r   Úfloat64r   rU   r"   rO   rÄ   r�   ÚconcatenaterË   rÌ   Úfloat32Ú	complex64rm   r4   Úint16Úint32Úint64r9   r;   rÎ   rß   r¤   r5   r3   ú<module>r)     sh  ðõ4 ã ß JÑ JÛ Û ß VÑ Và	ˆHˆB�H‰H�R—Z‘ZÓ ×$Ñ$Ñ$€Ø�—‘˜B  tÔ,€Ø€b‡k�k�#�q˜r sÔ+€Ø€b‡n�n�s™4˜R˜4‘y�j 1 # sÐ+Ó,€ð ‡�×Ñ˜ CÐ)CÐ#DÓEØ‡�×Ñ˜ 2§:¡:¨r¯z©z¸2¿<¹<ÈÏÉÐ"WÓXñ8ó Yó Fð8ð. ‡�×Ñ˜ CÐ)CÐ#DÓEØ‡�×Ñ˜ 2§8¡8¨R¯X©X°r·x±xÐ"@ÓAñ4ó Bó Fð4÷{7ñ {7÷|@ñ @÷|9ò |9r5   