
    (ph                         S SK r S SKrS SKJrJrJr  S SKJr  S SKJ	r	J
r
JrJrJrJrJrJrJr  S SKJr  S rS rS r\ R.                  R1                  S	/ S
Q5      S 5       rg)    N)assert_assert_equalassert_array_almost_equallambertw)	naninfpieisnanlogr_array
complex128)FuncDatac            
         [        [        [        [        5      5      5        [	        [        [
        S5      R                  [
        5        [	        [        [
        S5      R                  S[        -  5        [	        [        [
        * S5      R                  [
        5        [	        [        [
        * S5      R                  S[        -  5        [	        [        S5      [        SS5      5        / SPSP[
        S[
        4PSS[
        * 4PSS[
        * 4PSS[
        * 4P[        SS4PS	P[        * S-  SS
[        -  S-  4P[        S5      * S-  S[        S5      * 4PSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPSPS PS!PS"PS#PS$PS%PS&PS'PS(PS)PS*P[        SS+4PS,PS-PS.PS/Pn [        U [        S09n S1 n[        R                  " S2S39   [        XS4SS5S6S79R!                  5         S S S 5        g ! , (       d  f       g = f)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                 R    [        XR                  R                  [        5      5      $ N)r   realastypeint)xys     T/var/www/html/venv/lib/python3.13/site-packages/scipy/special/tests/test_lambertw.pywtest_values.<locals>.wQ   s    66==-..    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_valuesr:      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&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/		"%U;AAC 
#	"	"s   G55
Hc                  X    [        [        [        S[        S4   5      [        S   5        g )N        r   )r<   r   r   )r   r   r   r    r0   r-   
test_ufuncr>   W   s%    B2I$@!ACr0   c                     [         R                  " [         R                  5      n [        [	        SSS5      R                  U 5        [        [	        S/SS5      R                  U 5        [        [	        SS/S5      R                  U 5        [        [	        SSS/5      R                  U 5        [        [	        S/S/S/5      R                  U 5        g )Nr   )r6   r%   r   r   r   )dts    r-   "test_lambertw_ufunc_loop_selectionrA   \   s    	"--	 B!Q"(("-1#q!$**B/!aS!$**B/!Q$**B/1#sQC(..3r0   z)ga4    y            y+q          c                 (    [        U 5      nX:X  d   eg r'   r   )rB   r.   s     r-   test_lambertw_subnormal_k0rD   f   s     	A 6M6r0   )pytestnumpyr6   numpy.testingr   r   r   scipy.specialr   r   r	   r
   r   r   r   r   r   r   scipy.special._testutilsr   r:   r>   rA   markparametrizerD   r=   r0   r-   <module>rL      s\      J J " D D D -BDJC
4 AB Cr0   