p7]c@ sddlmZddlmZddlZddlmZmZm Z m Z ddl Z ddl j ZddlmZddlmZdefd YZd efd YZd efd YZdefdYZdefdYZdefdYZe jjdefdYZdefdYZdefdYZdefdYZdefdYZ dS(i(tdivision(t SP_GTE_019N(tassert_t assert_equaltassert_array_equaltassert_allclosei(tRandHIE(tCheckModelMixint CheckGenericcB sweZdZdZdZdZdZdZdZdZ dZ d Z e j jd ZRS( cC s)t|jj|jjdddddS(Ntatolgh㈵>trtol(Rtres1tparamstres2(tself((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_paramsscC s)t|jj|jjdddddS(NR gh㈵>R (RR tllfR (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_llfscC s,t|jj|jjdddddS(NR gMbP?R gh㈵>(RR tconf_intR (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_conf_intscC s)t|jj|jjdddddS(NR gMbP?R (RR tbseR (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_bsescC s)t|jj|jjdddddS(NR g{Gz?R (RR taicR (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_aicscC s)t|jj|jjdddddS(NR g?R (RR RR (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_bicscC sDtj|jjj}|jj|}t|jj|jdS(N( tnptidentityR R tsizett_testRttvaluesttvalue(Rt unit_matrixR((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_t!scC s|jj}tjt|jj}d|d)|jd|ddddd}t|jd|jjdd d d d dS( Niitalphag{Gz?tdisptmaxiteriiR g?R (R tmodelRtonestlenR tfit_regularizedR(RR$R!tres_reg((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_fit_regularized&s   "cC si|jjj}tt|jt|jx.|jjD]\}}t|||qDWdS(N( R R$t_get_init_kwdsRtsettkeyst init_keyst init_kwdstitems(RR.tkeytvalue((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_init_keys0s"cC s|jj|jjjj}|jjjj}t|jt|jjjdft t j |dt t j |ddS(Nii( R tllnulltres_nullR$texogt exog_inflRtshapeR&RRtptp(Rt exog_nulltexog_infl_null((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_null6s  cC s+|jj}dt|ks'tdS(NsCovariance Type:(R tsummarytstrtAssertionError(Rtsumm((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_summaryAs(t__name__t __module__RRRRRRR R)R2R;tpytesttmarktsmokeR@(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRs         tTestZeroInflatedModel_logitcB seZedZRS(cC stjjjdt}|j|_tj|jddddfdt}tj|jdddfdt}tj|j|d|ddj d d d d d d|_ t |j j _ ddddg|_idd6|_tj}||_dS(Nt as_pandasiitprependiR6t inflationtlogittmethodtnewtonR#iR"texposuretoffset(tsmtdatasetstrandhietloadtFalsetendogt add_constantR5tZeroInflatedPoissontfitR tTruet_resultst_attach_nullmodelR-R.Rtzero_inflated_poisson_logitR (tclstdataR5R6R ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt setup_classIs .(  (RARBt classmethodR^(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRFHstTestZeroInflatedModel_probitcB seZedZRS(cC stjjjdt}|j|_tj|jddddfdt}tj|jdddfdt}tj|j|d|ddj d d d d d d|_ t |j j _ ddddg|_idd6|_tj}||_dS(NRGiiRHiR6RItprobitRKRLR#iR"RMRN(RORPRQRRRSRTRUR5RVRWR RXRYRZR-R.Rtzero_inflated_poisson_probitR (R\R]R5R6R ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^Zs .(  (RARBR_R^(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR`YstTestZeroInflatedModel_offsetcB s eZedZdZRS(c C stjjjdt}|j|_tj|jddddfdt}tj|jdddfdt}tj|j|d|d|jdddfj d d d d d d|_ t |j j _ ddddg|_idd6|_tj}||_dS(NRGiiRHiR6RNiRKRLR#iR"RMRIRJ(RORPRQRRRSRTRUR5RVRWR RXRYRZR-R.Rtzero_inflated_poisson_offsetR (R\R]R5R6R ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^ks .(+  c C s4|jj}|j}tj|j|jd|jdtj |}|j d|jj dddddd }t |j |jj d d d d |jj }|jj }t ||d d d d |j}|j}|j}|jj d |d|d|} |j d |d|dtj |} t | | d d d d |jddd}|jddd}|ddd}|jj d |d|d|} |j d |d|dtj |} t | | d d d d t | |dddd d d d t | |dddd d d d dS(NR6RMt start_paramsRKRLR#iR"iR gư>R R5RNi i(R R$RNRORVRTR5R6RtexpRWR Rtpredict( Rtmodel1RNtmodel3tres3tfitted1tfitted3textex_inflt fitted1_0t fitted3_0t fitted1_2t fitted3_2((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_exposure|s8  "     &(RARBR_R^Rs(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRcjstTestZeroInflatedModelPandascB s eZedZdZRS(c C s6tjjj}|j|_||_tj|jjddddfdt }tj|jjdddfdt }t j ddddd d g}tj |j|d |d d }|j d|dddddd|_t|jj_d dd dg|_id d 6|_tj}||_dS(NiiRHig̥w?gNr(Xg. g?gA~Wpg`·?R6RIRJReRKRLR#iR"RMRN(RORPRQt load_pandasRTR]RUR5tilocRSRtasarrayRVRWR RXRYRZR-R.RR[R (R\R]R5R6ReR$R ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^s"  1+  cC sddddddg}t|jjj|t|jjjj|t|jjjj|tj |j j j ddddfd t }tj |j j j ddd fd t }ddddddg}tj|j j|d |d d }t|j|dS(Ntinflate_lncoinst inflate_consttidptlpitfmdetconstiiRHiR6RIRJ(RR R$t exog_namesR tindexttolistRRORUR]R5RvRXRVRT(Rt param_namesR5R6R$((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_namess  4.  (RARBR_R^R(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRtstTestZeroInflatedPoisson_predictcB s2eZedZdZdZdZRS(c C sddg}tjjdd}tj|df}d|d|ddf<|j|}tjjj|dd|j |_ tj |j |}|j dd d d d d d d|_ dS(Nig?i{iig?RRKtbfgsR#itmaxfunR"i(RtrandomtseedR%tdotROt distributionst zipoissontrvsR7RTRVRWtres(R\texpected_paramstnobsR5tmu_trueR$((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^s cC s5t|jjj|jjdddddS(NR g{Gz?R (RRRgtmeanRT(R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_means$cC sEt|jjj|jjj|jjdddddS(NR g?R (RRRgRt_dispersion_factorRTtvar(R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_varscC s|j}|jj}|jdd}tjjjtj ddddf|jdj }t ||dddddS(Ntwhichtprobig?R R ( RR$RTRgRORRtpmfRtarangetNonetTR(RRRTtprtpr2((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_predict_probs   ((RARBR_R^RRR(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRs  t"TestZeroInflatedGeneralizedPoissoncB sDeZedZdZdZdZdZdZRS(cC s tjjjdt}|j|_tj|jddddfdt}tj|jdddfdt}tj|j|d|ddj dd d d d d|_ t |j j _ dd dddg|_idd6dd6|_tj}||_dS(NRGiiRHiR6tpRKRLR#iR"RMRIRNRJ(RORPRQRRRSRTRUR5tZeroInflatedGeneralizedPoissonRWR RXRYRZR-R.Rt!zero_inflated_generalized_poissonR (R\R]R5R6R ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^s .(* cC sdS(N((R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRscC sdS(N((R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRscC sdS(N((R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRscC sDtj|jjj}|jj|}t|jj|jdS(N( RRR R RRRRR(RRR((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR sc C s|jj}|jjd}|jd|dddddddd }t|j|jjd d d d t|j|jjd d d dt|j dt k|jd|dddddddd }t|j|jjd d d dt|j|jjd d d dt|j dt kt j j d}trKi|d6ni}|jd|dddddddddd dd|}t|j|jjd dd dt|j|jjd d d ddS(NReRKtminimizet min_methods trust-ncgR#iR"iR gMbP?R g{Gz?g333333?t convergedtdoglegg~jth?iRt basinhoppingtnitertstepsizeg?t niter_successtintervalg-C6?(R R$t mle_settingsRWRR R RRt mle_retvalsRXRRt RandomStateRR( Rtreset_randomstateR$Retres_ncgtres_dogt random_stateRtres_bh((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyt test_minimizes6        ( RARBR_R^RRRR R(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRs     t*TestZeroInflatedGeneralizedPoisson_predictcB s2eZedZdZdZdZRS(c C sdddg}tjjdd}tj|df}d|d|ddf<|j|d }tjjj||dddd|j |_ tj |j |dd}|j d d d d d d dd|_ dS(Nig?iiiiRRRKRR#iRR"i(RRRR%RRORt zigenpoissonRR7RTRRWR(R\RRR5RR$((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^7scC s5t|jjj|jjdddddS(NR g-C6?R (RRRgRRT(R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRDs$cC sEt|jjj|jjj|jjdddddS(NR g?R g?(RRRgRRRTR(R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRHscC s|j}|jj}|jdd}tjjjtj ddddf|jdddj }t ||dddd dS( NRRi g?iR g{Gz?R g?( RR$RTRgRORtzinegbinRRRRRR(RRRTRR((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRMs   ((RARBR_R^RRR(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR6s  t!TestZeroInflatedNegativeBinomialPcB sDeZedZdZdZdZdZdZRS(c C s.tjjjdt}|j|_tj|jdddfdt}tj|jdddfdt}tj ddddd g}tj |j|d |d d j d |dddddddd|_ t |j j_d dddd g|_idd6d d 6|_tj}||_dS(NRGiRHigGz?g(\$gɿg= ףp=?gq= ףp?R6RiReRKtnmtxtolgư>R#iR"RMRIRNRJ(RORPRQRRRSRTRUR5RtarraytZeroInflatedNegativeBinomialPRWR RXRYRZR-R.Rtzero_inflated_negative_binomialR (R\R]R5R6tspR ((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^Ws (( cC s)t|jj|jjdddddS(NR gMbP?R (RR R R (R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRjscC sdS(N((R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRnscC sdS(N((R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRqscC s|jj}tjt|jj}d|d)|jd|ddddd}t|jd|jjdd d d d dS( NiiR!g{Gz?R"R#iiR g?R (R R$RR%R&R R'R(RR$R!R(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR)ts   "c C s|jj}|jjd}|jd|dddddddd }t|j|jjd d d d t|j|jjd d d dt|j dt k|jd|dddddddd }t|j|jjd d d dt|j|jjd d d dt|j dt k|jd|dddddddd }t|j|jjd dd dt|j|jjd d d d dS(NReRKRRs trust-ncgR#iR"iR gMbP?R gQ?gQ?RRg~jth?gy&1|?RRig-C6?ga2U0*3?( R R$RRWRR R RRRRX(RRR$ReRRR((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRs2         ( RARBR_R^RRRR)R(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRVs     t)TestZeroInflatedNegativeBinomialP_predictcB s;eZedZdZdZdZdZRS(c C sdddg}tjjdd}tj|df}d|d|ddfF/Y˯?gsn?gkv錵?gz?gO.?R6RiReRKRR#iR"i( RORPRQRRRSRTR5RRRRWR(R\R]R5RetmodR((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR^s  cC s/t|jjj|jjdddS(NR g{Gz?(RRRgRRT(R((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyR's$cC s^|jj}d|jjddj|jjddj}t||dddS(NiRs prob-zeros mean-nonzeroR g?(RTRRRgR(Rtmean1tmean2((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyttest_zero_nonzero_mean+s(RARBR_R^RR(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyRs (!t __future__Rtstatsmodels.compat.scipyRtnumpyRt numpy.testingRRRRRCtstatsmodels.apitapiROtresults.results_discreteRt test_discreteRRRFR`RcRttobjectRRDtslowRRRRR(((sJlib/python2.7/site-packages/statsmodels/discrete/tests/test_count_model.pyts& " :>' F Oo