o
    Ebi                     @   s|   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dS )    N)assert_assert_equalassert_array_almost_equal)lambertw)	naninfpieisnanlogr_arraycomplex_)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   realZastypeint)xy r'   C/usr/lib/python3/dist-packages/scipy/special/tests/test_lambertw.pywP   s   ztest_values.<locals>.wignore)all)r   r   g|=gvIh%<=)ZrtolZatol)r   r
   r   r   r   r   r#   imagr   r	   r   r   r   npZerrstater   check)datar)   r'   r'   r(   test_values   s   


	
 !"#$%&'()*+,/0124"r0   c                   C   s    t ttdtdf td  d S )N        r   )r1   r   r   )r   r   r   r	   r'   r'   r'   r(   
test_ufuncV   s   r2   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   )r-   r"   Z
complex128r   r   )Zdtr'   r'   r(   "test_lambertw_ufunc_loop_selection[   s   r3   )Znumpyr-   Znumpy.testingr   r   r   Zscipy.specialr   r   r   r   r	   r
   r   r   r   r   Zscipy.special._testutilsr   r0   r2   r3   r'   r'   r'   r(   <module>   s   ,E