3
Odi                 @   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           3   C   sx  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 dUdWtdtfddXt fddt fddt ft	ddfdYt d ddt d ft
d d dt
d fdZd]dadedidmdpdtdwd{dddddddddddddddddddddddtddFfddddg0} t| tdM} dNdO }tjdPdQ t|| dddRdSdTj  W d Q R X d S )N         g      ?r   y                W	&?y              ?      ?烝{?g?i'  g"{,7?gѤX9@gHgtm@y        3]N@g}@y        &@   gӸ?y        ~dY?gg[?y        @  r6@gAj@y        T'>@   g(o-@y        n!=@y              @gAF?y        Z|?g?y        ?g)u?y        %+?g(t?y        mȻy@gO^?y        "@g      ?g~}?h?y        e?g:㢥'@y        %K8@g8	U'@i gk)j^-@y        xe/:@gb/.@y        >ꮺF@@
   g0j;?y        "B`?gm.[	@y        <?@gsqM@y        1mP@gE&\@y        57@g#u*?y        -pF?g>M|@y        Wn2?gǁc@y        'FyE@gfI3@y        I5fԪ4@g7-?y        Mb`?g7W@?y        ;ūԎ?gy&1?g)!faY?y        NJuM?)dtypec             S   s   t | |jjtS )N)r   realZastypeint)xy r    U/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_lambertw.pywP   s    ztest_values.<locals>.wignore)allg|=gvIh%<=)ZrtolZatol)r   r   r                   )r%   r   r   )r   r   r   )r   r   r         пֿ)r'   r   r(   g      -C6"{,7)r)   r   r*         пr&   ѤX9)r+   r&   r,   r&   gHgtmHgtm3]N)r   r&   r-         пr&   ѤX9)r.   r&   r/   gHgtmHgtm3]N@)r   r   r0         пg}}&@)r1   r   r2   Ӹ?~dY?)r3   r   r4   r3   gg[g[忼@)r3   r   r5   r3   r&   Ӹ?~dY)r3   r&   r6   )r   r   r   Aj@T'>@)r   r   r7   r&   Aj@T'>)r   r&   r8   (o-@n!=@)r   r   r9         @      @AF?Z|?)r:   r   r;   gٿٿ?g)u)u%+?)r<   r   r=         @      @g(t(tmȻy@)r>   r   r?         @      @r&   O^?")r@   r&   rA         r&   g~}?h~}?he)rB   r&   rC   g      -C6g:㢥':㢥'%K8@)rD   r   rE   g      -C6r&   8	U')rF   r&   rG   g      -C6g:㢥':㢥'%K8)rH   rI   rJ   g      h㈵gk)j^-k)j^-xe/:@)rK   r   rL   g      h㈵gb/.b/.>ꮺF@@)rM   r   rN   y       @      ???0j;?"B`?)rO   r   rP   y       @      ???gm.[	m.[	<?@)rQ   r   rR   y       @      ???r&   gsqMsqM1mP)rS   r&   rT   y       @      ???gE&\E&\57@)rU   r   rV   y       @      ?y             ɿg#u*Ϳ#u*Ϳ-pFſ)rW   r   rX   y       @      ?y             ɿg>M|>M|Wn2?)rY   r   rZ   y       @      ?y             ɿr&   gǁcǁc'FyE)r[   r&   r\   y       @      ?y             ɿgfI3fI3I5fԪ4@)r]   r   r^   g            Mb`?g7W@7W@;ūԎ?)r_   r   r`   g            Mb`g7W@7W@;ūԎ)ra   r   rb   gy&1ܿy&1ܿ?g)!faY)!faYNJuM?)rc   r   rd   gy&1ܿy&1ܿٿg)!faY)!faYNJuM)re   r   rf   )r   r   )r   r
   r   r   r   r   r   imagr   r	   r   r   r   npZerrstater   check)datar"   r    r    r!   test_values   st    


rk   c               C   s    t ttdtdf td  d S )N              ?W	&?)rl   rm   rn   )r   r   r   r	   r    r    r    r!   
test_ufuncV   s    ro   c              C   s   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   )rh   r   Z
complex128r   r   )dtr    r    r!   "test_lambertw_ufunc_loop_selection[   s    rq   )Znumpyrh   Znumpy.testingr   r   r   Zscipy.specialr   r   r   r   r	   r
   r   r   r   r   Zscipy.special._testutilsr   rk   ro   rq   r    r    r    r!   <module>	   s   ,E