
p7]c           @   s  d  Z  d d l m Z d d l Z d d l Z d d l Z d d l Z d d l	 m
 Z
 m Z m Z m Z m Z m Z m Z d d l Z d d l m Z m Z m Z m Z m Z m Z m Z m Z d d l m Z m Z d d l j Z  d d l! j" j Z# d d l$ m% Z% m& Z& m' Z' m( Z( d d	 l) m* Z* m+ Z+ d d
 l, m- Z- y d d l. Z. e/ Z0 Wn e1 k
 rne2 Z0 n Xd Z3 d Z4 d Z5 d Z6 d Z7 d Z8 d Z9 d Z: d e; f d     YZ< d e< f d     YZ= d e= f d     YZ> d e; f d     YZ? d e> f d     YZ@ d e> f d     YZA d e> f d     YZB d  e> f d!     YZC d" e> f d#     YZD d$ e> f d%     YZE d& e> f d'     YZF d( e> f d)     YZG d* e> f d+     YZH d, e> f d-     YZI d. e; f d/     YZJ d0 eJ f d1     YZK d2 eJ f d3     YZL d4 eJ f d5     YZM e jN jO e0 d6 d7 d8 e; f d9     Y ZP d: e; f d;     YZQ d< e; f d=     YZR d> eR f d?     YZS d@ eR f dA     YZT dB eR f dC     YZU dD eR f dE     YZV dF eR f dG     YZW dH eR f dI     YZX dJ e; f dK     YZY dL eY f dM     YZZ dN eZ f dO     YZ[ dP eZ f dQ     YZ\ dR eY f dS     YZ] dT e> e? f dU     YZ^ dV e? f dW     YZ_ dX e> e? f dY     YZ` dZ e= f d[     YZa d\ e; f d]     YZb d^ eb e= f d_     YZc d` eb e= f da     YZd db eb e= f dc     YZe dd eb e= f de     YZf df eb e= f dg     YZg dh e= f di     YZh dj eh f dk     YZi dl eh f dm     YZj dn   Zk do   Zl dp   Zm dq   Zn dr   Zo ds   Zp dt   Zq du   Zr dv   Zs dw   Zt dx   Zu dy   Zv dz e; f d{     YZw d| e; f d}     YZx d~ e; f d     YZy d e; f d     YZz d eb e= f d     YZ{ d eb e= f d     YZ| d eb e= f d     YZ} d eb e= f d     YZ~ d eR f d     YZ d e; f d     YZ d e; f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d e f d     YZ d   Z d   Z d   Z d   Z d S(   s   
Tests for discrete models

Notes
-----
DECIMAL_3 is used because it seems that there is a loss of precision
in the Stata *.dta -> *.csv output, NOT the estimator for the Poisson
tests.
i(   t   rangeN(   t   assert_t   assert_raisest   assert_almost_equalt   assert_equalt   assert_array_equalt   assert_allcloset   assert_array_less(   t   Logitt   Probitt   MNLogitt   Poissont   NegativeBinomialt
   CountModelt   GeneralizedPoissont   NegativeBinomialP(   t   _iscountt   _isdummyi   (   t   Spectort
   DiscreteL1t   RandHIEt   Anes(   t   PerfectSeparationErrort   SpecificationWarning(   t   nbinomi   i
   i	   i   i   i   i    t   CheckModelMixinc           B   s   e  Z d    Z RS(   c         C   s<   |  j  j } t j t d d  | j d d  Wd  QXd  S(   Nt   matchs   is not supported, use eithert   methodt   foo(   t   res1t   modelt   pytestt   raisest
   ValueErrort   fit_regularized(   t   selfR   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt#   test_fit_regularized_invalid_method9   s    (   t   __name__t
   __module__R$   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   4   s   t   CheckModelResultsc           B   s   e  Z d  Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d	   Z e j j d
 d d e d e  d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z RS(   sm   
    res2 should be the test results from RModelWrap
    or the results as defined in model_results_data
    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   paramst   res2t	   DECIMAL_4(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_paramsG   s    c         C   s&   t  |  j j   |  j j d d d  S(   Nt   rtolgh㈵?(   R   R   t   conf_intR)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_conf_intJ   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   tvaluesR)   t   zR*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_zstatM   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   pvaluesR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_pvaluesP   s    c         C   sG   t  |  j d  s$ t j d   n  t |  j j   |  j j t  d  S(   Nt
   cov_paramss   TODO: implement res2.cov_params(   t   hasattrR)   R   t   skipR   R   R4   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_cov_paramsS   s
    	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   llfR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_llfZ   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   llnullR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_llnull]   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   llrR)   t	   DECIMAL_3(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_llr`   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t
   llr_pvalueR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_llr_pvaluec   s    	t   reasons-   Test has not been implemented for this class.t   strictR    c         C   s
   t   d  S(   N(   t   NotImplementedError(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_normalized_cov_paramsh   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   bseR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_bsem   s    c         C   s6   t  |  j j |  j j  t  |  j j |  j j  d  S(   N(   R   R   t   df_modelR)   t   df_resid(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dofp   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   aicR)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_aict   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   bicR)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_bicw   s    c         C   s/   t  |  j j j |  j j  |  j j t  d  S(   N(   R   R   R   t   predictR(   R)   t   phatR*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predictz   s    c         C   s5   t  |  j j j |  j j d t |  j j t  d  S(   Nt   linear(	   R   R   R   RN   R(   t   TrueR)   t   yhatR*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predict_xb~   s    c         C   s;   |  j  j j |  j  j  j   } t | |  j  j t  d  S(   N(   R   R   t
   loglikeobsR(   t   sumR   R8   t
   DECIMAL_14(   R#   t   llobssum(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_loglikeobs   s    !c         C   sS   |  j  j j |  j  j  j d  } |  j  j j |  j  j  } t | | t  d  S(   Ni    (   R   R   t	   score_obsR(   RV   t   scoreR   t	   DECIMAL_9(   R#   t   jacsumR[   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_jac   s    $(   R%   R&   t   __doc__R+   R.   R1   R3   R7   R9   R;   R>   R@   R   t   markt   xfailRR   RC   RD   RF   RI   RK   RM   RP   RT   RY   R^   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR'   A   s(   																t   CheckBinaryResultsc           B   s8   e  Z d    Z d   Z d   Z e j j d    Z RS(   c         C   s    t  |  j j   |  j j  d  S(   N(   R   R   t
   pred_tableR)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_pred_table   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t	   resid_devR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_resid_dev   s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t   resid_generalizedR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_resid_generalized   s    c         C   s   |  j  j d  S(   N(   R   t   resid_response(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_resid_response   s    (	   R%   R&   Rd   Rf   Rh   R   R`   t   smokeRj   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRb      s   			t   CheckMargEffc           B   s   e  Z d  Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d	   Z d
   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z RS(   s9   
    Test marginal effects (margeff) and its options
    c         C   s   |  j  j   } t | j |  j j t  t | j |  j j t  | j	   } | d j
 } t | | j d d t | j | j j d f  d  S(   Ns   dy/dxR,   gvIh%<=i   (   R   t   get_margeffR   t   margeffR)   t   margeff_nodummy_dydxR*   t
   margeff_set   margeff_nodummy_dydx_set   summary_framet   valuesR   R   t   shapet   size(   R#   t   met   me_framet   eff(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dydxoverall   s    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   Nt   att   mean(	   R   Rm   R   Rn   R)   t   margeff_nodummy_dydxmeanR*   Rp   t   margeff_nodummy_dydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dydxmean   s
    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   t   median(	   R   Rm   R   Rn   R)   t   margeff_nodummy_dydxmedianR*   Rp   t   margeff_nodummy_dydxmedian_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dydxmedian   s
    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   t   zero(   R   Rm   R   Rn   R)   t   margeff_nodummy_dydxzeroR*   Rp   (   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dydxzero   s
    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   NR   t   dyex(	   R   Rm   R   Rn   R)   t   margeff_nodummy_dyexR*   Rp   t   margeff_nodummy_dyex_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dyexoverall   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R{   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_dyexmeanR*   Rp   t   margeff_nodummy_dyexmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dyexmean   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_dyexmedianR*   Rp   t   margeff_nodummy_dyexmedian_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dyexmedian   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_dyexzeroR*   Rp   t   margeff_nodummy_dyexzero_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_dyexzero   s
    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   NR   t   eydx(	   R   Rm   R   Rn   R)   t   margeff_nodummy_eydxR*   Rp   t   margeff_nodummy_eydx_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eydxoverall   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R{   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eydxmeanR*   Rp   t   margeff_nodummy_eydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eydxmean   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eydxmedianR*   Rp   t   margeff_nodummy_eydxmedian_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eydxmedian   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eydxzeroR*   Rp   t   margeff_nodummy_eydxzero_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eydxzero   s
    		c         C   sK   |  j  j d d  } t | j |  j j t  t | j |  j j t  d  S(   NR   t   eyex(	   R   Rm   R   Rn   R)   t   margeff_nodummy_eyexR*   Rp   t   margeff_nodummy_eyex_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eyexoverall   s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R{   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eyexmeanR*   Rp   t   margeff_nodummy_eyexmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eyexmean  s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eyexmedianR*   Rp   t   margeff_nodummy_eyexmedian_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eyexmedian  s
    		c         C   sQ   |  j  j d d d d  } t | j |  j j t  t | j |  j j t  d  S(   NRz   R   R   R   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_eyexzeroR*   Rp   t   margeff_nodummy_eyexzero_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_eyexzero  s
    		c         C   sK   |  j  j d t  } t | j |  j j t  t | j |  j j	 t  d  S(   Nt   dummy(
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_dydxR*   Rp   t   margeff_dummy_dydx_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_dydxoverall  s
    		c         C   sQ   |  j  j d d d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NRz   R{   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_dydxmeanR*   Rp   t   margeff_dummy_dydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_dydxmean"  s
    		c         C   sQ   |  j  j d d d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NR   R   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_eydxR*   Rp   t   margeff_dummy_eydx_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_eydxoverall)  s
    		c         C   sW   |  j  j d d d d d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NRz   R{   R   R   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_eydxmeanR*   Rp   t   margeff_dummy_eydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_eydxmean0  s
    !		c         C   sK   |  j  j d t  } t | j |  j j t  t | j |  j j	 t  d  S(   Nt   count(
   R   Rm   RR   R   Rn   R)   t   margeff_count_dydxR*   Rp   t   margeff_count_dydx_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_count_dydxoverall7  s
    		c         C   sQ   |  j  j d t d d  } t | j |  j j t  t | j |  j j	 t  d  S(   NR   Rz   R{   (
   R   Rm   RR   R   Rn   R)   t   margeff_count_dydxmeanR*   Rp   t   margeff_count_dydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_count_dydxmean>  s
    		c         C   sQ   |  j  j d t d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NR   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_count_dummy_dydxoverallR*   Rp   t"   margeff_count_dummy_dydxoverall_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_count_dummy_dydxoverallE  s
    		c         C   sW   |  j  j d t d t d d  } t | j |  j j t  t | j |  j j	 t  d  S(   NR   R   Rz   R{   (
   R   Rm   RR   R   Rn   R)   t   margeff_count_dummy_dydxmeanR*   Rp   t   margeff_count_dummy_dydxmean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_count_dummy_dydxmeanL  s
    !		(   R%   R&   R_   Ry   R~   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRl      s2   																							t   TestProbitNewtonc           B   s   e  Z e d     Z RS(   c         C   ss   t  j j j d t  } t  j | j d t | _ t | j | j  j	 d d d d  |  _
 t j } | |  _ d  S(   Nt	   as_pandast   prependR   t   newtont   dispi    (   t   smt   datasetst   spectort   loadt   Falset   add_constantt   exogR	   t   endogt   fitR   R   t   probitR)   (   t   clst   dataR)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   setup_classV  s
    *	(   R%   R&   t   classmethodR   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   T  s   t   TestProbitBFGSc           B   s   e  Z e d     Z RS(   c         C   ss   t  j j j d t  } t  j | j d t | _ t | j | j  j	 d d d d  |  _
 t j } | |  _ d  S(   NR   R   R   t   bfgsR   i    (   R   R   R   R   R   R   R   R	   R   R   R   R   R   R)   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   e  s    	(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   c  s   t   TestProbitNMc           B   s   e  Z e d     Z RS(   c         C   sy   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j d d d d d d  |  _ d  S(	   NR   R   R   t   nmR   i    t   maxiteri  (   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   q  s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   o  s   t   TestProbitPowellc           B   s   e  Z e d     Z RS(   c         C   sy   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j d d d d d d  |  _ d  S(	   NR   R   R   t   powellR   i    t   ftolg:0yE>(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   }  s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   {  s   t   TestProbitCGc           B   s   e  Z e d     Z RS(   c         C   s(  t  j j j d t  } t  j | j d t | _ t j } | |  _	 d d l
 m } | | j  } | | j  } t | j |  j d d d d d	 d
 d d  } | j | j  } t | | j d d d d t | j | j  j d | d d d	 d
 d d d d  |  _ t |  j j d d  d  S(   NR   R   i(   t   StandardizeTransformR   t   cgR   i    R   i  t   gtolg:0yE>R,   gh㈵>t   atolgư>t   start_paramst   fcallsid   (   R   R   R   R   R   R   R   R   R   R)   t!   statsmodels.tools.transform_modelR   R	   R   R   t   transform_paramsR(   R   R   R   t   mle_retvals(   R   R   R)   R   t   transft   exog_stt   res1_stR   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s"    					(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t   TestProbitNCGc           B   s   e  Z e d     Z RS(   c      	   C   s   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j d d d d d d d	 t  |  _ d  S(
   NR   R   R   t   ncgR   i    t   avextolg:0yE>t   warn_convergence(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t   TestProbitBasinhoppingc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j } t j j d  | d d d d d d	 d
 i d d 6d d 6 |  _ d  S(   NR   R   i   R   t   basinhoppingR   i    t   niteri   t	   minimizers   L-BFGS-Bg:0yE>t   tol(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   t   npt   randomt   seedR   (   R   R   R)   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t   TestProbitMinimizeDefaultc           B   s   e  Z e d     Z RS(   c      	   C   s   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j } | d d d d d d d	 d
  |  _ d  S(   NR   R   R   t   minimizeR   i    R   i   R   g:0yE>(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t   TestProbitMinimizeDoglegc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j } | d d d d d d d	 d
 d d  |  _ d  S(   NR   R   R   R   R   i    R   i   R   g:0yE>t
   min_methodt   dogleg(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t#   TestProbitMinimizeAdditionalOptionsc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j d d d d d d d	 d
 d d d d  |  _ d  S(   NR   R   R   R   R   i    R   i  R   s   Nelder-Meadt   xatolg-C6?t   fatol(   R   R   R   R   R   R   R   R   R   R)   R	   R   R   R   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		$(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s   t   CheckLikelihoodModelL1c           B   sD   e  Z d  Z d   Z d   Z d   Z d   Z d   Z d   Z RS(   s>   
    For testing results generated with L1 regularization
    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RE   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   t
   nnz_paramsR)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nnz_params  s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RJ   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RL   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRM     s    (	   R%   R&   R_   R+   R.   RF   R  RK   RM   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   					t   TestProbitL1c           B   s    e  Z e d     Z d   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t j	 d d d d g  } t
 | j | j  j d d d	 | d
 d d d d d d d d d  |  _ t j } | |  _ d  S(   NR   R   g?g?g333333?i
   R   t   l1t   alphaR   i    t	   trim_modet   autot   auto_trim_tolg{Gz?t   accg|=R   i  (   R   R   R   R   R   R   R   RR   R   t   arrayR	   R   R"   R   R   R   R)   (   R   R   R  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R4   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7     s    (   R%   R&   R   R   R7   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   t   TestMNLogitL1c           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } | j } t  j | d t } t  j | j |  } d t	 j
 | j d | j f  } d | d d  d   f <| j d d d	 | d
 d d d d d d d  |  _ t j } | |  _ d  S(   NR   R   g      $@i   i    iR   R  R  R	  R
  R  g{Gz?R  g|=R   (   R   R   t   anes96R   R   R   R   R
   R   R   t   onest   Jt   KR"   R   R   t   mnlogitR)   (   R   t	   anes_datat	   anes_exogt
   mlogit_modR  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	#		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   t   TestLogitL1c           B   s    e  Z e d     Z d   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ d t j	 d d d d g  |  _
 t | j | j  j d d d |  j
 d	 d
 d d d d d d d d  |  _ t j } | |  _ d  S(   NR   R   i   g        g      ?R   R  R  R   i    R	  Ru   t   size_trim_tolgh㈵>R  g|=R   i  (   R   R   R   R   R   R   R   RR   R   R  R  R   R   R"   R   R   t   logitR)   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   $  s    "	c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R4   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7   /  s    (   R%   R&   R   R   R7   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  "  s   RA   s1   Skipped test_cvxopt since cvxopt is not availablet
   TestCVXOPTc           B   s    e  Z e d     Z d   Z RS(   c         C   sP   t  s t j d  n  t j j j   |  _ t j |  j j	 d t
 |  j _	 d  S(   Ns1   Skipped test_cvxopt since cvxopt is not availableR   (   t
   has_cvxoptR   R6   R   R   R   R   R   R   R   RR   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   8  s    c         C   s   d t  j d d d d g  |  _ t |  j j |  j j  j d d d |  j d d d	 d
 d d d d  } t |  j j |  j j  j d d d |  j d d d d
 d d d d d d  } t | j	 | j	 t
  d  S(   Ng      @i    i   g      ?R   R  R  R   R  g|=R   i  R	  R
  t   l1_cvxopt_cpt   abstolR  g{Gz?(   R   R  R  R   R   R   R   R"   R   R(   R*   (   R#   t	   res_slsqpt
   res_cvxopt(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_cvxopt_versus_slsqp?  s    "!	(   R%   R&   R   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  4  s   t   TestSweepAlphaL1c           B   s    e  Z e d     Z d   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j	 | j  |  _
 t j d d d d g d d d d g d d d d g g  |  _ t j |  _ d  S(   NR   R   g?g?g      ?i   (   R   R   R   R   R   R   R   RR   R   R   R   R   R  t   alphasR   t   sweepR   (   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   M  s    c         C   s   x} t  d  D]o } |  j | d  d   f } |  j j d d d | d d d d d	 d
 d d  } t | j |  j j | t  q Wd  S(   Ni   R   R  R  R   i    R  g|=R	  t   offR   i  (   R    R"  R   R"   R   R(   R   R*   (   R#   t   iR  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_sweep_alphaX  s    (   R%   R&   R   R   R&  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR!  K  s   t   CheckL1Compatabilityc           B   sD   e  Z d  Z d   Z d   Z d   Z d   Z d   Z d   Z RS(   s   
    Tests compatability between l1 and unregularized by setting alpha such
    that certain parameters should be effectively unregularized, and others
    should be ignored by the model.
    c         C   sd   |  j  } t |  j j |  |  j j |  t  |  j j j j d } t d |  j j | | !t  d  S(   Ni   i    (	   t   mR   t	   res_unregR(   t   res_regR*   R   R   Rt   (   R#   R(  t   kvars(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   g  s
    	!c         C   s[   |  j  } t |  j j   d  |  d  |  f |  j j   d  |  d  |  f t  d  S(   N(   R(  R   R)  R4   R*  t	   DECIMAL_1(   R#   R(  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7   p  s
    	""c         C   s6   t  |  j j |  j j  t  |  j j |  j j  d  S(   N(   R   R)  RG   R*  RH   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dfx  s    c         C   s   |  j  } |  j } t |  d d  } |  j j t j t |  j j    } |  j	 j t j | |   } t
 | j |  | j |  t  t
 | j |  | j |  t  t
 t j | j |  t | j |  | j |  d d t
 t j | j |  d  S(   Nt   k_extrai    R   g~jth?(   R(  R+  t   getattrR)  t   t_testR   t   eyet   lenR(   R*  R   t   effectR=   t   sdt   nanR   t   tvalue(   R#   R(  R+  t   extrat   t_unregt   t_reg(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_t_test|  s    		'!c         C   s   |  j  } |  j } t |  d d  } |  j j t j t |  j j   |   } |  j	 j t j | |  |   } t
 | j | j d d d d t | j | j t  d  S(   NR.  i    R,   giUMu>R   gMbP?(   R(  R+  R/  R)  t   f_testR   R1  R2  R(   R*  R   t   fvalueR   t   pvalueR=   (   R#   R(  R+  R7  t   f_unregt   f_reg(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_f_test  s    		+#c         C   s,   |  j  } t t |  j j t j |   d  S(   N(   R+  R   R!   R*  R;  R   R1  (   R#   R+  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_bad_r_matrix  s    	(	   R%   R&   R_   R+   R7   R-  R:  R@  RA  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR'  a  s   						
t   TestPoissonL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   s+  d |  _  d |  _ t j j j d t  } | j j t	  j
 t | j  d  } t j | d t } | d  d   d  |  j  f } t j | j |  } | j d d d t  |  _ d t | j  t j |  j   } d	 | |  j *t j | j |  j d d
 d | d t d d d d d d  |  _ d  S(   Ni
   i   R   iR   R   R   R   i    R  R  R  g|=R   i  R	  R
  (   R+  R(  R   R   t   randhieR   R   R   t   viewt   floatt   reshapeR2  R   RR   R   R   R   R)  R   R  R"   R*  (   R   t	   rand_datat	   rand_exogt   exog_no_PSIt	   mod_unregR  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		'#(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRB    s   t#   TestNegativeBinomialL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   sh  d |  _  d |  _ t j j j d t  } | j j t	  j
 t | j  d  } | | j d  | j d  } t j | d t } | d  d   d  |  j  f } t j | j |  } | j d d d	 t  |  _ d t | j  t j |  j  d
  } d | |  j *d | d <t j | j |  } | j d d d | d	 t d d d d d d  |  _ d
 |  _ d  S(   Ni
   i   R   ii    R   R   R   R   i   R  R  R  g|=R   i  R	  R
  (   R+  R(  R   R   RC  R   R   R   RD  RE  RF  R2  R{   t   stdR   RR   R   R   R   R)  R   R  R"   R*  R.  (   R   RG  RH  t   rand_exog_stRI  RJ  R  t   mod_reg(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s"    		' '
	(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK    s   t&   TestNegativeBinomialGeoL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   sM  d |  _  d |  _ t j j j d t  } | j j t	  j
 t | j  d  } t j | d t } | d  d   d  |  j  f } t j | j | d d } | j d d	 d
 t  |  _ d t | j  t j |  j   } d | |  j *t j | j | d d } | j d d d | d
 t d d d d d d  |  _ t | j d  d  S(   Ni
   i   R   iR   t   loglike_methodt	   geometricR   R   R   i    R  R  R  g|=R   i  R	  R
  (   R+  R(  R   R   RC  R   R   R   RD  RE  RF  R2  R   RR   R   R   R   R)  R   R  R"   R*  R   RP  (   R   RG  RH  RI  RJ  R  RN  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s"    		'	#		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRO    s   t   TestLogitL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   s   d |  _  d |  _ t j j j d t  } t j | j d t	 | _ t
 j d d d d g  } t | j | j  j d d d	 | d
 d d d d d d d  |  _ | j d  d   d  |  j  f } t | j |  j d
 d d d  |  _ d  S(   Ni   i   R   R   i    i
   R   R  R  R   R  gV瞯<R   i  R	  R
  R   (   R+  R(  R   R   R   R   R   R   R   RR   R   R  R   R   R"   R*  R   R)  (   R   R   R  RI  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		"(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRR    s   t   TestMNLogitL1Compatabilityc           B   s;   e  Z e d     Z d   Z e j j d  d    Z RS(   c         C   s   d |  _  d |  _ t j j j d t  } t j | j d t	 | _ t
 j d d d d g  } t | j | j  j d d d	 | d
 d d d d d d d  |  _ | j d  d   d  |  j  f } t | j |  j d
 d d d d d d d  |  _ d  S(   Ni   i   R   R   i    i
   R   R  R  R   R  gV瞯<R   i  R	  R
  R   R   i  (   R+  R(  R   R   R   R   R   R   R   RR   R   R  R
   R   R"   R*  R   R)  (   R   R   R  RI  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		"c         C   s   |  j  } |  j } |  j j t j |   } |  j j t j |   } t | j | j |  t	  t | j
 | j
 |  t	  t t j | j
 |  t | j | j d  |  d  |  f t	  d  S(   N(   R(  R+  R)  R0  R   R1  R*  R   R3  R=   R4  R5  R6  (   R#   R(  R+  R8  R9  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR:    s    		s   Skipped test_f_test for MNLogitc         C   s   d  S(   N(    (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR@    s    (	   R%   R&   R   R   R:  R   R`   R6   R@  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRS    s   	
t   TestProbitL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   s   d |  _  d |  _ t j j j d t  } t j | j d t	 | _ t
 j d d d d g  } t | j | j  j d d d	 | d
 d d d d d d d  |  _ | j d  d   d  |  j  f } t | j |  j d
 d d d  |  _ d  S(   Ni   i   R   R   i    i
   R   R  R  R   R  gV瞯<R   i  R	  R
  R   (   R+  R(  R   R   R   R   R   R   R   RR   R   R  R	   R   R"   R*  R   R)  (   R   R   R  RI  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    		"(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT    s   t	   CompareL1c           B   s   e  Z d  Z d   Z RS(   s   
    For checking results for l1 regularization.
    Assumes self.res1 and self.res2 are two legitimate models to be compared.
    c         C   sH  t  |  j j |  j j t  t  |  j j   |  j j   t  t  |  j j   |  j j   t  t  |  j j |  j j t  t  |  j j   |  j j   t  t  |  j j	 |  j j	 t  t  |  j j
 |  j j
 t  t  |  j j |  j j t  t  |  j j |  j j t  t  |  j j |  j j t  t |  j j d t k  d  S(   Nt	   converged(   R   R   R(   R)   R*   R4   R-   R2   Rc   RE   R8   RJ   RL   R   R   RR   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_basic_results)  s    (   R%   R&   R_   RW  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRU  $  s   t   CompareL11Dc           B   s   e  Z d  Z d   Z RS(   s?   
    Check t and f tests.  This only works for 1-d results
    c         C   s   t  j t |  j j j     } t |  j j |  j |  j	 j |  j t
  t |  j j |  j |  j	 j |  j t
  d  S(   N(   R   R1  R2  R   R(   t   ravelR   R0  R=  R)   R*   R;  (   R#   t   restrictmat(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_tests?  s
    !(   R%   R&   R_   R[  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRX  ;  s   t   TestL1AlphaZeroLogitc           B   s    e  Z e d     Z d   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j	 | j  j
 d d d d d d d d	 d
 d d d d d  |  _ t | j	 | j  j d d d d	  |  _ d  S(   NR   R   R   R  R  i    R   R  gV瞯<R   i  R	  R
  R  g{Gz?R   (   R   R   R   R   R   R   R   RR   R   R   R"   R   R   R)   (   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   J  s    c         C   sW   |  j  j j d d d d d d d d d d	 d
 d d d  } t | j d t k  d  S(   NR   R  R  i    R   R  gV瞯<R   i   R	  R
  R  g{Gz?RV  (   R   R   R"   R   R   R   (   R#   t   res(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_convergedS  s    (   R%   R&   R   R   R^  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR\  G  s   	t   TestL1AlphaZeroProbitc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j	 | j  j
 d d d d d d d d	 d
 d d d d d  |  _ t | j	 | j  j d d d d	  |  _ d  S(   NR   R   R   R  R  i    R   R  gV瞯<R   i  R	  R
  R  g{Gz?R   (   R   R   R   R   R   R   R   RR   R	   R   R"   R   R   R)   (   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   _  s    (   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR_  \  s   t   TestL1AlphaZeroMNLogitc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j | j  j	 d d d d d d d d	 d
 d d d d d  |  _
 t | j | j  j d d d d	 d d d
 d  |  _ d  S(   NR   R   R   R  R  i    R   R  gV瞯<R   i  R	  R
  R  g{Gz?R   R   (   R   R   R  R   R   R   R   R
   R   R"   R   R   R)   (   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   k  s    $(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR`  i  s   t   TestLogitNewtonc           B   sD   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z RS(   c         C   ss   t  j j j d t  } t  j | j d t | _ t | j | j  j	 d d d d  |  _
 t j } | |  _ d  S(   NR   R   R   R   R   i    (   R   R   R   R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   y  s
    *	c         C   s    t  |  j j |  j j d  d  S(   Ni   (   R   R   t   resid_pearsonR)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_resid_pearson  s    c         C   sY   |  j  j d i d d 6d d 6 } t | j |  j j t  t | j |  j j t  d  S(   Nt   atexogg       @i    g      ?i   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_atexog1R*   Rp   t   margeff_nodummy_atexog1_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_exog1  s
    #		c         C   s_   |  j  j d i d d 6d d 6d d  } t | j |  j j t  t | j |  j j t  d  S(   NRd  g      5@i   i    i   Rz   R{   (	   R   Rm   R   Rn   R)   t   margeff_nodummy_atexog2R*   Rp   t   margeff_nodummy_atexog2_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_nodummy_exog2  s
    )		c         C   s_   |  j  j d i d d 6d d 6d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NRd  g       @i    g      ?i   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_atexog1R*   Rp   t   margeff_dummy_atexog1_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_exog1  s
    )		c         C   se   |  j  j d i d d 6d d 6d d d t  } t | j |  j j t  t | j |  j j	 t  d  S(	   NRd  g      5@i   i    i   Rz   R{   R   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_atexog2R*   Rp   t   margeff_dummy_atexog2_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_dummy_exog2  s    &			(	   R%   R&   R   R   Rc  Rg  Rj  Rm  Rp  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRa  w  s   				t   TestLogitNewtonPrependc           B   sD   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j	 | j  j
 d d d d  |  _ t j } | |  _ t j t j t |  j j   d  |  _ d  S(   NR   R   R   R   R   i    i   (   R   R   R   R   R   R   R   RR   R   R   R   R   R   R  R)   R   t   rollt   arangeR2  R(   t   slice(   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    *		c         C   s    t  |  j j |  j j d  d  S(   Ni   (   R   R   Rb  R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRc    s    c         C   sY   |  j  j d i d d 6d d 6 } t | j |  j j t  t | j |  j j t  d  S(   NRd  g       @i   g      ?i   (	   R   Rm   R   Rn   R)   Re  R*   Rp   Rf  (   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRg    s
    #		c         C   s_   |  j  j d i d d 6d d 6d d  } t | j |  j j t  t | j |  j j t  d  S(   NRd  g      5@i   i    i   Rz   R{   (	   R   Rm   R   Rn   R)   Rh  R*   Rp   Ri  (   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRj    s
    )		c         C   s_   |  j  j d i d d 6d d 6d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NRd  g       @i   g      ?i   R   (
   R   Rm   RR   R   Rn   R)   Rk  R*   Rp   Rl  (   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRm    s
    )		c         C   se   |  j  j d i d d 6d d 6d d d t  } t | j |  j j t  t | j |  j j	 t  d  S(	   NRd  g      5@i   i    i   Rz   R{   R   (
   R   Rm   RR   R   Rn   R)   Rn  R*   Rp   Ro  (   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRp    s    &			(	   R%   R&   R   R   Rc  Rg  Rj  Rm  Rp  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRq    s   					t   TestLogitBFGSc           B   s   e  Z e d     Z RS(   c         C   ss   t  j j j d t  } t  j | j d t | _ t j } | |  _	 t
 | j | j  j d d d d  |  _ d  S(   NR   R   R   R   R   i    (   R   R   R   R   R   R   R   R   R  R)   R   R   R   R   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s
    		(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRu    s   t   TestPoissonNewtonc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z e j	 j
 d d d e  d    Z RS(	   c         C   sm   t  j j j d t  } t  j | j d t } t | j |  j	 d d d d  |  _
 t j } | |  _ d  S(   NR   R   R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   t   poissonR)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s
    '	c         C   sE   |  j  j   } t | j |  j j t  t | j |  j j t  d  S(   N(	   R   Rm   R   Rn   R)   t   margeff_nodummy_overallR*   Rp   t   margeff_nodummy_overall_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_margeff_overall  s
    		c         C   sK   |  j  j d t  } t | j |  j j t  t | j |  j j	 t  d  S(   NR   (
   R   Rm   RR   R   Rn   R)   t   margeff_dummy_overallR*   Rp   t   margeff_dummy_overall_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_margeff_dummy_overall  s
    		c         C   s    t  |  j j |  j j d  d  S(   Ni   (   R   R   t   residR)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_resid  s    c         C   sr   t  j j t  j j t   } t  j j | d d  } t j | d d } |  j j	   d  } t
 | | d  d  S(   Nt   resultss   predict_prob_poisson.csvt	   delimitert   ,id   i   (   t   ost   patht   dirnamet   abspatht   __file__t   joinR   t   loadtxtR   t   predict_probR   (   R#   t   cur_dirR  t	   probs_rest   probs(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predict_prob  s
    RA   s+   res2.cov_params is a zero-dim array of NoneRB   c         C   s   t  t |   j   d  S(   N(   t   superRv  R7   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7     s    (   R%   R&   R   R   Rz  R}  R  R  R   R`   Ra   RR   R7   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRv    s   					t   CheckNegBinMixinc           B   s2   e  Z e j j d  d d e d e  d    Z RS(   RA   s.   pvalues do not match, in some cases wrong sizeRB   R    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R2   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR3   
  s    	(   R%   R&   R   R`   Ra   RR   t   AssertionErrorR3   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   t   TestNegativeBinomialNB2Newtonc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z RS(	   c         C   sp   t  j j j d t  } t  j | j d t } t | j | d  j	 d d d d  |  _
 t j } | |  _ d  S(   NR   R   t   nb2R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   t   negativebinomial_nb2_bfgsR)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s
    *	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RE   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+      s    c         C   sF   |  j  j t |  j  j |  j j t  t |  j  j |  j j t  d  S(   N(   R   RE   R   t   lnalphaR)   R*   t   lnalpha_std_err(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_alpha#  s
    
c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.   *  s    c         C   s$   t  |  j j d  |  j j t  d  S(   Ni(   R   R   R2   R)   t	   DECIMAL_2(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1   .  s    c         C   s(   t  |  j j d  |  j j d  t  d  S(   Ni
   (   R   R   t   fittedvaluesR)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_fittedvalues2  s    c         C   s4   t  |  j j   d  t j |  j j d   t  d  S(   Ni
   (   R   R   RN   R   t   expR)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP   6  s    c         C   s1   t  |  j j d t  d  |  j j d  t  d  S(   NRQ   i
   (   R   R   RN   RR   R)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT   :  s    (   R%   R&   R   R   RF   R+   R  R.   R1   R  RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   								t   TestNegativeBinomialNB1Newtonc           B   s   e  Z e d     Z d   Z d   Z d   Z d   Z e j	 j
 d d d e d e  d	    Z e j	 j
 d d d e d e  d
    Z RS(   c         C   s|   t  j j j d t  } t  j | j d t } t | j | d  } | j	 d d d d d d	  |  _
 t j } | |  _ d  S(
   NR   R   t   nb1R   R   R   id   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   t   negativebinomial_nb1_bfgsR)   (   R   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   A  s    !	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R/   R)   R0   R,  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1   J  s    c         C   sF   |  j  j t |  j  j |  j j d  t |  j  j |  j j t  d  S(   Ni   (   R   RE   R   R  R)   R  R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_lnalphaM  s    
c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   S  s    c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.   V  s    RA   s-   Test has not been implemented for this class.RB   R    c         C   s
   t   d  S(   N(   RC   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP   \  s    c         C   s
   t   d  S(   N(   RC   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT   a  s    (   R%   R&   R   R   R1   R  R+   R.   R   R`   Ra   RR   RC   RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  ?  s   					t   TestNegativeBinomialNB2BFGSc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z RS(	   c         C   sv   t  j j j d t  } t  j | j d t } t | j | d  j	 d d d d d d	  |  _
 t j } | |  _ d  S(
   NR   R   R  R   R   R   i    R   i  (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   i  s    	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RE   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF   t  s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   w  s    c         C   sF   |  j  j t |  j  j |  j j t  t |  j  j |  j j t  d  S(   N(   R   RE   R   R  R)   R*   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  z  s
    
c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s$   t  |  j j d  |  j j t  d  S(   Ni(   R   R   R2   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1     s    c         C   s(   t  |  j j d  |  j j d  t  d  S(   Ni
   (   R   R   R  R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    c         C   s4   t  |  j j   d  t j |  j j d   t  d  S(   Ni
   (   R   R   RN   R   R  R)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s1   t  |  j j d t  d  |  j j d  t  d  S(   NRQ   i
   (   R   R   RN   RR   R)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    (   R%   R&   R   R   RF   R+   R  R.   R1   R  RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  g  s   							t   TestNegativeBinomialNB1BFGSc           B   s   e  Z e d     Z d   Z d   Z d   Z d   Z e j	 j
 d d d e d e  d	    Z e j	 j
 d d d e d e  d
    Z RS(   c         C   sv   t  j j j d t  } t  j | j d t } t | j | d  j	 d d d d d d	  |  _
 t j } | |  _ d  S(
   NR   R   R  R   R   R   id   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R/   R)   R0   R,  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1     s    c         C   sF   |  j  j t |  j  j |  j j d  t |  j  j |  j j t  d  S(   Ni   (   R   RE   R   R  R)   R  R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    
c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R*   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    RA   s-   Test has not been implemented for this class.RB   R    c         C   s
   t   d  S(   N(   RC   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s
   t   d  S(   N(   RC   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    (   R%   R&   R   R   R1   R  R+   R.   R   R`   Ra   RR   RC   RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   
				t!   TestNegativeBinomialGeometricBFGSc           B   sz   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z d	   Z d
   Z d   Z RS(   c         C   sv   t  j j j d t  } t  j | j d t } t | j | d  } | j	 d d d d  |  _
 t j } | |  _ d  S(   NR   R   RQ  R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   t   negativebinomial_geometric_bfgsR)   (   R   R   R   t   modR)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RJ   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RL   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRM     s    c         C   s#   t  |  j j   |  j j t  d  S(   N(   R   R   R-   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s(   t  |  j j d  |  j j d  t  d  S(   Ni
   (   R   R   R  R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    c         C   s4   t  |  j j   d  t j |  j j d   t  d  S(   Ni
   (   R   R   RN   R   R  R)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R(   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   s1   t  |  j j d t  d  |  j j d  t  d  S(   NRQ   i
   (   R   R   RN   RR   R)   R  R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R/   R)   R0   R,  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R8   R)   R,  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR9     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   R<   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR>     s    c         C   s    t  |  j j |  j j t  d  S(   N(   R   R   RE   R)   R=   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    (   R%   R&   R   R   RK   RM   R.   R  RP   R+   RT   R1   R9   R>   RF   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   										t   CheckMNLogitBaseZeroc           B   st   e  Z d    Z d   Z d   Z d   Z d   Z d   Z d   Z d   Z	 e
 j j d d	 d
 e  d    Z RS(   c         C   s   |  j  j   } t | j |  j j d  t | j |  j j d  | j   } | d j	 j
 | j j d d } t | | j d d t | j t j | j  d f  d  S(   Ni   s   dy/dxt   ordert   FR,   gvIh%<=(   R   Rm   R   Rn   R)   t   margeff_dydx_overallRp   t   margeff_dydx_overall_seRr   Rs   RF  Rt   R   R   R   Ru   (   R#   Rv   Rw   Rx   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRz    s    "c         C   sK   |  j  j d d  } t | j |  j j d  t | j |  j j d  d  S(   NRz   R{   i   (   R   Rm   R   Rn   R)   t   margeff_dydx_meanRp   t   margeff_dydx_mean_se(   R#   Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_margeff_mean  s    c         C   s   |  j  } | j  d } t j | j | f  } t j | d t } t | j |  j	 d d d d  } | j
 d t  } t | j |  j j d  t | j |  j j d  | j
 d t d d	  } t | j |  j j d
  t | j |  j j d  d  S(   Nt   voteR   R   R   R   i    R   i   R   i   (   R   R   t   column_stackR   R   R   R   R
   R   R   Rm   RR   R   Rn   R)   t   margeff_dydx_dummy_overallRp   t   margeff_dydx_dummy_overall_set   margeff_eydx_dummy_overallt   margeff_eydx_dummy_overall_se(   R#   R   R  R   R]  Rv   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_margeff_dummy  s    	$		c         C   s    t  |  j j j |  j j  d  S(   N(   R   R   R   R  R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_j  s    c         C   s    t  |  j j j |  j j  d  S(   N(   R   R   R   R  R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_k  s    c         C   s6   t  |  j j d  d   d d d d d d d g  d  S(   Ni   s   y=1s   y=2s   y=3s   y=4s   y=5s   y=6(   R   R   t   _get_endog_namet   None(   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_endog_names  s    c        C   s%  d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d g} t  |  j j   j d  |  t  |  j j   j d  t j |   d d d d	 d	 d
 d g d d d d	 d	 d d
 g d d d d	 d	 d d g d
 d d d	 d	 d d g d d d d	 d	 d d g d d d d	 d	 d d g d d d d	 d	 d d g g } t  |  j j   |  d  S(   Ni   i   i    i   i   g     _@g     D@g       @g        g      (@g      3@g     @S@g     @R@g      @g      .@g     B@g     E@g      @g      "@g      ?g      @g      $@g      4@g      6@g      9@g      ?@g     Q@g      2@g     a@(   R   R   RN   t   argmaxRc   RV   R   t   bincount(   R#   t   pred(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRd   !  sn    ????????????????????????????????????????????B(c         C   s   t  |  j j |  j j  d  S(   N(   R   R   t   resid_misclassifiedR)   R~  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  `  s    RA   s+   res2.cov_params is a zero-dim array of NoneRB   c         C   s   t  t |   j   d  S(   N(   R  R  R7   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7   c  s    (   R%   R&   Rz  R  R  R  R  R  Rd   R  R   R`   Ra   RR   R7   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s   								?	t   TestMNLogitNewtonBaseZeroc           B   s   e  Z e d     Z RS(   c         C   s|   t  j j j d t  } | |  _ | j } t  j | d t } t | j	 |  j
 d d d d  |  _ t j } | |  _ d  S(   NR   R   R   R   R   i    (   R   R   R  R   R   R   R   R   R
   R   R   R   R   t   mnlogit_basezeroR)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   j  s    		'	(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  i  s   t   TestMNLogitLBFGSBaseZeroc           B   s   e  Z e d     Z RS(   c         C   s   t  j j j d t  } | |  _ | j } t  j | d t } t | j	 |  } | j
 d d d d d d d	 d
 d d d d d | j  |  _ t j } | |  _ d  S(   NR   R   R   t   lbfgsR   i    R   iP  R(  i(   t   pgtolg|=t   factrg      @t   loglike_and_score(   R   R   R  R   R   R   R   R   R
   R   R   R  R   R   R  R)   (   R   R   R   t   mymodelR)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   w  s    			(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  v  s   c    
   	   C   s  t  j j t  j j t   }  t  j j |  d d d d d  } t  j j |  } t j t  j j | d  d d d d	 } | d  d   d
 f } | d  d   d  d
  f } | | d k } | | d k } t j	 | d t
 } t | |  } t t | j d d t | _ d
 d l m } d
 d  l } | j d  d k } | r}|   * t j d  | j d t d d  }	 Wd  QXn | j d t d d  }	 t |	 j d  d  S(   Ns   ..t   genmodt   testsR  s   iris.csvR  R  t   skip_headeri   ii   R   R   i  (   t   assert_produces_warningi   t   alwaysR   i2   RV  (   i   i   (   R  R  R  R  R  R  R   t
   genfromtxtR   R   RR   R   R   R   R   R   t   raise_on_perfect_predictiont   pandas.util.testingR  t   syst   version_infot   warningst   simplefilterR   R   (
   R  t   iris_dirt   irist   yt   XR  R  R  t   PY3_g3R]  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_perfect_prediction  s,    !!		
!c          C   s  t  j j j d t  }  t  j |  j d t } t  j |  j	 |  j
 d d d d  } | j   } | j |  } t | |  | j | d d d d	 } t | |  | j | d d d d
 } t d
 | |  | j | d t j d
  d d	 } t d
 | |  d  S(   NR   R   R   R   R   i    t   offsett   exposurei   i   (   R   R   RC  R   R   R   R   RR   R   R   R   RN   R   R   t   log(   R   R   R]  t   pred1t   pred2t   pred3(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_poisson_predict  s    '$c          C   s1  d }  t  j j d  t  j j |  d  } t j | d t } t  j j t  j | j	 d    } t j
 | |  } d d l m } d d  l } | j d  d k } | r |   : t j d	  | j d
 t  j d  d d d d  } Wd  QXn( | j d
 t  j d  d d d d  } t | j d  d  S(   Ni'  i) i   R   i   i(   R  i   R  R   i   R   R   R   i    RV  (   i   i   (   R   R   R   t   randnR   R   RR   Rw  R  RV   R   R  R  R  R  R  R  R   R  R   R   (   t   nobst   xt   y_countR  R  R  R  R]  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_poisson_newton  s    $
1(c          C   s   t  j j j d t  }  |  j } | d  d   d  d  f } t  j | d t } t  j |  j	 |  j
 d d d d  } d j | j   j   j d  d	  } t j j t j j t   } t j j | d
 d  } t | d   } | j   } Wd  QXt j j | | d   | j   d  S(   NR   iR   R   R   R   i    s   
i	   R  s   mn_logit_summary.txtt   r(   R   R   R  R   R   R   R   RR   R
   R   R   R  t   summaryt   as_textt   splitR  R  R  R  R  t   opent   readR   t   testingR   t   summary2(   R   R   R   t   smryR  t   test_case_filet   fdt	   test_case(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_issue_339  s    	'(c          C   s   t  j j j d t  }  |  j } | d  d   d  d  f } t  j | d t } t  j |  j	 |  j
 d d d d  } | d } t j j | j |  j d
  t j j | j | d   j d  d  S(   NR   iR   R   R   R   i    i   i   (   i   i   (   i   i   (   R   R   R  R   R   R   R   RR   R
   R   R   R   R  R   RN   Rt   R  (   R   R   R   R  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_issue_341  s    	'
c          C   s   t  j j d
  }  t  j j d d d d |  d  d   d f <t  j j d d d d |  d  d   d f <t  j j d d d d |  d  d   d f <t  j j d	 d d d |  d  d   d f <t |   } t | d d g  d  S(   Ni2   i
   i   Ru   i   i   i    i   i(   i2   i
   (   R   R   t   randintR   R   (   R  t	   count_ind(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_iscount  s    ++++c          C   s   t  j j d
  }  t  j j d d d d |  d  d   d f <t  j j d d d d |  d  d   d f <t  j j d d d d |  d  d   d f <t  j j d	 d d d |  d  d   d f <t |   } t | d d g  d  S(   Ni2   i
   i   Ru   i   i    i   i   i(   i2   i
   (   R   R   R  R   R   (   R  R  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_isdummy  s    ++++c          C   sJ   d d d d d d g }  t  j j d d  } t  j j t t |  |  d  S(   Ni   i   i   (   R   R   R  R  R   R!   R   (   R  R  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_non_binary  s    c    
      C   s]  t  j j j   }  |  j j t t t d  d    |  d <t  j	 |  j
 d t } t  j |  j |  } | j d d  } | j } | j   } | j | j d  d  d  d   f  } t j d d	 j |  j
 j  |  j  } | j d d  } | j } | j   } t | | d
 d | j |  j
 j d  d  d  d   f  }	 t |	 | d
 d d  S(   Ni   t   ABCDEFGR   R   R   i    i   s   PID ~ s    + R,   g|=(   R   R   R  t   load_pandasR   t   replacet   dictt   zipR    R   R   RR   R
   R   R(   R  RN   t   iloct   smfR  R  t   columnsR   R   (
   t   dtaR   R  R]  R(   R  t	   predictedR)   t   params_ft   predicted_f(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_mnlogit_factor  s    +	((	+c          C   s  i d d d d g d 6d d d t  j g d 6d g d d	 6t  j j d
 d  d 6d d d d g d 6}  t j |   } t j d d | d | d } t t	 | j
  t  j k d d t j t  j j d
 d   } d | j d <t t t j | j | d	 d g d | d  S(   Ni   i   i
   i   t   Fooi   t   Bari   t   constantRu   R  g      ?R  s	   Foo ~ BarR   t   msgs   Exposure is not ndarrayi   (   i   R  (   R   R5  R   t   uniformt   pdt	   DataFrameR  Rw  R   t   typeR  t   ndarrayt   Seriest   locR   R!   R   R   R  (   t   dt   dft   mod1R  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_formula_missing_exposure  s    ,$""c    
      C   sD  d d  l  }  i d d d d g d 6d d d d g d	 6d g d d
 6t j d  g d d 6d d d d g d 6} |  j |  } t j d d | d | d } t j d d g  } | j | d t } | d
 d	 g j	 } t j
 | |  d } t | |  | j | d t j d  g d d t } | d }	 t | |	  d  S(   Nii   i   i
   i   R  i   i   R  R  R  g      ?R  s	   Foo ~ BarR   g?RQ   (   t   pandasR   R  R	  R   t   from_formulaR  RN   RR   Rs   t   dotR   (
   R  R  R  R  R(   R  R  t   expectedR  t	   expected2(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predict_with_exposure.  s    )%+
c          C   s   d }  t  j |   } d | d d g <t | t  j |    j d d  } t  j d d g d d g g  } t | j   |  t | t  j |    j d d  } t  j d d g d d g g  } t | j   |  d  S(	   Ni
   i   i   R   i    g       @g        g       @(	   R   t   zerosR   R  R   R  R   Rc   R
   (   R  R  R]  R  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_binary_pred_table_zerosL  s    $!$!t   TestGeneralizedPoisson_p2c           B   sq   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z d	   Z d
   Z RS(   c         C   s   t  j j j d t  } t  j | j d t | _ t | j | j d d } | j	 d d d d  |  _
 t j } | |  _ d  S(	   NR   R   t   pi   R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   t   generalizedpoisson_gp2R)   (   R   R   R  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   ^  s    	c         C   s#   t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   RE   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF   g  s    c         C   s#   t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   j  s    c         C   s<   t  |  j j |  j j  t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  m  s    c         C   s&   t  |  j j   |  j j d d d  S(   NR   gMbP?(   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.   r  s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RJ   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK   v  s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RL   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRM   y  s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RG   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR-  |  s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   R8   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR9     s    c         C   sa   |  j  j t j t |  j  j   d   } t | j |  j j	  t | j
 |  j j d d d  S(   NiR   gV瞯<(   R   t	   wald_testR   R1  R2  R(   R   t	   statisticR)   t   wald_statisticR=  t   wald_pvalue(   R#   t   result(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt	   test_wald  s    +c         C   sD   t  j |  j j j  } |  j j |  } t |  j j | j  d  S(   N(	   R   t   identityR   R(   Ru   R0  R   R/   R6  (   R#   t   unit_matrixR0  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_t  s    (   R%   R&   R   R   RF   R+   R  R.   RK   RM   R-  R9   R"  R%  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  [  s   										t"   TestGeneralizedPoisson_transparamsc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z RS(	   c         C   sy   t  j j j d t  } t  j | j d t | _ t | j | j d d j	 d d d d  |  _
 t j } | |  _ d  S(	   NR   R   R  i   R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	c         C   s#   t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   RE   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    c         C   s#   t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   s<   t  |  j j |  j j  t  |  j j |  j j d d d  S(   NR   gh㈵>(   R   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    c         C   s&   t  |  j j   |  j j d d d  S(   NR   gMbP?(   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RJ   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK     s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RL   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRM     s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   RG   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR-    s    c         C   s   t  |  j j |  j j  d  S(   N(   R   R   R8   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR9     s    (   R%   R&   R   R   RF   R+   R  R.   RK   RM   R-  R9   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR&    s   								t   TestGeneralizedPoisson_p1c           B   sM   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 RS(   c         C   sv   t  j j j d t  |  _ t  j |  j j d t |  j _ t |  j j	 |  j j d d j
 d d d d  |  _ d  S(	   NR   R   R  i   R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   (   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    !c         C   s   t  j |  j j |  j j  j |  j j d   } t  j |  j j |  j j d d j t	 |  j j d   d g  } t
 | |  d  S(   NiR  i   i    (   R   R   R   R   R   t   loglikeR   R(   R   t   listR   (   R#   t   poisson_llft   genpoisson_llf(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR9     s     c         C   s   t  j |  j j |  j j  j |  j j d   } t  j |  j j |  j j d d j t	 |  j j d   d g  } t
 | d  | d d d  S(   NiR  i   i    R   g&.>(   R   R   R   R   R   R[   R   R(   R   R)  R   (   R#   t   poisson_scoret   genpoisson_score(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_score  s     c         C   s   t  j |  j j |  j j  j |  j j d   } t  j |  j j |  j j d d j t	 |  j j d   d g  } t
 | d  d  d  d  f | d d d  S(   NiR  i   i    R   g|=(   R   R   R   R   R   t   hessianR   R(   R   R)  R   (   R#   R,  R-  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_hessian  s     c         C   sD   t  j |  j j j  } |  j j |  } t |  j j | j  d  S(   N(	   R   R#  R   R(   Ru   R0  R   R/   R6  (   R#   R$  R0  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR%    s    c         C   s<  |  j  j } t j t |  j  j   } d | d )| j d | d d d  } | j d | d d d  } | j d | d d d  } t | j |  j  j d d	 t | j |  j  j d d
 t |  j  j d  d j	   d  t | j d  d j	   d  t | j d  d j	   d  t | j d  d j	   d  d  S(   Ni    iR  g{Gz?R   id   i  R   g-C6
?gh㈵>i   g.?gI?g'~ۅ?gM5rK7?(
   R   R   R   R  R2  R(   R"   R   RE   R{   (   R#   R   R  t   res_reg1t   res_reg2t   res_reg3(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_fit_regularized  s    
!c         C   s7   |  j  j j   } t d | k  t | d d  d  S(   NR  i   (   R   R   t   _get_init_kwdsR   R   (   R#   t   kwds(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_init_kwds  s    (
   R%   R&   R   R   R9   R.  R0  R%  R4  R7  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR'    s   								t&   TestGeneralizedPoisson_underdispersionc           B   s;   e  Z e d     Z d   Z d   Z d   Z d   Z RS(   c         C   s   d d d g |  _  t j j d  d } t j | d f  } d | d  | d  d f <t j | j |  j  d    } t j j	 j
 | |  j  d d d t |  |  _ t j |  j | d	 d } | j d
 d d d d d d d d d  |  _ d  S(   Ni   g      gi  i   i   iRu   R  R   R   t   xtolgư>R   i  t   maxfunR   i    (   t   expected_paramsR   R   R   R  R  R  R   t   distributionst   genpoisson_pt   rvsR2  R   R   R   R]  (   R   R  R   t   mu_truet   model_gp(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    "c         C   s   |  j  } | j j } t | j   d d d t | j   d d d t | j |  j d d d d t | j	 d t
 k  t | j	 d	 d
 d d d  S(   NgQ?R,   gMbP?gaTR'?R   gQ?g?RV  t   foptg6?g{Gz?(   R]  R   R   R   R{   t   varR(   R;  R   R   RR   (   R#   R]  R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt
   test_basic	  s    	c         C   s   |  j  } | j j d | j d d d d  } t | j j | j  t j t | j   d d t | j j | j  t j t | j   d d t | j | j d d	 d  S(
   NR   R   R   R   i    R   g{Gz?g|=g-C6?(	   R]  R   R   R(   R   R[   R   R  R2  (   R#   R]  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_newton  s    	$c         C   sv   t  |  j j   j   |  j j   d d d d t  |  j j   j   |  j j j   |  j j   d d d d d  S(   NR   g?R,   g?(   R   R]  RN   R{   R   t   _dispersion_factorRB  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_mean_var  s
    $"c         C   s   |  j  } | j j } t j | j t   } | j d d  } t j	 j
 j t j d  d  d   d  f | j   | j d d  j } t | | d d d d d d	 l m } | j | | j d
   } t | d d d d  S(   Nt   whicht   probi   ii   R,   g|=R   (   t   statsi    g2Ld?gn?g{Gz?(   g2Ld?gn?(   R]  R   R   R   R  t   astypet   intRN   R   R<  R=  t   pmfRs  R  R(   t   TR   t   scipyRI  t	   chisquareRV   (   R#   R]  R   t   freqt   prt   pr2RI  t   chi2(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  '  s    	((   R%   R&   R   R   RC  RD  RF  R  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR8    s
   		
	t   TestNegativeBinomialPNB2Newtonc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z RS(	   c         C   sy   t  j j j d t  } t  j | j d t } t | j | d d } | j	 d d d d  |  _
 t j } | |  _ d  S(	   NR   R   R  i   R   R   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   9  s    	c         C   s)   t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   RE   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF   C  s    c         C   s#   t  |  j j |  j j d d d  S(   NR   gHz>(   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   G  s    c         C   sF   |  j  j t |  j  j |  j j  t |  j  j |  j j d d d  S(   NR   gHz>(   R   RE   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  K  s
    
c         C   s,   t  |  j j   |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.   R  s    c         C   s-   t  |  j j d  |  j j d d d d d  S(   NiR   g{Gzt?R,   (   R   R   R2   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1   V  s    c         C   s%   t  |  j j d  |  j j d   d  S(   Ni
   (   R   R   R  R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  Z  s    c         C   s1   t  |  j j   d  t j |  j j d    d  S(   Ni
   (   R   R   RN   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP   ^  s    c         C   s.   t  |  j j d d  d  |  j j d   d  S(   NRG  RQ   i
   (   R   R   RN   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT   b  s    (   R%   R&   R   R   RF   R+   R  R.   R1   R  RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT  7  s   
							t   TestNegativeBinomialPNB1Newtonc           B   sM   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 RS(   c         C   s   t  j j j d t  } t  j | j d t } t | j | d d } | j	 d d d d d	 d
  |  _
 t j } | |  _ d  S(   NR   R   R  i   R   R   R   id   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R  R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   i  s    !	c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g{Gzt?R,   (   R   R   R/   R)   R0   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1   r  s    c         C   s@   |  j  j t |  j  j |  j j  t |  j  j |  j j  d  S(   N(   R   RE   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR  v  s    
c         C   s   t  |  j j |  j j  d  S(   N(   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+   |  s    c         C   s,   t  |  j j   |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s=   t  |  j j   d  t j |  j j d   d d d d d  S(   Ni
   R   gMbP?R,   (   R   R   RN   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s:   t  |  j j d d  d  |  j j d  d d d d d  S(   NRG  RQ   i
   R   gMbP?R,   (   R   R   RN   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    (
   R%   R&   R   R   R1   R  R+   R.   RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRU  g  s   						t   TestNegativeBinomialPNB2BFGSc           B   s_   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z RS(	   c         C   sy   t  j j j d t  } t  j | j d t } t | j | d d j	 d d d d d	 d
  |  _
 t j } | |  _ d  S(   NR   R   R  i   R   R   R   i    R   i  (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    	c         C   s)   t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   RE   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   sX   |  j  j t |  j  j |  j j d d d d t |  j  j |  j j d d d d d  S(   NR   gh㈵>R,   (   R   RE   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    
c         C   s,   t  |  j j   |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s-   t  |  j j d  |  j j d d d d d  S(   NiR   g{Gzt?R,   (   R   R   R2   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1     s    c         C   s1   t  |  j j d  |  j j d  d d d d d  S(   Ni
   R   g-C6?R,   (   R   R   R  R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s    c         C   s=   t  |  j j   d  t j |  j j d   d d d d d  S(   Ni
   R   gMbP?R,   (   R   R   RN   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s:   t  |  j j d d  d  |  j j d  d d d d d  S(   NRG  RQ   i
   R   gMbP?R,   (   R   R   RN   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    (   R%   R&   R   R   RF   R+   R  R.   R1   R  RP   RT   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRV    s   							t   TestNegativeBinomialPNB1BFGSc           B   s   e  Z e d     Z d   Z d   Z d   Z d   Z d   Z d   Z	 d   Z
 d   Z d	   Z d
   Z d   Z d   Z RS(   c         C   sy   t  j j j d t  } t  j | j d t } t | j | d d j	 d d d d d	 d
  |  _
 t j } | |  _ d  S(   NR   R   R  i   R   R   R   id   R   i    (   R   R   RC  R   R   R   R   R   R   R   R   R   R  R)   (   R   R   R   R)   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    !	c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g{Gzt?R,   (   R   R   RE   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRF     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g      ?R,   (   R   R   RJ   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRK     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g      ?R,   (   R   R   RL   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRM     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R8   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR9     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R8   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR>     s    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g      ?R,   (   R   R   R/   R)   R0   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR1     s    c         C   sN   t  |  j j |  j j d d d d t  |  j j |  j j d d d d d  S(   NR   gMbP?R,   (   R   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR    s
    c         C   s)   t  |  j j |  j j d d d d d  S(   NR   g?R,   (   R   R   R(   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR+     s    c         C   s,   t  |  j j   |  j j d d d d d  S(   NR   g?R,   (   R   R   R-   R)   (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR.     s    c         C   s=   t  |  j j   d  t j |  j j d   d d d d d  S(   Ni
   R   g{Gzt?R,   (   R   R   RN   R   R  R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRP     s    c         C   s:   t  |  j j d d  d  |  j j d  d d d d d  S(   NRG  RQ   i
   R   g{Gzt?R,   (   R   R   RN   R)   R  (   R#   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRT     s    c         C   s7   |  j  j j   } t d | k  t | d d  d  S(   NR  i   (   R   R   R5  R   R   (   R#   R6  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR7    s    (   R%   R&   R   R   RF   RK   RM   R9   R>   R1   R  R+   R.   RP   RT   R7  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRW    s   
											t$   TestNegativeBinomialPL1Compatabilityc           B   s   e  Z e d     Z RS(   c         C   sh  d |  _  d |  _ t j j j d t  } | j j t	  j
 t | j  d  } | | j d  | j d  } t j | d t } | d  d   d  |  j  f } t j | j |  } | j d d d	 d  |  _ d t | j  t j |  j  d
  } d | |  j *d | d <t j | j |  } | j d d d | d	 t d d d d d d  |  _ d
 |  _ d  S(   Ni
   i   R   ii    R   R   R   R   i   R  R  R  g|=R   i  R	  R
  (   R+  R(  R   R   RC  R   R   R   RD  RE  RF  R2  R{   RL  R   RR   R   R   R   R)  R   R  R"   R*  R.  (   R   RG  RH  RM  RI  RJ  R  RN  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s"    		' '
	(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRX  
  s   t    TestNegativeBinomialPPredictProbc           B   s   e  Z d    Z d   Z RS(   c         C   s  d d g } t  j j d  d } t  j | d f  } d | d  | d  d f <t  j | j |   } d } d | | } | | | } t j | | d t |  } t	 j
 | |  j d	 d
  }	 |	 j   }
 d | |
 } | | |
 } |	 j d d  } t | t j t  j d  d  d   d  f | |  j d d d d |	 j d | d
 d g d d  } t | | d
 d g d d d d d  S(   Ni   g      i  i   i   g?g      ?Ru   R   i    RG  RH  i   R   g{Gz?R,   R   ig|=gV瞯<(   R   R   R   R  R  R  R   R>  R2  R   R   R   RN   R   RL  Rs  R  RM  (   R#   R;  R  R   R?  R  Ru   RH  R   R]  t   muR  t   probs_ex(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predict_prob_p1%  s(    1"c         C   sA  d d g } t  j j d  d } t  j | d f  } d | d  | d  d f <t  j | j |   } d } d | } | | | } t j | | d t |  } t	 j
 | | d	 d j d
 d  }	 |	 j   }
 d | } | | |
 } t |	 j d d  t j t  j d  d  d   d  f | |  j d d d d d  S(   Ni   g      i  i   i   g?g      ?Ru   R  R   i    RG  RH  i   R   g{Gz?R,   (   R   R   R   R  R  R  R   R>  R2  R   R   R   RN   R   RL  Rs  R  RM  (   R#   R;  R  R   R?  R  Ru   RH  R   R]  RZ  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_predict_prob_p2?  s"    
$
1(   R%   R&   R\  R]  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRY  #  s   	t	   CheckNullc           B   s    e  Z e d     Z d   Z RS(   c         C   s|   t  j d d d d d d d g  } t |  } t  j t  j |  | f  } t  j d d	 d
 d d d d g  } | | f S(   Ng      4@g      9@g      >@g     A@g      D@g     F@g      I@i  i  i  i  i@  i  i  (   R   R  R2  R  R  (   R   R  R  R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt	   _get_dataX  s
    $$c         C   s   |  j  j d |  j d d  } t | j _ | j } | j } t | | j	 d d |  j } t | | j	 d d t | j
 | j
 d d d  S(   NR   R   i    R,   gư>g-C6
?(   R   R   R   RR   t   _resultst   _attach_nullmodelR:   t   res_nullR   R8   R(   (   R#   R]  t   llf0t	   res_null0t	   res_null1(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR;   `  s    			(   R%   R&   R   R_  R;   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR^  V  s   t   TestPoissonNullc           B   s   e  Z e d     Z RS(   c         C   sn   |  j    \ } } t | |  |  _ t | | d  d   d f  j d d g d d  |  _ d d g |  _ d  S(   Ni    R   g      !@R   (   R_  R   R   R   Rb  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   p  s    7(   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRf  n  s   t   TestNegativeBinomialNB1Nullc           B   s   e  Z e d     Z RS(   c         C   s   |  j    \ } } t | | d d |  _ t | | d  d   d f d d |  _ |  j j d d d g d d d	 d
 d d d d  |  _ t j d d d g  |  _ d  S(   NRP  R  i    R   i   i  R   R   R   g:0yE>R   i,  R   g0@g?$?g     @(	   R_  R   R   t
   model_nullR   Rb  R   R  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR   {  s    (   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRg  y  s   t   TestNegativeBinomialNB2Nullc           B   s   e  Z e d     Z RS(   c         C   s   |  j    \ } } t | | d d |  _ t | | d  d   d f d d |  _ |  j j d d d g d d d	 d
 d d d d  |  _ t j d d d g  |  _ d  S(   NRP  R  i    R   i   g      ?R   R   R   gư>R   i,  R   g#$ @g7<$BD?gki,?(	   R_  R   R   Rh  R   Rb  R   R  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    (   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRi    s   t   TestNegativeBinomialNBP2Nullc           B   s    e  Z e d     Z d   Z RS(   c         C   s   |  j    \ } } t | | d d |  _ t | | d  d   d f d d |  _ |  j j d d d g d d d	 d
 d d d d  |  _ t j d d d g  |  _ d  S(   NR  i   i    R   i   i   R   R   R   gư>R   i,  R   g#$ @g7<$BD?gki,?(	   R_  R   R   Rh  R   Rb  R   R  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    (c         C   s`   |  j  j |  j  j } } t | | d d } | j   } |  j  j   } t | | d d d  S(   NRP  R  R,   g-q=(   R   R   R   R   t   _get_start_params_nullR   (   R#   R   R   t	   model_nb2t   sp1t   sp0(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_start_null  s
    (   R%   R&   R   R   Ro  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRj    s   
t   TestNegativeBinomialNBP1Nullc           B   s    e  Z e d     Z d   Z RS(   c         C   s   |  j    \ } } t | | d d |  _ t | | d  d   d f d d |  _ |  j j d d d g d d d	 d
 d d d d  |  _ t j d d d g  |  _ d  S(   NR  g      ?i    i   R   i   R   R   R   gư>R   i,  R   g0@g?$?g     @(	   R_  R   R   Rh  R   Rb  R   R  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    (c         C   s`   |  j  j |  j  j } } t | | d d } | j   } |  j  j   } t | | d d d  S(   NRP  R  R,   g-q=(   R   R   R   R   Rk  R   (   R#   R   R   Rl  Rm  Rn  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRo    s
    (   R%   R&   R   R   Ro  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRp    s   
t   TestGeneralizedPoissonNullc           B   s   e  Z e d     Z RS(   c         C   s   |  j    \ } } t | | d d |  _ t | | d  d   d f d d |  _ |  j j d d d g d d d	 d
 d d d d  |  _ t j d d d g  |  _ d  S(   NR  g      ?i    R   g @i   R   R   R   g:0yE>R   i,  R   gY$@g^?g3 7I?(	   R_  R   R   Rh  R   Rb  R   R  R   (   R   R   R   (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyR     s    ((   R%   R&   R   R   (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyRq    s   c          C   s  d }  t  j d  } d | d  |  d  d f <t  j | j d   } t  j j |  } t | |  j d t  j d d g  d d  } | j	 } t
 t | d  t k  | j d	 t  | j	 } t | j j d t  j | j    d
 d t | j j d d  t | |  | j d	 t d d g d d  | j	 } t | j j d d g d
 d t | j j d d  | j   t
 d | j k  t | j d |  t
 d | j k  t | j d |  | j d d  t
 d | j k  t | j d d  d  S(   Ni
   i   i   i    i   R   R   t
   res_llnullt   attach_resultsR,   g|=t	   optimizerR   g      ?R   R   t	   prsquaredR:   i  (   i   i   (   R   R  R  RV   R   Rw  R   R   R  R:   R   R5   R   t   set_null_optionsRR   R   Rb  t   mle_settingsR{   R   R  t   _cache(   R  R   RZ  R   R]  t   llnull0t   lln(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_null_options  s6    0			
c          C   s  t  j j t  j j t   }  t  j j |  d d  } t j |  } d g } | | | | j   | | j	   } | d j
 } t j | | j    } d  } t d d d d  } t | | d	 | }	 |	 j d
 d |  }
 t | | d	 | }	 |	 j d
 d d |  } t | j d d  |
 j d d t | j d |  t | j   j   | j   d d t d d d d  } t | | d	 | d d }	 |	 j d
 d d |  } t | j d d  |
 j d d t | j d |  t | j   j   | j   d d t | | d	 | d d }	 |	 j d
 d d d d |  } t | j d d  |
 j d d t | j d |  t | j   j   | j   d d d  S(   NR  s
   sm3533.csvt   ppt   numR   R   R   i  R  R   i    t   optim_kwds_prelimR   iR,   g-C6?g?R  i   g      ?i   (   R  R  R  R  R  R  R  t   read_csvR{   RL  Rs   R   R   t   copyR  R  R   R   R   R   Rw  R(   R   RN   R   R   (   R  t   filepathR  t   featuresR  R  R   R  R~  R   t   res_poiR]  (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_optim_kwds_prelim  s<    	&%%c          C   sZ  d d  l  }  t j d    t j j j d t  } Wd  QX|  j	 d  t j
 | j | j d d } t j d d d d	 d
 d d d d d g
  } | j d | d d  } t | j d  t j d d d d d d d d d d g
  } | j d d d | d d  } t | j d k  | j d | d d  } t | j | j  t | j | j  d  S(   NiR   t   errorRP  R  gXgBP'˿g)my|?gV6㢞gqmP?g˯?ghh?g,y?g˨?gI"$?R   R   i    i   g|~!<ڨgS9ÿg1&7i?gmqdgo ?gT
}?g$&9?g        g,[?R  id   (   R  R   t   warnsR  R   R   RC  R   R   R  R   R   R   R   R  R   R   RG   R"   R   RH   (   R  R   R   R(   R   t
   reg_paramsR)   t   res3(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt"   test_unchanging_degrees_of_freedom#	  s"    c          C   s   t  j i d d d d d g d 6d d d d d g d 6 }  t j d d d	 |   j   } t j t d
 d  | j   j	   } Wd  QXd | k s t
  d  S(   Ng        g?i    t   Ai   t   Bt   formulas   A ~ BR   R   s+   endog contains values are that not int-likes   A=1.1(   R  R	  R  R  R   R   R  R   R  R  R  (   R  R!  t   summ(    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   test_mnlogit_float_name?	  s    ;
(   R_   t   statsmodels.compat.pythonR    R  R  t   numpyR   R  R  t   numpy.testingR   R   R   R   R   R   R   R   t#   statsmodels.discrete.discrete_modelR   R	   R
   R   R   R   R   R   t%   statsmodels.discrete.discrete_marginsR   R   t   statsmodels.apit   apiR   t   statsmodels.formula.apiR  R  t   results.results_discreteR   R   R   R   t   statsmodels.tools.sm_exceptionsR   R   t   scipy.statsR   t   cvxoptRR   R  t   ImportErrorR   RW   t
   DECIMAL_10R\   R*   R=   R  R,  t	   DECIMAL_0t   objectR   R'   Rb   Rl   R   R   R   R   R   R   R   R   R   R   R  R  R  R  R`   t   skipifR  R!  R'  RB  RK  RO  RR  RS  RT  RU  RX  R\  R_  R`  Ra  Rq  Ru  Rv  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R  R&  R'  R8  RT  RU  RV  RW  RX  RY  R^  Rf  Rg  Ri  Rj  Rp  Rq  R{  R  R  R  (    (    (    sG   lib/python2.7/site-packages/statsmodels/discrete/tests/test_discrete.pyt   <module>	   s   4:"

N7 ,/*-(/)7s						
	
					2(D>0)5E3	&	2	