Ë
    7^(hà  ã                   óÔ   — d dl mZ d dlmZ d dlZ ed«      r
d dlmc mZ ndZ	d dl
Z
d dlmZmZmZ d„ Zd„ Zd	„ Zd
„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z y)é    )Úimport_module)ÚraisesNÚllvmliteT)ÚaÚbÚnc                 óJ   — d}d}t        | |z
  «      ||t        |«      z  z   k  S )Ngñhãˆµøä>g:Œ0âŽyE>)Úabs)r   r   ÚrtolÚatols       ú_/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sympy/printing/tests/test_llvmjit.pyÚiscloser      s,   € Ø€DØ€DÜˆq�‰s‹8�t˜d¤3 q£6™kÑ)Ñ)Ð)ó    c                  óÔ   — t         dz   } t        j                  t         g| «      }t        | j	                  t         di«      j                  «       «      } |d«      }t        ||«      sJ ‚y )Ng      ð?ç      @)r   ÚgÚllvm_callableÚfloatÚsubsÚevalfr   ©ÚeÚfÚresÚjit_ress       r   Útest_simple_exprr      sV   € Ü	ˆC‰€AÜ	�‰œ˜˜QÓ€AÜ
�—‘œ˜3�xÓ ×&Ñ&Ó(Ó
)€CÙ�‹f€Gä�7˜CÔ Ð Ñ r   c                  ó   — dt         z  t        z   dz   } t        j                  t         t        g| «      }t	        | j                  t         dt        di«      j                  «       «      } |dd«      }t        ||«      sJ ‚y )Nr   g      @©r   r   r   r   r   r   r   r   r   s       r   Útest_two_argr      sg   € ØŒA‰”‰	�C‰€AÜ	�‰œœA˜ Ó"€AÜ
�—‘œ˜3¤ 3Ð'Ó(×.Ñ.Ó0Ó
1€CÙ��S‹k€Gä�7˜CÔ Ð Ñ r   c                  óü   — dt        j                  t         «      z  } t        j                  t        g| «      }t        | j                  t        di«      j                  «       «      } |d«      }t        ||«      sJ ‚y )Nr   ç      ø?)	ÚsympyÚexpr   r   r   r   r   r   r   r   s       r   Ú	test_funcr$   (   sa   € ØŒE�I‰I”q�b‹MÑ€AÜ	�‰œ˜˜QÓ€AÜ
�—‘œ˜3�xÓ ×&Ñ&Ó(Ó
)€CÙ�‹f€Gä�7˜CÔ Ð Ñ r   c                  óH  — dt        j                  t         «      z  t        j                  t        «      z   } t	        j
                  t        t        g| «      }t        | j                  t        dt        di«      j                  «       «      } |dd«      }t        ||«      sJ ‚y ©Nr   r!   ç       @)
r"   r#   r   r   r   r   r   r   r   r   r   s       r   Útest_two_funcr(   1   sw   € ØŒE�I‰I”q�b‹MÑœEŸI™I¤a›LÑ(€AÜ	�‰œœA˜ Ó"€AÜ
�—‘œ˜3¤ 3Ð'Ó(×.Ñ.Ó0Ó
1€CÙ��S‹k€Gä�7˜CÔ Ð Ñ r   c                  óF  — dt        j                  t        «      z  t        j                  t        «      z   } t	        j
                  t        t        g| «      }t        | j                  t        dt        di«      j                  «       «      } |dd«      }t        ||«      sJ ‚y r&   )
r"   Úsqrtr   r   r   r   r   r   r   r   r   s       r   Útest_two_sqrtr+   :   su   € ØŒE�J‰J”q‹MÑœEŸJ™J¤q›MÑ)€AÜ	�‰œœA˜ Ó"€AÜ
�—‘œ˜3¤ 3Ð'Ó(×.Ñ.Ó0Ó
1€CÙ��S‹k€Gä�7˜CÔ Ð Ñ r   c                  ó   — t         dz  t        dz  z   } t        j                  t         t        g| «      }t	        | j                  t         dt        di«      j                  «       «      } |dd«      }t        ||«      sJ ‚y )Nr!   é   r'   r   r   s       r   Útest_two_powr.   C   sg   € Ü	ˆ3‰”�A‘‰€AÜ	�‰œœA˜ Ó"€AÜ
�—‘œ˜3¤ 3Ð'Ó(×.Ñ.Ó0Ó
1€CÙ��S‹k€Gä�7˜CÔ Ð Ñ r   c                  óL  — t         dz   } t        j                  t         g| d¬«      }t        j                  d«      }t        j
                  dz  }t         di} ||t            «      } |||«      }t        | j                  |«      j                  «       «      }t        ||«      sJ ‚y )Nç333333ó?úscipy.integrate.test©Úcallback_typeé   çš™™™™™@©
r   r   r   ÚctypesÚc_intÚc_doubler   r   r   r   ©r   r   ÚmÚ
array_typeÚinpÚarrayr   r   s           r   Útest_callbackr?   L   sˆ   € Ü	ˆC‰€AÜ	�‰œ˜˜QÐ.DÔE€AÜ�‰�Q‹€AÜ—‘ 1Ñ$€JÜˆcˆ(€CÙ�sœ1‘vÓ€EÙ��5‹k€Gä
�—‘�s“×!Ñ!Ó#Ó
$€Cä�7˜CÔ Ð Ñ r   c                  óv  — t         dz   } t        j                  t         g| d¬«      }t        j                  d«      }t        j
                  dz  }t         di} ||t            «      } |d«      } |||d ||«      }|dk(  sJ ‚t        | j                  |«      j                  «       «      }t        |d   |«      sJ ‚y )Nr0   Úcubaturer2   r4   r5   g        r   r6   )	r   r   r;   r<   r=   r>   Ú	out_arrayÚjit_retr   s	            r   Útest_callback_cubaturerD   Z   s§   € Ü	ˆC‰€AÜ	�‰œ˜˜Q¨jÔ9€AÜ�‰�Q‹€AÜ—‘ 1Ñ$€JÜˆcˆ(€CÙ�sœ1‘vÓ€EÙ˜3“€IÙ��5˜$  9Ó-€Gà�aŠ<Ðˆ<ä
�—‘�s“×!Ñ!Ó#Ó
$€Cä�9˜Q‘< Ô%Ð%Ñ%r   c                  ó€  — dt         z  t        z  } t        j                  t         t        g| d¬«      }t	        j
                  d«      }t        j                  dz  }t         dt        di} ||t            |t           «      } |||«      }t        | j                  |«      j                  «       «      }t        ||«      sJ ‚y )Né   r1   r2   é   çš™™™™™É?ç333333û?)r   r   r   r   r7   r8   r9   r   r   r   r   r:   s           r   Útest_callback_tworJ   k   s™   € Ø	Œ!‰ŒA‰€AÜ	�‰œœA˜ Ð1GÔH€AÜ�‰�Q‹€AÜ—‘ 1Ñ$€JÜˆc”1�cÐ
€CÙ�sœ1‘v˜s¤1™vÓ&€EÙ��5‹k€Gä
�—‘�s“×!Ñ!Ó#Ó
$€Cä�7˜CÔ Ð Ñ r   c                  ó–  — t        j                  d«      } d| d   z  | d   z  }t        j                  t        | g|d¬«      }t        j                  d«      }t
        j                  dz  }| d   d| d   d	i} ||| d      || d      «      } |||«      }t        |j                  |«      j                  «       «      }t        ||«      sJ ‚y )
NÚdrF   r   r4   r1   r2   rG   rH   rI   )r"   ÚIndexedBaser   r   r   r7   r8   r9   r   r   r   r   )	rL   r   r   r;   r<   r=   r>   r   r   s	            r   Útest_callback_alt_tworN   y   sÁ   € Ü×Ñ˜#Ó€AØ	ˆ!ˆA‰$‰ˆq�‰t‰€AÜ	�‰œ˜A˜ Ð1GÔH€AÜ�‰�Q‹€AÜ—‘ 1Ñ$€JØˆQ‰4��a˜‘d˜CÐ
 €CÙ�s˜1˜Q™4‘y # a¨¡d¡)Ó,€EÙ��5‹k€Gä
�—‘�s“×!Ñ!Ó#Ó
$€Cä�7˜CÔ Ð Ñ r   c                  ó
  — t         dt        z  fgt         dz   gg} t        j                  t        g| «      }| d   d   d   j	                  t        di«      }t        | d   d   j	                  t         |i«      j                  «       «      } |d«      }t        ||«      sJ ‚t        j                  t        g| d¬«      }t        j                  d«      }t        j                  dz  } |d«      } |||«      }	t        |	|«      sJ ‚y )Nr   é   r   r4   r!   r1   r2   )r   r   r   r   r   r   r   r   r7   r8   r9   )
r   r   Úb_valr   r   Ú
f_callbackr;   r<   r>   Újit_callback_ress
             r   Útest_multiple_statementsrT   ˆ   sä   € äˆc”!‰eˆ*ˆœ˜A™�wÐ€AÜ	�‰œ˜˜QÓ€AØˆa‰D�‰G�A‰J�O‰OœQ ˜HÓ%€EÜ
��!‘�Q‘—‘œa ˜ZÓ(×.Ñ.Ó0Ó
1€CÙ�‹f€GÜ�7˜CÔ Ð Ð ä—‘¤!  aÐ7MÔN€JÜ�‰�Q‹€AÜ—‘ 1Ñ$€JÙ�s‹O€EÙ! ! UÓ+ÐÜÐ# SÔ)Ð)Ñ)r   c                  óš  — t         t         z  t        t        z  z   t        j                  t          t         z  t        t        z  z
  «      z   } t        j                  | «      }t        j                  t         t        g|«      }t        | j                  t         dt        di«      j                  «       «      } |dd«      }t        ||«      sJ ‚y )Ngffffff@çš™™™™™¹?)r   r   r"   r#   Úcser   r   r   r   r   r   )r   Úe2r   r   r   s        r   Útest_cserY   ™   s�   € Ü	Œ!‰Œa”‰c‰	”E—I‘Iœq˜b¤™d¤Q¤q¡S™jÓ)Ñ)€AÜ	�‰�1‹€BÜ	�‰œœA˜ Ó#€AÜ
�—‘œ˜3¤ 3Ð'Ó(×.Ñ.Ó0Ó
1€CÙ��S‹k€Gä�7˜CÔ Ð Ñ r   c                 óÚ   — i }| d   D ],  \  }}|j                  |«      }|j                  |«      }|||<   Œ. | d   D � cg c]"  } | j                  |«      j                  |«      ‘Œ$ c} S c c} w )Nr   r4   )r   )r   Úsub_dictÚtmp_dictÚtmp_nameÚtmp_exprrX   Úe3s          r   Úeval_cser`   £   ss   € Ø€HØ ™dò  Ñˆ�(Ø�]‰]˜8Ó$ˆØ�W‰W�XÓˆØˆ�Òð ð 67°q±TÖ:°ˆA�F‰F�8Ó×!Ñ! (Õ+Ò:Ð:ùÒ:s   ¾'A(c                  óœ  ‡— t         t         z  } t         t         z  t        t        z  z   }t        j                  | |g«      Št	        t
        ˆfd„«       t        j                  t         t        g‰«      } |dd«      }t        |«      dk(  sJ ‚t        ‰t         dt        di«      }t        |d   |d   «      sJ ‚t        |d   |d   «      sJ ‚y )Nc                  óH   •— t        j                  t        t        g‰ d¬«      S )Nzscipy.integrater2   )r   r   r   r   )r_   s   €r   ú<lambda>z#test_cse_multiple.<locals>.<lambda>²   s   ø€ ”1—?‘?¤A¤q 6¨2Ð=NÔO€ r   rV   r!   rG   r   r4   )r   r   r"   rW   r   ÚNotImplementedErrorr   r   Úlenr`   r   )Úe1rX   r   r   r   r_   s        @r   Útest_cse_multiplerg   ¬   s¶   ø€ Ü	
Œ1‰€BÜ	
Œ1‰Œq”‰s‰€BÜ	�‰�B˜�8Ó	€Bä
ÔÛOôQô 	
�‰œœA˜ Ó#€AÙ��S‹k€GÜˆw‹<˜1ÒÐÐÜ
�2œ˜3¤ 3Ð'Ó
(€CÜ�3�q‘6˜7 1™:Ô&Ð&Ð&Ü�3�q‘6˜7 1™:Ô&Ð&Ñ&r   c                  ó„  — t         t         z  } t         t         z  t        t        z  z   }t        j                  | |d|z  g«      }t	        j
                  t         t        g|d¬«      }d}d}t        j                  |«      }t        j                  |«      }t        j                  |z  }t        j                  |z  }	t         dt        di}
 ||
t            |
t           «      } |	«       } |||d ||«      }|dk(  sJ ‚t        ||
«      }t        |d   |d   «      sJ ‚t        |d	   |d	   «      sJ ‚t        |d   |d   «      sJ ‚y )
Né   rA   r2   rG   rF   rH   r!   r   r4   )r   r   r"   rW   r   r   r7   r8   r9   r`   r   )rf   rX   r_   r   ÚndimÚoutdimr;   Úfdimr<   Úout_array_typer=   r>   rB   rC   r   s                  r   Útest_callback_cubature_multiplern   ¼   s*  € Ü	
Œ1‰€BÜ	
Œ1‰Œq”‰s‰€BÜ	�‰�B˜˜A˜b™D�>Ó	"€BÜ	�‰œœA˜ °*Ô=€Að €Dà€Fä�‰�TÓ€AÜ�<‰<˜Ó€DÜ—‘ 4Ñ'€JÜ—_‘_ vÑ-€NÜˆc”1�cÐ
€CÙ�sœ1‘v˜s¤1™vÓ&€EÙÓ €IÙ��5˜$  iÓ0€Gà�aŠ<Ðˆ<ä
�2�sÓ
€Cä�9˜Q‘<  Q¡Ô(Ð(Ð(Ü�9˜Q‘<  Q¡Ô(Ð(Ð(Ü�9˜Q‘<  Q¡Ô(Ð(Ñ(r   c                  óT   ‡ — t         t         z  t        z   Š t        t        ˆ fd„«       y )Nc                  ó:   •— t        j                  t        g‰ «      S )N©r   r   r   ©r   s   €r   rc   z'test_symbol_not_found.<locals>.<lambda>Û   s   ø€ ¤§¡´°°QÓ 7€ r   )r   r   r   ÚLookupErrorrr   s   @r   Útest_symbol_not_foundrt   Ù   s   ø€ Ü	Œ!‰Œa‰€AÜ
Œ;Ó7Õ8r   c                  ó8   ‡ — t         Š t        t        ˆ fd„«       y )Nc                  ó>   •— t        j                  t        g‰ d¬«      S )NÚbad_callbackr2   rq   rr   s   €r   rc   z#test_bad_callback.<locals>.<lambda>à   s   ø€ œqŸ™´¨s°AÀ^ÔT€ r   )r   r   Ú
ValueErrorrr   s   @r   Útest_bad_callbackry   Þ   s   ø€ Ü	€AÜ
Œ:ÓTÕUr   )!Úsympy.externalr   Úsympy.testing.pytestr   r7   Úsympy.printing.llvmjitcodeÚprintingÚllvmjitcoder   Údisabledr"   Ú	sympy.abcr   r   r   r   r   r   r$   r(   r+   r.   r?   rD   rJ   rN   rT   rY   r`   rg   rn   rt   ry   © r   r   ú<module>r‚      s‡   ðÝ (Ý 'Û ñ �Ôß*Ñ*à€Hã ß Ñ ò*ò!ò!ò!ò!ò!ò!ò!ò&ò"!ò!ò*ò"!ò;ò'ò )ò:9ó
Vr   