
    tKg                         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.                  j1                  d	g d
      d        Zy)    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      j!                          d d d        y # 1 sw Y   y xY w)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    t        | |j                  j                  t                    S N)r   realastypeint)xys     e/home/alanp/www/video.onchill/myenv/lib/python3.12/site-packages/scipy/special/tests/test_lambertw.pywztest_values.<locals>.wQ   s    166==-..    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      sB   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&1s1vg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%U;AAC 
#	"	"s    G>>Hc                  X    t        t        t        dt        df         t        d          y )N        r   )r;   r   r   )r   r   r   r    r/   r-   
test_ufuncr=   W   s%    B2I$@!A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                  |        y )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(..3r/   z)ga4    y            y+q          c                 (    t        |       }|| k(  sJ y r'   r   )rA   r.   s     r-   test_lambertw_subnormal_k0rC   f   s     	A 6M6r/   )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 " D D D -BDJC
4 AB Cr/   