U
    HvfW                     @   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   A/tmp/pip-unpacked-wheel-96ln3f52/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| |dd d S )Ng 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   sl   t tjdksttt tj s,tt ddks>tt dtjksRttt dddgd d S )Nr   g      g      ?ii)r
   r   r   infAssertionErrorisnanr   r   r   r   r   test_zetac_special_cases   s
    r   c                   C   s   t tt jsttt jdks*ttddks<tttdddddg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   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.specialspecialr
   Znumpyr   Znumpy.testingr   r   r   r   r   r"   r$   r   r   r   r   <module>   s   	