o
    ?Hh                     @   s   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d Zdd	 Zd
d Ze jdg ddd ZdS )    N)assert_assert_equalassert_array_almost_equallambertw)	naninfpieisnanlogr_array
complex128)FuncDatac               	   C   s  t ttt tttdjt tttdjdt  ttt djt ttt djdt  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 dd
t d ft
d d 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d0} d1d2 }tjd3d4 t|| d5dd6d7d8  W d    d S 1 sw   Y  d S )9N               ?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                 S   s   t | |jtS N)r   realastypeint)xy r*   a/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/special/tests/test_lambertw.pywQ   s   ztest_values.<locals>.wignore)all)r   r   g|=gvIh%<=)rtolatol)r   r   r   r   r   r   r%   imagr	   r
   r   r   r   nperrstater   check)datar,   r*   r*   r+   test_values   s   


	
 !"#$%&'()*+,/0124"r6   c                   C   s    t ttdtdf td  d S )N        r   )r7   r   r   )r   r   r   r
   r*   r*   r*   r+   
test_ufuncW   s   r8   c                  C   s   t 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   )r2   r#   r   r   r   )dtr*   r*   r+   "test_lambertw_ufunc_loop_selection\   s   r:   z)ga4    y            y+q          c                 C   s   t | }|| ks
J d S r$   r   )r;   r,   r*   r*   r+   test_lambertw_subnormal_k0f   s   r<   )pytestnumpyr2   numpy.testingr   r   r   scipy.specialr   r   r   r	   r
   r   r   r   r   r   scipy.special._testutilsr   r6   r8   r:   markparametrizer<   r*   r*   r*   r+   <module>   s   ,E
