3
Od                 @   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 j| |||}t j||  ||| |}t j||  |||| }t j| |||}t||dd t||dd t||dd d} d}d}d}t j| |||}tj| || }	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   Z/var/www/html/virt/lib64/python3.6/site-packages/scipy/stats/tests/test_discrete_distns.pytest_hypergeom_logpmf   s$    r   c              C   s   t jdd} d}tj| d|}| dk}t|| t jd}d}tj| ||}ddddddddg}t||d	d
 tj| ||}ddddddddg}t||d	d
 d S )N   r	   r   grh|?r         r   gvIh%<=)Zrtol   g$I$I?g$I$I?g$I$I?g$I$I?g۶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 j| d }t| ddj|}t j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 jd}td| |j|}t| | |  j|}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   Znumpyr   Znumpy.testingr   r   r   r   r#   r&   r)   r   r   r   r   <module>   s   
