p7]c@s:dZddlmZmZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlZddlmZddlmZmZddlmZmZmZmZmZddlmZdd lmZd Zd Z d Z!d Z"dZ#dZ$yddl%Z%e&Z'Wne(k r5e)Z'nXde*fdYZ+de+fdYZ,de+fdYZ-de*fdYZ.de*fdYZ/de*fdYZ0de*fdYZ1de*fdYZ2d e*fd!YZ3d"e+fd#YZ4d$e*fd%YZ5d&e*fd'YZ6d(e6fd)YZ7d*e*fd+YZ8d,e*fd-YZ9d.e+fd/YZ:d0Z;d1Z<d2e+fd3YZ=d4Z>d5e+fd6YZ?d7e+fd8YZ@d9e*fd:YZAd;e+fd<YZBd=eBfd>YZCd?eBfd@YZDdAeBfdBYZEdCeBfdDYZFdEZGdFZHdGZIdHZJdIZKdJe*fdKYZLdLZMdMZNdNZOdOZPdPZQdQZRdRZSdSZTdTZUdUZVejWjXe' dVdWdXZYdS(Ys& Test functions for models.regression i(tlongtlrangeN(tassert_almost_equaltassert_t assert_raisest assert_equaltassert_allclose(ttoeplitz(t add_constantt categorical(tOLStWLStGLSt yule_walkertburg(tlongley(ttiiiiiitCheckRegressionResultscBsCeZdZeZdZeZdZeZdZ eZ dZ eZ dZ eZdZeZdZdZeZd ZeZd ZeZd ZeZd ZeZd ZeZdZeZdZ eZ!dZ"eZ#dZ$eZ%dZ&eZ'dZ(eZ)dZ*eZ+dZ,RS(s res2 contains results from Rmodelwrap or were obtained from a statistical packages such as R, Stata, or SAS and were written to model_results cCs#t|jj|jj|jdS(N(Rtres1tparamstres2tdecimal_params(tself((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_params'scCs#t|jj|jj|jdS(N(RRtbseRtdecimal_standarderrors(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_standarderrors,scCs|jj}|jj}xptt|D]\}t||d||ddd|j t||d||ddd|j q1WdS(Nitrtoli i(Rtconf_intRtrangetlenRtdecimal_confidenceintervals(Rtconf1tconf2ti((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_confidenceintervals1scCs]t|jjdkrY|jjdd}|jjdd!}t|||jndS(Nitcolsii(ii(RRRRRtdecimal_conf_int_subset(Rtci1tci2((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_conf_int_subset<s cCs#t|jj|jj|jdS(N(RRtscaleRt decimal_scale(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_scaleEscCs#t|jj|jj|jdS(N(RRtrsquaredRtdecimal_rsquared(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_rsquaredJscCs#t|jj|jj|jdS(N(RRt rsquared_adjRtdecimal_rsquared_adj(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_rsquared_adjOscCs<t|jjj|jjt|jjj|jjdS(N(RRtmodeltdf_modelRtdf_resid(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_degreesSscCs#t|jj|jj|jdS(N(RRtessRt decimal_ess(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_essXscCs#t|jj|jj|jdS(N(RRtssrRt decimal_ssr(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_sumof_squaredresids^scCs#t|jj|jj|jdS(N(RRt mse_modelRtdecimal_mse_resid(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_mse_residbscCs#t|jj|jj|jdS(N(RRt mse_residRtdecimal_mse_model(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_mse_modelhscCs-t|jj|jj|jdd|dS(Nterr_msgs Test class %s(RRt mse_totalRtdecimal_mse_total(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_mse_totalms cCs#t|jj|jj|jdS(N(RRtfvalueRtdecimal_fvalue(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_fvaluesscCs#t|jj|jj|jdS(N(RRtllfRtdecimal_loglike(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_loglikezscCs#t|jj|jj|jdS(N(RRtaicRt decimal_aic(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_aic~scCs#t|jj|jj|jdS(N(RRtbicRt decimal_bic(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_bicscCs#t|jj|jj|jdS(N(RRtpvaluesRtdecimal_pvalues(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_pvaluesscCs#t|jj|jj|jdS(N(RRtwresidRtdecimal_wresid(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_wresidscCs#t|jj|jj|jdS(N(RRtresidRtdecimal_resids(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_residsscCs#t|jj|jj|jdS(N(RRt resid_pearsonRtdecimal_norm_resids(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_norm_residss(-t__name__t __module__t__doc__t DECIMAL_4RRRRRR#R%R(R*R+R-R.R0R1R5R7R8R:R;R=R>R@RARDRERGRHRJRKRMRNRPRQRSRTRVRWRYRZR\R](((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR sT                    tTestOLScBseZedZdZdZdZdZdZdZ dZ dZ d Z d Z d Zd ZRS( c CsAddlm}tjdt}t|jdt|_t|j|jj }|}|j |_ ||_ ||_ t|j|jj dd}t|j|j}t jj|j\}}|||_|_t jjt j|j||_t jj||_|j dd} ||_| |_dS(Ni(tLongleyt as_pandastprependtmethodtqr(tresults.results_regressionRcRtloadtFalseRtexogR tendogtfitRURRtnptlinalgRgtexog_Qtexog_RtinvtdottTtnormalized_cov_paramst matrix_ranktranktres_qrt res_qr_manual( tclsRctdataRRRxtmodel_qrtQtRtres_qr2((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt setup_classs"    !$ cCsI|jj|jj}||jj:}tj|}t||tdS(N(Rxt eigenvalsRyRnt zeros_likeRt DECIMAL_7(Rteigenval_perc_difftzeros((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_eigenvaluess   cCsRt|jjd |jjd tttj|jjd|jjddS(Ni(RRtHC0_seRRaRRntround(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_HC0_errorsscCsOt|jjd |jjd tt|jjd|jjddddS(NiRgHך>(RRtHC1_seRRaR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_HC1_errorss cCsOt|jjd |jjd tt|jjd|jjddddS(NiRgƠ>(RRtHC2_seRRaR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_HC2_errorss cCsOt|jjd |jjd tt|jjd|jjddddS(NiRgv !>(RRtHC3_seRRaR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_HC3_errorss cCs t|jj|jjddS(Ni(RRRRx(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_qr_paramss cCs3ttj|jj|jj|jjddS(Ni(RRnt ones_likeRRuRx(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_qr_normalized_cov_paramss cCstjdt}t|jdt|_tj|jdddgi (RRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRfscCst|jjddS(Ni (RRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRiscCst|jjddS(Ni(RRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRls( R^R_R`RRRHRRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRQs    t TestTtestcBsSeZdZedZdZdZdZdZdZ dZ RS(si Test individual t-tests. Ie., are the coefficients significantly different than zero. cCstjdt}t|jdt|_t|j|jj|_t j d}|jj ||_ d}|jj ||_ dS(NRdReis9x1 = 0, x2 = 0, x3 = 0, x4 = 0, x5 = 0, x6 = 0, const = 0(RRiRjRRkR RlRmRRnRtt_testtTtesttNewTTest(RzR{R~R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRuscCst|jj|jjdS(N(RRttvalueR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_new_tvaluescCs t|jj|jjtdS(N(RRRRttvaluesRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_tvaluescCs t|jj|jjtdS(N(RRtsdRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_sdscCsBt|jjtjtj|jj|jj j dt dS(Ni( RRRt student_ttsfRntabsRRR2R4Ra(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs%cCs t|jj|jjjdS(N(RRRRR2R4(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCst|jj|jjdS(N(RRteffectRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_effects( R^R_R`RRRRRRRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRos     t TestTtest2cBsJeZdZedZdZdZdZdZdZ RS(s~ Tests the hypothesis that the coefficients on POP and YEAR are equal. Results from RPy using 'car' package. cCs}tjd}ddg|dd+tjdt}t|jdt|_t|j|jj }|j ||_ dS(NiiiiiRdRe( RnRRRiRjRRkR RlRmRtTtest1(RzR~R{R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs cCst|jjdtdS(NgN-(RRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCst|jjdtdS(Ng9y@v|@(RRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCst|jjdtdS(Nig6X%X?g6X%h?(RRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCst|jjddS(Ni (RRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCst|jjdtdS(Ng<,nϔ(RRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs( R^R_R`RRRRRRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs    tTestGLScBseZdZedZdZdZdZdZdZ dZ dZ d Z d Z d Zd ZRS( sA These test results were obtained by replication with R. c Csddlm}tjdt}ttj|jdddf|jdddffdt}t |j |j }tj |j d|j d dd}ttjd}||}t|j |d |j }||_||_||_||_|j |_ dS( Ni(t LongleyGlsRdiReiiitsigma(RhRRRiRjRRnt column_stackRkR RlRmtcorrcoefRXRtarangeR RRR( RzRR{Rkt tmp_resultstrhotorderRt GLS_results((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs >     cCs't|jjd|jjdddS(NiRgMbp?(RRRLR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRNscCs#t|jj|jjdddS(NRgQ?(RRROR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRQscCs t|jj|jjtdS(N(RRRIRt DECIMAL_0(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRKscCs t|jj|jjtdS(N(RRRRt DECIMAL_1(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCs t|jj|jjtdS(N(RRRXRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_residscCs t|jj|jjtdS(N(RRR)RRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR+scCs t|jj|jjtdS(N(RRRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_tvaluesscCs t|jj|jjtdS(N(RRRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCs t|jj|jjtdS(N(RRt fittedvaluesRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_fittedvaluesscCs t|jj|jjtdS(N(RRRRRRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRTscCs|jj}tj|dddg Test that GLS with no argument is equivalent to OLS. cCstjdt}t|jdt|_t|j|jj}t|j|jj}t|j|jdd}|j|_|j|_||_ ||_ ||_ dS(NRdReRi( RRiRjRRkR RlRmR RRtres3(RzR{tols_restgls_restgls_res_scalar((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs    c Cs?t|j}ttt|j|jdtj|ddS(NRi(RRlRt ValueErrorR RkRntones(Rtn((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_wrong_size_sigma_1dsc CsIt|j}ttt|j|jdtj|d|dfdS(NRi(RRlRRR RkRnR(RR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_wrong_size_sigma_2d s(R^R_R`RRRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs tTestLMcBsDeZedZdZdZdZdZdZRS(cCstjjdd}tjd}tjjdd}tj|||}t||j|_t||dddfj|_|jj d|_ |jj d|_ ||_ ||_ dS(NidiiitHC0(ii(RnRRRRsR Rmt res1_fulltres1_restrictedtget_robustcov_resultst res2_fulltres2_restrictedtXtY(RzRtbteR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs( c Cs|jj}|jd}|j}tj|||tj|j||}tjj|}tj ||dddfd}|tjtj|||j}|j j |j}|d} t || tdS(Ni(RRURRRnRsRtRoRrtmeantNoneRtcompare_lm_testRR( RRXRRtStSinvtstLMstatt LMstat_OLStLMstat2((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_LM_homoskedastic+s   -&% c Cs|jj}|jd}|j}||dddf}tj|j||}tjj |}tj |d}|tjtj|||j}|j j |j dt} | d} t|| tdS(Nitdemean(RRURRRRnRsRtRoRrRRRRRjRR( RRXRRtscoresRRRRRR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_LM_heteroskedastic_nodemean7s   %  c Cs|jj}|jd}|j}||dddf}||jd}tj|j||}tj j |}tj|d}|tjtj|||j} |j j |j } | d} t| | tdS(Ni(RRURRRRRnRsRtRoRrRRRRR( RRXRRR t scores_demeanRRRRRR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_LM_heteroskedastic_demeanEs   % c Cs|jj}|jj}|jd}|j}||dddf}tj|d}||dddf}tj|j ||}tj j |}|tjtj|||j } |j j |jdt} | d} t| | tdS(Nituse_lr(RRURRRRRnRRsRtRoRrRRRRRR( RRXt resid_fullRRR RRRRRR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt!test_LM_heteroskedastic_LRversionSs    %  cCstt|jj|jdS(N(RRRRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_LM_nonnestedds( R^R_RRRR R RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs    tTestOLS_GLS_WLS_equivalencecBs;eZedZdZdZdZdZRS(cCstjdt}t|jdt|_|j}|j}|jd}tj|}g|_ |j j t ||j |j j t |||j |j j t||d|j |j j t||tjd|j dS(NRdReiidg?(RRiRjRRkRlRRnRRtappendR RmR R tdiag(RzR{RRRtw((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRks    "&cCstjg|jD]}|j^q}tj||jdj}t||ttjg|jD]}|j^qe}tj||jdj}t||ttjg|jD]}|j^q}tj||jdj}t||tdS(Ni( RnRRRIRRRRLRO(RtrRItllf_1tictic_1((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_llys(((cCsbtjg|jD]}|j^q}tj|jdjgt|j}t||dS(Ni(RnRRRRR(RRRtparams_1((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs()cCsbtjg|jD]}|j^q}tj|jdjgt|j}t||dS(Ni(RnRRRRR(RRRtbse_1((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_sss()cCsetjg|jD]}|j^q}tj|jdjgt|j}t||tdS(Ni(RnRRR,RRR(RRR,t rsquared_1((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR.s()(R^R_RRRRRR.(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRis   tTestGLS_WLS_equivalencecBseZedZRS(cCs%tjdt}t|jdt|_|j}|j}|jd}tjj dtjj dd|}d|}g|_ |j j t |||j|j j t ||d|j|j j t||d |j|j j t||tjd |jdS( NRdReiig?ig?g{Gz?idg?(RRiRjRRkRlRRnRtseedtuniformRRR RmR R(RzR{RRRRtw_inv((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs     "&&(R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRst TestNonFitcBs eZedZdZRS(cCs^tjdt}t|jdt|_|j|_|j|_t|j|j|_dS(NRdRe(RRiRjRRkRlR t ols_model(RzR{((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs   cCs;|jjd|jjd}t|jjtddS(Niii (RlRRkRR#R4R(RR4((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_df_resids(R^R_RRR$(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR"stTestWLS_CornerCasescBs eZedZdZRS(cCsRtjd|_tjd|_d}t|j|jd|j|_dS(Nitweights(i(i(RnRRkRlR Rmtwls_res(RzR&((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRscCs2tjd}ttt|j|jd|dS(Ni R&(i i (RnRRRR RlRk(RR&((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_wrong_size_weightss(R^R_RRR((((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR%stTestWLSExogWeightscBseZedZRS(cCs;ddlm}ddlm}|dt}t|jdt|_d}d|jdddf}|||j}t|j |jd |j |_ ||_ |d |j j |j _d|j j|j j}|j j|8_|j j|8_|j jd tjtj|j jj7_dS( Ni(tCCardWLSi(RiRdRegR@iR&g?(RhR*tstatsmodels.datasets.ccardRiRjRRkRR RlRmRRRXRURIRLRORntlogR2R&(RzR*RitdtatnobsR&tscaled_weightstcorr_ic((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs$ (R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR)scCsdddddddg}tdd}t|dt}t||dtddj}t|jd d t|jd d dS( NiiiiiiReR&gg@L?ig,@g@(RRRjR RmRRFR)(RRt wls_model((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_wls_examples $cCs tjddddddg}ddgddgddgddgddgddgg}t|t|dtj}tjdddg}ddgddgddgg}tjdddg}t|t|dtd |j}t|j|jdS( NiiiiReg7@ig@R&( RnRR RRjRmR Rt centered_tss(RRtols_modtywtXwRtwls_mod((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_wls_tsss!<!!'tTestWLSScalarVsArraycBseZedZRS(cCsddlm}|dt}t|jdt|_t|j|jdddj}dd gt |j}t|j|jd|j}||_ ||_ dS( Ni(RiRdReR&g?iig@( tstatsmodels.datasets.longleyRiRjRRkRR RlRmRRR(RzRiR-t wls_scalarR&t wls_array((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs%! (R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR9scCsddlm}|dt}|j}tj|ddgst TestGLS_OLScBs eZedZdZRS(cCsjtjdt}t|jdt|_t|j|jj|_t |j|jj|_ dS(NRdRe( RRiRjRRkR RlRmRR R(RzR{((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR+scCst||tdS(N(RRa(RR R!((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR?2s(R^R_RRR?(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR@*stTestYuleWalkercBs eZedZdZRS(cCs_ddlm}|dt}t|jdddd\|_|_dd d d g|_dS( Ni(RiRdRiRftmleg."0?g3Eܿg@**ʿg0?(tstatsmodels.datasets.sunspotsRiRjR RlRRtR_params(RzRiR{((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRGs cCst|j|jtdS(N(RRRDRa(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRPs(R^R_RRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRAFs tTestDataDimensionscBs eZedZdZRS(cCstjjdtjjdddd|_|jdddf|_tjjdddd|_|jdddf|_|jd |_ t |j|j|_ |j j d7_ |j j |_t |j|j|_|jj d7_ |jj |_dS(Ni1iitsizeiii(RnRRR tendog_n_Rt endog_n_onetexog_n_t exog_n_onet degen_exogR tmod1R3RmRtmod2R(Rz((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRUscCst||tdS(N(RRa(RR R!((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyR?fs(R^R_RRR?(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRETstTestGLS_large_datacBs2eZedZdZdZdZRS(cCstt|jd}tjj|d}tjj|d}tj|}t||d|j|_ t||ddj|_ t||j|_ t ||j|_ dS(NiiiR(tsuperRNRRnRRRR RmRRt gls_res_noneR R(RzR.RRR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRkscCs t|jj|jjtdS(N(RRRRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_large_equal_paramswscCs t|jj|jjtdS(N(RRRIRR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_large_equal_loglikezscCs t|jj|jjtdS(N(RRRRPR(R((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_large_equal_params_none}s(R^R_RRRQRRRS(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRNjs  tTestNxNxcBseZedZRS(cCsStt|jt|j|j|_|jjd7_|jj|_ dS(Ni( RORTRR RGRIRMR3RmR(Rz((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs(R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRTst TestNxOneNxcBseZedZRS(cCsStt|jt|j|j|_|jjd7_|jj|_ dS(Ni( RORURR RHRIRMR3RmR(Rz((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs(R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRUst TestNxNxOnecBseZedZRS(cCsStt|jt|j|j|_|jjd7_|jj|_ dS(Ni( RORVRR RGRJRMR3RmR(Rz((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRs(R^R_RR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRVscCsCtjjdtjjddd}ttt||ddS(Ni1iiii(RnRRR RRR (R{((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_bad_sizescCs tjjdtjjdddd}t|dt}tj|ddd gtjjdd}t|t |ddd dfd tj }t||d tj }t |j |j d |j jjd kst|j jjd ks tdS(Ni90iiRFiRg?g@g@iRethasconsti (RnRRtrandintR RRstnormalR RRmRR,R2R{t k_constanttAssertionError(RRtrescR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_const_indicators.7cCstjjdtjjdddd}t|dt}d|dddf |$t$|$} & \textbf{[0.025} & \textbf{0.975]} \\ \midrule \textbf{GNPDEFL} & 15.0619 & 84.915 & 0.177 & 0.863 & -177.029 & 207.153 \\ \textbf{GNP} & -0.0358 & 0.033 & -1.070 & 0.313 & -0.112 & 0.040 \\ \textbf{UNEMP} & -2.0202 & 0.488 & -4.136 & 0.003 & -3.125 & -0.915 \\ \textbf{ARMED} & -1.0332 & 0.214 & -4.822 & 0.001 & -1.518 & -0.549 \\ \textbf{POP} & -0.0511 & 0.226 & -0.226 & 0.826 & -0.563 & 0.460 \\ \textbf{YEAR} & 1829.1515 & 455.478 & 4.016 & 0.003 & 798.788 & 2859.515 \\ \textbf{constant} & -3.482e+06 & 8.9e+05 & -3.911 & 0.004 & -5.5e+06 & -1.47e+06 \\ \bottomrule \end{tabular} \begin{tabular}{lclc} \textbf{Omnibus:} & 0.749 & \textbf{ Durbin-Watson: } & 2.559 \\ \textbf{Prob(Omnibus):} & 0.688 & \textbf{ Jarque-Bera (JB): } & 0.684 \\ \textbf{Skew:} & 0.420 & \textbf{ Prob(JB): } & 0.710 \\ \textbf{Kurtosis:} & 2.434 & \textbf{ Cond. No. } & 4.86e+09 \\ \bottomrule \end{tabular} %\caption{OLS Regression Results} \end{center} Warnings: \newline [1] Standard Errors assume that the covariance matrix of the errors is correctly specified. \newline [2] The condition number is large, 4.86e+09. This might indicate that there are \newline strong multicollinearity or other numerical problems.(treRt load_pandasRkRlR Rmtpytesttwarnst UserWarningtsummarytas_latextsubR(RjR-RRRttabletexpected((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_summary_as_latexs         (tTestRegularizedFitcBs#eZdZdZdZRS(cCstjjdd}tjjd|}tjjd|df}xHtttfD]7}|||}|jdd}t|j dqVWdS(NiidRFitalphaig( RnRRRZR R R tfit_regularizedRR(RRRlRkRzR2tresult((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_empty_modelsc Cs ddl}ddlm}|jj|jjt}tj|jj |dddd}gt |D]}|j drn|^qn}xt|D]l}t ||}|d } |d} t |d } t |d } |d jtj} |d t| d f}|d t| dt| df}||j}||jd d:}||jd }||jd d d:}xptttfD]_}|||}|jd| d| }t|j| dd |jd| d| dtqWqWdS(Nii(tglmnet_r_resultsRslasso_data.csvt delimitert,trslt_iiiitddoftL1_wtRvtdecimalt profile_scale(tosRRztpathtdirnametabspatht__file__Rntloadtxttjointdirt startswithtgetattrtfloattastypetfloat64tintRtstdR R R RwRRR(RRRztcur_dirR{RtteststtesttvecRtpRtlamRRlRkRzRtrslt((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_regularized s2  .   ,cCstjjdtjjdd}|dddf|dddftjjdd}tjjdd}|dddf|dddftjjdd}tj|||f}tj|||f}tj||f}tj||f}tjd} d| dd+tjd| } xdD]} xdD]} t||} | j d | d | }t ||d | }|j d | d | }t ||d | }|j d | d | }t |j |j ddt |j |j ddqQWqDWdS(NiRFidiiiiig?RRvR&RR(idi(idi(ig?i(ii(RnRRRZtvstackt concatenateRRR RwR R RR(Rtexog1tendog1texog2tendog2texog_atendog_atexog_btendog_btwgtsRRRvRLtrslt1RMtrslt2tmod3trslt3((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_regularized_weights3s,==   (R^R_RyRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyRus &cCsddlj}ddlm}ddlm}|jjjj }t j |j |j ddf<|ddd|j}|j}|ddd|}|j}t|jj|jjt||dd|d d dS( Ni(tols(t PatsyErroritfirmtformulas&value ~ invest + capital + firm + yearR{Rtraise(tstatsmodels.apitapitstatsmodels.formula.apiRtpatsyRtdatasetstgrunfeldRkR{RnRtloctindextdropnaRmRRtvaluesR(tsmRRR-RRRMR((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_formula_missing_catTs     cCsd}tjdd|}tjtjg}tjitj||fd6}tjddg}tjjd|d}|d|d|d||d( RnRt RandomStateRRRRRjtallR\(trndRRRsR"tart_t as_nodemean((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyt test_burgs  c CstjtttjdWdQXtjtttjjddWdQXtjtttjjddWdQXdS(Nidiitapple(idi(RltraisesRRRnRRR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_burg_errorss treasonssqrt_lasso requires cvxoptcCstjjdd}d}tj|}dtjtjj||}tjj|}tj tjj d||f|j }tj |}dddddg|dd+d d l m}d tj||jdd d |}tj ||dtjj d|} idt6dt6} itjd"t6tjd#t6} x3ttfD]%} t| |jddd|d| } | j|}tjtj |tj ||}t| |ddddfj}tj |}|j|dd!|dd+tjtj |tj ||}t||| | dddd t| jdd!| | dd!dd!q\WdS($Niidig?RFiiii(tnormg?g?ig?igtr}?gˑ?g9B?gR ?g-  ?gg<5o?gY}|?gF?gT\Z:?gbj?Rft sqrt_lassoRvRRRg?gh㈵>(gtr}?gˑ?g9B?gR ?g-  ?(gg<5o?gY}|?gF?gT\Z:?gbj?(RnRRRRtsubtracttouterRotcholeskyRsRZRtRtscipy.stats.distributionsRRtppfRjRtr_R RwRRmR(RRRtcxtcxrRRRRvRtexpected_oracletexpected_paramsRRterrtnumertoraclet oracle_errtdenom((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyttest_sqrt_lassos6"*,)' '+' (ZR`tstatsmodels.compat.pythonRRRRctnumpyRnt numpy.testingRRRRRRlt scipy.linalgRtstatsmodels.tools.toolsRR RR R R R Rtstatsmodels.datasetsRt scipy.statsRRRat DECIMAL_3t DECIMAL_2RRRtcvxoptRt has_cvxoptt ImportErrorRjtobjectRRbRRRRRRRRRRRR"R%R)R2R8R9R=R>R@RARERNRTRURVRWR^RbRhRtRuRRRRRRRRRRtmarktskipifR(((sKlib/python2.7/site-packages/statsmodels/regression/tests/test_regression.pyts   ( (    |l#$!B S-        :V