3
Od                 @   sl   d dl Zd dlmZmZ d dlmZ d dlmZm	Z	m
Z
 dd Zdd Zd	d
 Zdd Zdd Zdd ZdS )    N)assert_allcloseassert_)FuncData)gammagammalnloggammac              C   sv   t jddddddg} | j }t j| |\} }| d|  j }t j|t|fj}dd }t||ddd	d	d
j	  d S )Ng     X@g      #@g      ?y              ?c             S   s   t jt| S )N)npexpr   )z r   U/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_loggamma.pyf   s    ztest_identities1.<locals>.fr      g+=)rtolatolg     Xg      #g      )
r   arraycopymeshgridflattenvstackr   Tr   check)xyr
   datasetr   r   r   r   test_identities1   s    r   c              C   s   t jddddddg} | j }t j| |\} }| d|  j }t j|t j|t| fj}dd }t	||ddd	d	d
j
  d S )Ng     X@g      #@g      ?y              ?c             S   s   t | d S )Nr   )r   )r
   r   r   r   r      s    ztest_identities2.<locals>.fr   r   g+=)r   r   g     Xg      #g      )r   r   r   r   r   r   logr   r   r   r   )r   r   r
   r   r   r   r   r   test_identities2   s    r   c              C   s\   t jt jdd	 t jd
df d } t j| t| fj}dd }t||dddddj  d S )N
   g      ?c             S   s   t j| dd} t| jS )NZ
complex128)Zdtype)r   r   r   real)r
   r   r   r   r   +   s    z)test_complex_dispatch_realpart.<locals>.fr   r   g+=)r   r   ii)r   Zr_logspacer   r   r   r   r   )r   r   r   r   r   r   test_complex_dispatch_realpart$   s    $r!   c              C   sd   t jddd } t j| t| fj}tt|dddddj  ttdt j	k tt j
td d S )	Nr   g      ?r   r   g+=)r   r   i)r   r    r   r   r   r   r   r   r   infisnan)r   r   r   r   r   test_real_dispatch2   s
    r%   c              C   s4   t tdd} t tdd	}t| |j ddd d S )
Ng333333@        gV瞯<r   )r   r   g333333r&   g333333g       )r   complexr   	conjugate)r
   zbarr   r   r   test_gh_6536;   s    r*   c              C   sP   t jdd
d } t jdd | D }t jdd | D }t||j ddd	 d S )Ni,     d   c             S   s   g | ]}t |d qS )g        )r'   ).0x0r   r   r   
<listcomp>D   s    z#test_branch_cut.<locals>.<listcomp>c             S   s   g | ]}t |dqS )g        g       )r'   )r-   r.   r   r   r   r/   E   s    gV瞯<r   )r   r   i)r   r    Zasarrayr   r(   )r   r
   r)   r   r   r   test_branch_cutA   s    r0   )Znumpyr   Znumpy.testingr   r   Zscipy.special._testutilsr   Zscipy.specialr   r   r   r   r   r!   r%   r*   r0   r   r   r   r   <module>   s   	