ó
áp7]c           @   sz   d  Z  d d l m Z d d l Z d d l m Z d d l Z d d l m	 Z	 d d l
 j j Z d e f d „  ƒ  YZ d S(   s=   
Created on Fri Sep 15 13:38:13 2017

Author: Josef Perktold
iÿÿÿÿ(   t   divisionN(   t   assert_allclose(   t   Poissont   TestCountDiagnosticc           B   s5   e  Z e d  „  ƒ Z d „  Z e j j d „  ƒ Z RS(   c   	      C   så   d d d g } t  j j d ƒ d } t  j | d f ƒ } d | d  | d … d f <d } t  j | j | d  ƒ | ƒ } t  j j | d ƒ } t | | ƒ } | j d	 d
 d d d d ƒ } | |  _	 | |  _
 | |  _ | |  _ d  S(   Ni   g      à?ió iô  i   i    iÿÿÿÿi   t   methodt   bfgst   maxiteriˆ  t   maxfun(   t   npt   randomt   seedt   onest   expt   dott   poissonR   t   fitt   exogt   endogt   rest   nobs(	   t   clst   expected_paramsR   R   t   offsett   mu_truet	   endog_poit	   model_poit   res_poi(    (    sI   lib/python2.7/site-packages/statsmodels/discrete/tests/test_diagnostic.pyt   setup_class   s     			c         C   s‚   t  j |  j ƒ } t  j |  j ƒ } t | d  | d d | d f d d ƒt  j |  j |  j ƒ } d	 } t | | d d ƒd  S(
   Ni   i    i   t   rtolgñhãˆµøä>glÿ	
mŽé?gkW
wå?gü©ñÒMb@?(   glÿ	
mŽé?gkW
wå?i   i   (   t   diat   test_poisson_zeroinflationR   t    test_poisson_zeroinflation_brockR   R   (   t   selft   tzi1t   tzi2t   tzi3t   tzi3_1(    (    sI   lib/python2.7/site-packages/statsmodels/discrete/tests/test_diagnostic.pyt
   test_count/   s    )c         C   s“   |  j  } |  j j ƒ  } t j |  j ƒ | } t j |  j | d  d  … d  d … f ƒ } d } t | d  | d d ƒt j	 | | j
 d ƒ ƒ d  S(   Ni   g™gÎ<ÑØ?gø$hö6á?R   g-Cëâ6
?i    (   g™gÎ<ÑØ?gø$hö6á?(   R   R   t   predict_probR   t   bincountR   R   t   test_chisquare_probR   t
   plot_probst   mean(   R    t   close_figuresR   t   probst   freqt   tziR!   (    (    sI   lib/python2.7/site-packages/statsmodels/discrete/tests/test_diagnostic.pyt
   test_probs=   s    	+(	   t   __name__t
   __module__t   classmethodR   R%   t   pytestt   markt
   matplotlibR/   (    (    (    sI   lib/python2.7/site-packages/statsmodels/discrete/tests/test_diagnostic.pyR      s   	(   t   __doc__t
   __future__R    t   numpyR   t   numpy.testingR   R3   t#   statsmodels.discrete.discrete_modelR   t'   statsmodels.discrete._diagnostics_countt   discretet   _diagnostics_countR   t   objectR   (    (    (    sI   lib/python2.7/site-packages/statsmodels/discrete/tests/test_diagnostic.pyt   <module>   s   