o
    EbW                     @   sP   d dl m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ddtjd d d dd d S )N         -q=rtol)r   sczetanppi r   r   ?/usr/lib/python3/dist-packages/scipy/special/tests/test_zeta.py	test_zeta   s   (r   c                  C   s(   g d} g d}t t| |dd d S )N)g g?gH.?	   2   K   )g^ңge
gp#	6gj-s`?gXk   <g    @;r   r   )r   r
   zetac)xdesiredr   r   r   
test_zetac
   s   r   c                   C   sj   t tjdks
J tt tj sJ t ddksJ t dtjks)J tt g dd d S )Nr   g      g      ?)ii)r
   r   r   infisnanr   r   r   r   r   test_zetac_special_cases   s
   r   c                   C   s   t tt jsJ tt jdksJ tddksJ ttg dd ttdt jd d dd ttd	t jd	 d
 dd d S )Nr   r   g      )r   iiir   r   r   r      Z   )	r   r   r
   r   nanr   r   r   r   r   r   r   r   test_riemann_zeta_special_cases"   s   "r!   c                  C   s"   d} d}t t| |ddd d S )Ng    @pgkr   gvIh%,=)Zatolr	   )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    	