3
Od
                 @   sF   d dl Z d dlZd dlmZ d dljZd dlm	Z	 G dd de
ZdS )    N)assert_allclose)FuncDatac               @   s   e Zd Zejjdejddfdejdfddejfdejdfejddfddejfejddfejddfgdd Zejjdej	 dfej	dfgdd Z
d	d
 Zdd Zejjddddej	fdgdd Zejjddddgdd ZdS )TestVoigtProfilezx, sigma, gamma   r   c             C   s   t jtj|||std S )N)npisnanscvoigt_profileAssertionError)selfxsigmagamma r   U/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_faddeeva.pytest_nan   s    zTestVoigtProfile.test_nanz
x, desiredc             C   s   t j|dd|kstd S )Nr   )r   r	   r
   )r   r   desiredr   r   r   test_inf   s    zTestVoigtProfile.test_infc             C   s   t jd/dddgd0dddgd1d
ddgd2dddgddddgd3dddgd4dddgdddd gd!d"d#d$gd5d&d'd(gg
}ttj|d6d,d)d-d.j  d S )7Ng(\@gHzF@gp=
ף@gk?g?gQ@gzG!8@ge9?g(\+@gGz0@g(\/E@g3y?gRQ)@gzG?gHzG@g,<i?gGz&@g      @g(\5@gI?gQ'@gffffff4@gHz>@gxD?gףp=
W"@g\(9@gp=
ף @g
ڇ?gףp=
0@g2@g      @gZA.Y?gQ8"@g(\ @g
ףp=C@g|͵d?g
ףp=E@g333333?gףp=
F@ghsn?r   r         gV瞯<)atolrtolg(\gg(\+gRQ)gQ'gףp=
W"g
ףp=E)r   r   r   )r   arrayr   r   r	   check)r   Zpointsr   r   r   test_against_mathematica   s$    








z)TestVoigtProfile.test_against_mathematicac             C   s8   t jddd}ttj|ddtj| ddddd d S )Nr   
      r   gV瞯<)r   r   )r   linspacer   r   r	   )r   r   r   r   r   test_symmetry9   s    zTestVoigtProfile.test_symmetryzx, sigma, gamma, desiredc             C   s   t j||||kstd S )N)r   r	   r
   )r   r   r   r   r   r   r   r   test_corner_casesB   s    z"TestVoigtProfile.test_corner_caseszsigma1, gamma1, sigma2, gamma2缉ؗҜ<c             C   s6   t jddd}ttj|||tj|||ddd d S )Nr   r   r   gؗҜ<)r   r   )r   r   r   r   r	   )r   Zsigma1Zgamma1Zsigma2Zgamma2r   r   r   r   test_continuityI   s    z TestVoigtProfile.test_continuityN)r   r   r   r   )r   r   r    r   )r   r   r   r    )r   r   r    r    )__name__
__module____qualname__pytestmarkZparametrizer   nanr   infr   r   r   r   r!   r   r   r   r   r   	   s*   







	r   )r%   Znumpyr   Znumpy.testingr   Zscipy.specialZspecialr   Zscipy.special._testutilsr   objectr   r   r   r   r   <module>   s
   
