3
OdW                 @   sN   d dl jZd dlZd dlmZmZ dd Zdd Z	dd Z
d	d
 Zdd ZdS )    N)assert_equalassert_allclosec               C   s(   t tjddtjd d d dd d S )N         g-q=)rtol)r   sczetanppi r   r   Q/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_zeta.py	test_zeta   s    r   c              C   s8   ddddddg} dddd
ddg}t tj| |dd d S )Ng @g?gH.?	   2   K   g^ң?ge
@gp#	6@gj-s`?gXk   <g    @;g-q=)r   g g^ңge
gp#	6)r   r   zetac)xdesiredr   r   r   
test_zetac
   s    r   c               C   sl   t jtjdksttjt jtj s,tt jddks>tt jdtjksRttt jd	d
dgd d S )Nr   g      ?g      ?r   r   d   r   g      ii)r   r   r
   infAssertionErrorisnanr   r   r   r   r   test_zetac_special_cases   s
    r   c               C   s   t jtjt jsttjt jdks*ttjddks<tttjdddddgd ttjdt j	d d d	d
 ttjdt j	d d d	d
 d S )Nr   r   g      ?r      r      
   g-q=)r   Z   g      r   iii)
r
   r   r   r	   nanr   r   r   r   r   r   r   r   r   test_riemann_zeta_special_cases"   s    r#   c              C   s"   d} d}t tj| |ddd d S )Ng    @p@gk}r   gvIh%,=)Zatolr   g    @pgk)r   r   r	   )sr   r   r   r    test_riemann_zeta_avoid_overflow.   s    r%   )Zscipy.specialZspecialr   Znumpyr
   Znumpy.testingr   r   r   r   r   r#   r%   r   r   r   r   <module>   s   
	