B
    ²ô`£  ã               @   sX   d dl mZmZmZmZ d dlZd dlmZm	Z	m
Z
 dd„ Zdd„ Zdd	„ Zd
d„ ZdS )é    )Ú	betabinomÚ	hypergeomÚ	bernoulliÚ	boltzmannN)Úassert_almost_equalÚassert_equalÚassert_allclosec        
      C   sÌ   d} d}d}d}t  | |||¡}t  ||  ||| |¡}t  ||  |||| ¡}t  | |||¡}t||dd t||dd t||dd d} d}d}d}t  | |||¡}t | || ¡}	t||	dd d S )Né   é2   é
   é   )Údecimalé   é   )r   Zlogpmfr   r   )
ÚkÚNÚKÚnZlogpmf1Zlogpmf2Zlogpmf3Zlogpmf4Zhypergeom_logpmfZbernoulli_logpmf© r   úU/home/dcms/DCMS/lib/python3.7/site-packages/scipy/stats/tests/test_discrete_distns.pyÚtest_hypergeom_logpmf   s$    r   c              C   s¢   t  dd¡} d}t | d|¡}| dk}t||ƒ t  d¡}d}t | ||¡}ddddd	d
ddg}t||dd t | ||¡}ddddddddg}t||dd d S )Néýÿÿÿr	   r   g°rh‘í|¿?r   é   é   g’$I’$Iâ?g’$I’$IÒ?g’$I’$IÂ?g‚vIhÂ%<=)ZrtolgÛ¶mÛ¶më?)ÚnpÚaranger   Úpmfr   Úlogr   Zcdf)r   r   ÚpÚexpectedZlamÚcr   r   r   Útest_boltzmann_upper_bound    s    

r!   c              C   sJ   d} t  | d ¡}t| ddƒ |¡}t  d| d  | d ¡}t||ƒ d S )Né   r   )r   r   r   r   Úrepeatr   )r   r   r   r   r   r   r   Útest_betabinom_a_and_b_unity3   s
    r$   c              C   sH   d} d}t  d¡}td| |ƒ |¡}t| | |  ƒ |¡}t||ƒ d S )Ngffffff@g)\Âõ(ä?r   r   )r   r   r   r   r   r   )ÚaÚbr   r   r   r   r   r   Útest_betabinom_bernoulli=   s    
r'   )Zscipy.statsr   r   r   r   Únumpyr   Znumpy.testingr   r   r   r   r!   r$   r'   r   r   r   r   Ú<module>   s   
