3
OédÖ  ã               @   s\   d dl Zd dlmZmZmZ d dlmZ d dl	m
Z
mZ G dd„ deƒZdd„ Zd	d
„ ZdS )é    N)Úassert_array_equalÚassert_array_almost_equal_nulpÚassert_almost_equal)Úraises)ÚgammalnÚmultigammalnc               @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestMultiGammaLnc             C   s4   t jjdƒ t jt jjƒ ƒ}tt|dƒt|ƒƒ d S )NiÒ  é   )ÚnpÚrandomÚseedÚabsÚrandnr   r   r   )ÚselfÚa© r   úX/var/www/html/virt/lib64/python3.6/site-packages/scipy/special/tests/test_spfun_stats.pyÚtest1   s    zTestMultiGammaLn.test1c             C   sL   t jddgƒ}t|dƒ}t jt jt jƒƒt|ƒ t|d ƒ }t||ƒ d S )Ng      @g      $@é   g      à?)r
   Úarrayr   ÚlogÚsqrtÚpir   r   )r   r   ÚresultÚexpectedr   r   r   Útest2   s    
&zTestMultiGammaLn.test2c             C   s   t ttddƒ d S )Ng      à?g333333ó?)Úassert_raisesÚ
ValueErrorr   )r   r   r   r   Útest_bararg   s    zTestMultiGammaLn.test_barargN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r   	   s   r   c             C   sX   t | |ƒ}t| j|jƒ | jƒ }|jƒ }x*t| jƒD ]}t|| t || |ƒƒ q4W d S )N)r   r   ÚshapeZravelÚrangeÚsizer   )r   Údr   Za1Zresult1Úir   r   r   Ú _check_multigammaln_array_result   s    
r'   c              C   s–   t jjdƒ t jt jjddƒƒd dft jt jjddƒƒd dft jddƒjdddƒdft jdgƒdft jdƒdfg} x| D ]\}}t	||ƒ q|W d S )	NiÒ  é   r   é   r	   g      $@g      2@g       @)
r
   r   r   r   r   ZarangeZreshaper   Zfloat64r'   )Zcasesr   r%   r   r   r   Útest_multigammaln_array_arg*   s    r*   )Znumpyr
   Znumpy.testingr   r   r   Zpytestr   r   Zscipy.specialr   r   Úobjectr   r'   r*   r   r   r   r   Ú<module>   s   