ó áp7]c@sdZddlZddlmZmZddlZddlmZddl j Z ddl m Z ddlmZddlmZdZd Zd Zd Zd efd „ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZ ej!j"d „ƒZ#dS(!s Test functions for sm.rlm iÿÿÿÿN(tassert_almost_equaltassert_allclose(tstats(tRLM(tnorms(t HuberScaleiiiitCheckRlmResultsMixincBsƒeZdZd„ZeZd„Zd„ZeZd„Z d„Z d„Z d„Z d„Z eZd „Zd „Zd „ZRS( sû res2 contains results from Rmodelwrap or were obtained from a statistical packages such as R, Stata, or SAS and written to results.results_rlm Covariance matrices were obtained from SAS and are imported from results.results_rlm cCs t|jj|jjtƒdS(N(Rtres1tparamstres2t DECIMAL_4(tself((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_paramsscCs#t|jj|jj|jƒdS(N(RRtbseR tdecimal_standarderrors(R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyttest_standarderrors scCsHt|jdƒs"tjdƒnt|jjƒ|jjƒtƒdS(Ntconf_intsResults from R(thasattrR tpytesttskipRRRR (R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyttest_confidenceintervals%scCs#t|jj|jj|jƒdS(N(RRtscaleR t decimal_scale(R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_scale.scCs t|jj|jjtƒdS(N(RRtweightsR R (R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_weights2scCs t|jj|jjtƒdS(N(RRtresidR R (R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyttest_residuals5scCsBt|jjj|jjtƒt|jjj|jjtƒdS(N(RRtmodeltdf_modelR R tdf_resid(R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_degrees8scCsBt|jdƒs"tjdƒnt|jj|jjtƒdS(Nt bcov_unscaledsNo unscaled cov matrix from SAS(RR RRRRR R (R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyttest_bcov_unscaled>s cCsat|jj|jj|jƒt|jj|jj|jƒt|jj|jj|jƒdS(N(RRt bcov_scaledR th1tdecimal_bcov_scaledth2th3(R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyttest_bcov_scaledGs   cCsEt|jdƒs"tjdƒnt|jj|jjddƒdS(NttvaluessNo tvalues in benchmarktrtolgú~j¼t“h?(RR RRRRR((R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_tvaluesOscCs´|jj}||jj}tjjtj|ƒƒd}tjjdƒ|jj}tj ||||fƒ}t |jj |ƒt |jj |ƒt |jj ƒ|ƒdS(Nigš™™™™™™?(RRR Rtnormtsftnptabstisft column_stackRR(tpvaluesR(R RR(R1t half_widthR((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_tpvaluesUs (t__name__t __module__t__doc__R R RRRRRRRRR!R$R'R*R3(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRs          tTestRlmcBs5eZed„ƒZd„Zejjd„ƒZRS(cCs×ddlm}|dtƒ|_tj|jjdtƒ|j_t|_t |_ t |jj |jjdt jƒƒ}|jƒ}|jddƒj}|jddƒj}||_||j_||j_dS( Niÿÿÿÿ(tloadt as_pandastprependtMtcovtH2tH3(tstatsmodels.datasets.stacklossR8tFalsetdatatsmt add_constanttexogt DECIMAL_1Rt DECIMAL_3RRtendogRtHuberTtfitR"RR%R&(tclsR8RtresultsR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt setup_classds!  '   cCs ddlm}|ƒ|_dS(Ni(tHuber(tresults.results_rlmRMR (R RM((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pytsetupuscCs|jjƒdS(N(Rtsummary(R ((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_summaryys( R4R5t classmethodRLRORtmarktsmokeRQ(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyR7cs t TestHampelcBs eZed„ƒZd„ZRS(cCs°tt|ƒjƒt|_t|_t|_t|j j |j j dt j ƒƒ}|jƒ}|jddƒj}|jddƒj}||_||j_||j_dS(NR;R<R=R>(tsuperRURLt DECIMAL_2RRFRR$RRARGRDRtHampelRIR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRLs   '   cCs ddlm}|ƒ|_dS(Ni(RX(RNRXR (R RX((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRU~stTestRlmBisquarecBs eZed„ƒZd„ZRS(cCsžtt|ƒjƒt|_t|jj|jjdt j ƒƒ}|j ƒ}|j ddƒj }|j ddƒj }||_ ||j _||j _dS(NR;R<R=R>(RVRYRLRERRRARGRDRt TukeyBiweightRIR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRL•s '   cCs ddlm}|ƒ|_dS(Ni(tBiSquare(RNR[R (R R[((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRO£s(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRY”stTestRlmAndrewscBs eZed„ƒZd„ZRS(cCs•tt|ƒjƒt|jj|jjdtjƒƒ}|j ƒ}|j ddƒj }|j ddƒj }||_ ||j _ ||j _ dS(NR;R<R=R>(RVR\RLRRARGRDRt AndrewWaveRIR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRL©s'   cCs ddlm}|ƒ|_dS(Ni(tAndrews(RNR^R (R R^((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROµs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyR\¨s t TestRlmHubercBs eZed„ƒZd„ZRS(cCsàddlm}|dtƒ|_tj|jjdtƒ|j_t|jj|jjdt j ƒƒ}|j dt ƒƒ}|j dddt ƒƒj }|j dd dt ƒƒj }||_||j_||j_dS( Niÿÿÿÿ(R8R9R:R;t scale_estR<R=R>(R?R8R@RARBRCRDRRGRRHRIRR"RR%R&(RJR8RRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRL¾s!'  cCs ddlm}|ƒ|_dS(Ni(t HuberHuber(RNRaR (R Ra((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROÌs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyR_½stTestHampelHubercBs eZed„ƒZd„ZRS(cCs°tt|ƒjƒt|jj|jjdtjƒƒ}|j dt ƒƒ}|j dddt ƒƒj }|j dddt ƒƒj }||_ ||j _ ||j _dS(NR;R`R<R=R>(RVRbRLRRARGRDRRXRIRR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRLÒs'  cCs ddlm}|ƒ|_dS(Ni(t HampelHuber(RNRcR (R Rc((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROÞs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRbÑs tTestRlmBisquareHubercBs eZed„ƒZd„ZRS(cCs°tt|ƒjƒt|jj|jjdtjƒƒ}|j dt ƒƒ}|j dddt ƒƒj }|j dddt ƒƒj }||_ ||j _ ||j _dS(NR;R`R<R=R>(RVRdRLRRARGRDRRZRIRR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRLäs'  cCs ddlm}|ƒ|_dS(Ni(t BisquareHuber(RNReR (R Re((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROðs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRdãs tTestRlmAndrewsHubercBs eZed„ƒZd„ZRS(cCs°tt|ƒjƒt|jj|jjdtjƒƒ}|j dt ƒƒ}|j dddt ƒƒj }|j dddt ƒƒj }||_ ||j _ ||j _dS(NR;R`R<R=R>(RVRfRLRRARGRDRR]RIRR"RR%R&(RJRRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRLös'  cCs ddlm}|ƒ|_dS(Ni(t AndrewsHuber(RNRgR (R Rg((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRfõs t TestRlmSresidcBs eZed„ƒZd„ZRS(cCsÝddlm}|dtƒ|_tj|jjdtƒ|j_t|_t |_ t |jj |jjdt jƒƒ}|jddƒ}|jdd ƒj}|jdd ƒj}||_||j_||j_dS( Niÿÿÿÿ(R8R9R:R;tconvtsresidR<R=R>(R?R8R@RARBRCRDRERRFRRRGRRHRIR"RR%R&(RJR8RRKR%R&((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRL s!  '  cCs ddlm}|ƒ|_dS(Ni(RM(RNRMR (R RM((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyROs(R4R5RRRLRO(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyRhscCsXddljj}iddddgd6dddtjgd6}|jd d |ƒdS( Niÿÿÿÿiii i•tFooitBars Foo ~ BarRA(tstatsmodels.formula.apitformulatapiR-tnantrlm(tsmftd((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyt test_missings/($R6tnumpyR-t numpy.testingRRRtscipyRtstatsmodels.apiRoRBt&statsmodels.robust.robust_linear_modelRtstatsmodels.robustRtstatsmodels.robust.scaleRR RFRWREtobjectRR7RURYR\R_RbRdRfRhRSRTRt(((s@lib/python2.7/site-packages/statsmodels/robust/tests/test_rlm.pyts.  P