o
    jgz  ã                   @   sr   d dl mZ d dlmZ d dlmZ d dlmZmZ d dl	m
Z
 d dlmZmZ edƒ\ZZZdd	„ Zd
d„ ZdS )é    )Úproduct)ÚS)Úsymbols)ÚexpÚlog)Úsrepr)Ú	logaddexpÚ
logaddexp2zx y zc            
      C   sè   t ttƒ} tttƒttƒ ƒ}ttttgtdƒƒD ]\}}|  ||¡| ||¡  	t¡ 
¡ dks2J ‚qdtdƒ d }dtdƒ d }t|ƒ}t|ƒ}t ||ƒ}| 	t¡dksYJ ‚| 
¡ dksaJ ‚t ddƒ}	t|	ƒt|	 
¡ ƒksrJ ‚d S )Né   r   é   é   )r   ÚxÚyr   r   r   ÚzÚrangeÚdiffÚrewriteÚsimplifyr   )
Úlae_xyÚref_xyÚwrtÚderiv_orderÚone_third_eÚtwo_thirds_eÚ	logThirdEÚlogTwoThirdsEÚlae_sum_to_eÚwas© r   ú\/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/codegen/tests/test_numpy_nodes.pyÚtest_logaddexp
   s"   


ÿ
ý

r    c                  C   sî   t ttƒ} tdt dt  ƒtdƒ }ttttgtdƒƒD ]\}}|  ||¡| ||¡  t¡ 	¡ dks6J ‚qdd„ }t
jd d }d| }||ƒ}||ƒ}t ||ƒ}	|	 t¡dks\J ‚|	 ¡ dksdJ ‚t ttƒ}
t|
ƒt|
 ¡ ƒksuJ ‚d S )Nr   r
   r   c                 S   s   t | ƒt dƒ S )Nr   )r   )r   r   r   r   Úlb'   s   ztest_logaddexp2.<locals>.lbr   )r	   r   r   r   r   r   r   r   r   Úcancelr   ÚOner   r   )Úlae2_xyÚref2_xyr   r   r!   Ú
two_thirdsÚfour_thirdsÚlbTwoThirdsÚlbFourThirdsÚlae2_sum_to_2r   r   r   r   Útest_logaddexp2   s$   


ÿ
ý

r+   N)Ú	itertoolsr   Úsympy.core.singletonr   Úsympy.core.symbolr   Ú&sympy.functions.elementary.exponentialr   r   Úsympy.printing.reprr   Úsympy.codegen.numpy_nodesr   r	   r   r   r   r    r+   r   r   r   r   Ú<module>   s    