ó
áp7]c           @   s‘   d  d l  j Z d  d l m Z m Z d  d l m Z d e f d „  ƒ  YZ	 d e f d „  ƒ  YZ
 d e f d	 „  ƒ  YZ d
 e f d „  ƒ  YZ d S(   iÿÿÿÿN(   t   poissont   nbinom(   t   assert_allcloset   TestGenpoisson_pc           B   s;   e  Z d  Z d „  Z d „  Z d „  Z d „  Z d „  Z RS(   s/   
    Test Generalized Poisson Destribution
    c         C   sG   t  j d d ƒ } t j j j d d d d ƒ } t | | d d ƒd  S(   Ni   i    t   rtolgVçž¯Ò<(   R    t   pmft   smt   distributionst   genpoisson_pR   (   t   selft   poisson_pmft   genpoisson_pmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_pmf_p1   s    c         C   sG   t  j d d ƒ } t j j j d d d d ƒ } t | | d d ƒd  S(   Ni   i    R   gVçž¯Ò<(   R    R   R   R   R   R   (   R	   R
   R   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_pmf_p2   s    c         C   sG   t  j d d ƒ } t j j j d d d d ƒ } t | | d d ƒd  S(   Ni
   i   gÙ}ÚõÐò¾:i   R   gê-™—q=(   R    R   R   R   R   R   (   R	   R
   t   genpoisson_pmf_5(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_pmf_p5   s    c         C   sG   t  j d d ƒ } t j j j d d d d ƒ } t | | d d ƒd  S(   Ni   i   i    i   R   gVçž¯Ò<(   R    t   logpmfR   R   R   R   (   R	   R
   R   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_logpmf_p1   s    c         C   sG   t  j d d ƒ } t j j j d d d d ƒ } t | | d d ƒd  S(   Ni   i   i    i   R   gVçž¯Ò<(   R    R   R   R   R   R   (   R	   R
   R   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_logpmf_p2   s    (   t   __name__t
   __module__t   __doc__R   R   R   R   R   (    (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR      s   				t   TestZIPoissonc           B   s2   e  Z d  Z d „  Z d „  Z d „  Z d „  Z RS(   s   
    c         C   sD   t  j d d ƒ } t j j j d d d ƒ } t | | d d ƒd  S(   Ni   i   i    R   gê-™—q=(   R    R   R   R   t	   zipoissonR   (   R	   R
   t   zipoisson_pmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_pmf_zero'   s    c         C   sD   t  j d d ƒ } t j j j d d d ƒ } t | | d d ƒd  S(   Ni   i   i    R   gê-™—q=(   R    R   R   R   R   R   (   R	   t   poisson_logpmft   zipoisson_logpmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_logpmf_zero,   s    c         C   sJ   t  j d d ƒ } t j j j d d d ƒ } t | | d d d d ƒd  S(   Ni   gš™™™™™¹?R   gš™™™™™©?t   atol(   R    R   R   R   R   R   (   R	   R
   R   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_pmf1   s    c         C   sJ   t  j d d ƒ } t j j j d d d ƒ } t | | d d d d ƒd  S(   Ni   i   gš™™™™™¹?R   gš™™™™™©?R   (   R    R   R   R   R   R   (   R	   R   R   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   test_logpmf6   s    (   R   R   R   R   R   R   R   (    (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   $   s
   			t   TestZIGeneralizedPoissonc           B   s,   e  Z d  „  Z d „  Z d „  Z d „  Z RS(   c         C   sV   t  j j j d d d d ƒ } t  j j j d d d d d ƒ } t | | d d ƒd  S(   Ni   i   i   i    R   gê-™—q=(   R   R   R   R   t   zigenpoissonR   (   R	   t   gp_pmft   zigp_pmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   =   s    !c         C   sV   t  j j j d d d d ƒ } t  j j j d d d d d ƒ } t | | d d ƒd  S(   Ni   i   i   i    R   gê-™—q=(   R   R   R   R   R!   R   (   R	   t	   gp_logpmft   zigp_logpmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   B   s    !c         C   s\   t  j j j d d d d ƒ } t  j j j d d d d d ƒ } t | | d d d d ƒd  S(   Ni   i   gš™™™™™¹?R   gš™™™™™©?R   (   R   R   R   R   R!   R   (   R	   R"   R#   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   G   s    !c         C   s\   t  j j j d d d d ƒ } t  j j j d d d d d ƒ } t | | d d d d ƒd  S(   Ni   i   i    gš™™™™™¹?R   gš™™™™™©?R   (   R   R   R   R   R!   R   (   R	   R$   R%   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   L   s    !(   R   R   R   R   R   R   (    (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR    <   s   			t	   TestZiNBPc           B   s2   e  Z d  Z d „  Z d „  Z d „  Z d „  Z RS(   s-   
    Test Truncated Poisson distribution
    c         C   st   t  j j j d d d ƒ \ } } t j d | | ƒ } t  j j j d d d d d ƒ } t | | d d d d ƒd  S(	   Ni   gš™™™™™¹?i   id   g{®Gáz„?R   gñhãˆµøä>R   (   R   R   t   zinegbint   convert_paramsR   R   R   (   R	   t   nt   pt   nb_pmft   tnb_pmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   V   s    !!c         C   st   t  j j j d d d ƒ \ } } t j d | | ƒ } t  j j j d d d d d ƒ } t | | d d d d ƒd  S(   Ni
   i   i   iÈ   g{®Gáz„?R   R   (   R   R   R'   R(   R   R   R   (   R	   R)   R*   t	   nb_logpmft
   tnb_logpmf(    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   \   s    !!c         C   sr   t  j j j d d d ƒ \ } } t j d | | ƒ } t  j j j d d d d d ƒ } t | | d d d ƒd  S(   Ni   gÍÌÌÌÌÌì?i   g      à?R   gH¯¼šò×z>(   R   R   R'   R(   R   R   R   (   R	   R)   R*   R-   R,   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   b   s    !!c         C   st   t  j j j d d d ƒ \ } } t j d | | ƒ } t  j j j d d d d d ƒ } t | | d d d d ƒd  S(   Ni   i   i   g{®Gázt?R   g{®Gáz„?R   (   R   R   R'   R(   R   R   R   (   R	   R)   R*   R-   R.   (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR   h   s    !!(   R   R   R   R   R   R   R   (    (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyR&   R   s
   			(   t   statsmodels.apit   apiR   t   scipy.statsR    R   t   numpy.testingR   t   objectR   R   R    R&   (    (    (    sL   lib/python2.7/site-packages/statsmodels/distributions/tests/test_discrete.pyt   <module>   s   