
    ^Mh                         d dl Z d dlZd dlmZmZmZ d dlmZ d dlm	Z	m
Z
mZmZmZmZmZmZmZ d dlmZ d Zd Zd Ze j                            d	g d
          d             ZdS )    N)assert_assert_equalassert_array_almost_equallambertw)	naninfpieisnanlogr_array
complex128)FuncDatac            	         t          t          t          t                                         t	          t          t
          d          j        t
                     t	          t          t
          d          j        dt          z             t	          t          t
           d          j        t
                     t	          t          t
           d          j        dt          z             t	          t          d          t          dd                     g ddt
          dt
          fddt
           fddt
           fddt
           ft          ddfd	t           dz  dd
t          z  dz  ft          d           dz  dt          d           fddddddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*t          dd+fd,d-d.d/} t          | t          0          } d1 }t          j        d23          5  t          || d4dd5d67                                           d d d            d S # 1 swxY w Y   d S )8N               ?r   )r   r   r   )y                r   r   )r   r   W	&?y              ?)      ?r   g{?)      пr   gֿ)-C6r   g"{,7)r   r   gѤX9)r   r   yHgtm3]N)r   r   yHgtm3]N@)r   r   y}&@)r   yӸ?~dY?)r   r   yg[忼@)r   r   yӸ?~dY)  r   gr6@)r   r   yAj@T'>@)r   r   yAj@T'>)r      y(o-@n!=@)      @      @r   yAF?Z|?)yٿ?r   y)u%+?)r   r   y(tmȻy@)r   r   yO^?")g      r   y~}?he)r   r   y:㢥'%K8@)r   r   g8	U')r   y:㢥'%K8)h㈵   yk)j^-xe/:@)r!   r   yb/.>ꮺF@@)??r   y0j;?"B`?)r#   r   ym.[	<?@)r#   r   ysqM1mP)r#   r"   yE&\57@)ɿr   y#u*Ϳ-pFſ)r$   r   y>M|Wn2?)r$   r   yǁc'FyE)r$   r"   yfI3I5fԪ4@g7-?)y      Mb`?r   y7W@;ūԎ?)y      Mb`r   y7W@;ūԎ)yy&1ܿ?r   y)!faYNJuM?)yy&1ܿٿr   y)!faYNJuM)dtypec                 \    t          | |j                            t                              S N)r   realastypeint)xys     a/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/scipy/special/tests/test_lambertw.pywztest_values.<locals>.wQ   s     16==--...    ignore)all)r   r   g|=gvIh%<=)rtolatol)r   r   r   r   r   r	   r(   imagr
   r   r   r   r   nperrstater   check)datar.   s     r-   test_valuesr9      s    E(3--  !!!#a%s+++#a%qt,,,3$q!!&,,,3$q!!&"---"xA///333 
Q3 
2t	3
 
1sd3 
1sd3 
1a3 	$3 Qq"R%'3 a&&1s1vvg3 	'3 	)3 	03 	)3 	=3  	)!3" 	<#3$ 	=%3& 	9'3( 	:)3* 	:+3, 	&-3. 	;/30 	<132 	:334 	=536 	A738 	<93: 	<;3< 	?=3> 	B?3@ 	-A3B 	CC3D 	BE3F 	HG3H 	BI3J 	AK3L 	BM3N 	@O3P 	DQ3R 	BS3T 	CU3V 	AW3X 
AGHY3^ 	K_3` 	Ka3b 	Kc3d 	Ke3Dh Z(((D/ / /		"	"	" D DD%U;;;AACCCD D D D D D D D D D D D D D D D D Ds   (IIIc                  |    t          t          t          dt          df                   t          d                    d S )N        r   )r;   r   r   )r   r   r   r    r/   r-   
test_ufuncr=   W   s@    B2I$@!AC C C C Cr/   c                     t          j        t           j                  } t          t	          ddd          j        |            t          t	          dgdd          j        |            t          t	          ddgd          j        |            t          t	          dddg          j        |            t          t	          dgdgdg          j        |            d S )Nr   )r5   r%   r   r   r   )dts    r-   "test_lambertw_ufunc_loop_selectionr@   \   s    	"-	 	 B!Q""("---1#q!$$*B///!aS!$$*B///!Q$$*B///1#sQC((.33333r/   z)ga4    y            y+q          c                 4    t          |           }|| k    sJ d S r'   r   )rA   r.   s     r-   test_lambertw_subnormal_k0rC   f   s"     	A 666666r/   )pytestnumpyr5   numpy.testingr   r   r   scipy.specialr   r   r	   r
   r   r   r   r   r   r   scipy.special._testutilsr   r9   r=   r@   markparametrizerC   r<   r/   r-   <module>rK      s        J J J J J J J J J J " " " " " " D D D D D D D D D D D D D D D D D D D D D D - - - - - -BD BD BDJC C C
4 4 4 AAABB  CB  r/   