ó
î&]\c           @` sv   d  d l  m Z m Z m Z d  d l m Z m Z m Z d  d l Z	 d  d l
 m Z m Z m Z d „  Z d „  Z d S(   i    (   t   divisiont   print_functiont   absolute_import(   t	   hypergeomt	   bernoullit	   boltzmannN(   t   assert_almost_equalt   assert_equalt   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(   Ni   i2   i
   t   decimali   i   i   (   R   t   logpmfR   R   (
   t   kt   Nt   Kt   nt   logpmf1t   logpmf2t   logpmf3t   logpmf4t   hypergeom_logpmft   bernoulli_logpmf(    (    sE   lib/python2.7/site-packages/scipy/stats/tests/test_discrete_distns.pyt   test_hypergeom_logpmf   s$      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(   Niýÿÿÿi   i   g°rh‘í|¿?i    i   i   i   i   t   rtolg‚vIhÂ%<=i   g’$I’$Iâ?g’$I’$IÒ?g’$I’$IÂ?g’$I’$Iâ?gÛ¶mÛ¶më?(   t   npt   arangeR   t   pmfR   t   logR   t   cdf(   R   R   t   pt   expectedt   lamt   c(    (    sE   lib/python2.7/site-packages/scipy/stats/tests/test_discrete_distns.pyt   test_boltzmann_upper_bound"   s    (   t
   __future__R    R   R   t   scipy.statsR   R   R   t   numpyR   t   numpy.testingR   R   R   R   R    (    (    (    sE   lib/python2.7/site-packages/scipy/stats/tests/test_discrete_distns.pyt   <module>   s
   	