U
    Hvf                     @   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 ddddddg} |  }t | |\} }| d|   }t |t|fj}dd	 }t||d
dddd	  d S )N     X      #            ?      #@     X@              ?c                 S   s   t t| S )N)npexpr   z r   E/tmp/pip-unpacked-wheel-96ln3f52/scipy/special/tests/test_loggamma.pyf   s    ztest_identities1.<locals>.fr      +=ZrtolZatol)
r   arraycopymeshgridflattenvstackr   Tr   checkxyr   datasetr   r   r   r   test_identities1   s    r$   c                  C   s   t ddddddg} |  }t | |\} }| d|   }t |t |t| fj}dd	 }t	||d
dddd
  d S )Nr   r	   r
   r   r   r   r   c                 S   s   t | d S )Nr   )r   r   r   r   r   r      s    ztest_identities2.<locals>.fr   r   r   r   )r   r   r   r   r   r   logr   r   r   r   r    r   r   r   test_identities2   s    r&   c                  C   s\   t jt dd t ddf d } t | t| fj}dd }t||ddddd	  d S )
N
   r   c                 S   s   t j| dd} t| jS )NZ
complex128)Zdtype)r   r   r   realr   r   r   r   r   +   s    z)test_complex_dispatch_realpart.<locals>.fr   r   r   r   )r   Zr_logspacer   r   r   r   r   )r!   r#   r   r   r   r   test_complex_dispatch_realpart$   s    $r+   c                  C   sd   t ddd } t | t| fj}tt|ddddd  ttdt j	k tt 
td d S )	Nr(   r'   r   r   r   r   r   )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| | ddd d S )Ng333333               V瞯<r   r   )r   complexr   	conjugate)r   zbarr   r   r   test_gh_6536;   s    r6   c                  C   sP   t ddd } t dd | D }t dd | D }t|| ddd	 d S )
Ni,  id   c                 S   s   g | ]}t |d qS )r0   r3   .0Zx0r   r   r   
<listcomp>D   s     z#test_branch_cut.<locals>.<listcomp>c                 S   s   g | ]}t |d qS )r1   r8   r9   r   r   r   r;   E   s     r2   r   r   )r   r*   Zasarrayr   r4   )r!   r   r5   r   r   r   test_branch_cutA   s    r<   )Znumpyr   Znumpy.testingr   r   Zscipy.special._testutilsr   Zscipy.specialr   r   r   r$   r&   r+   r/   r6   r<   r   r   r   r   <module>   s   	