ó áp7]c@sôddlZddlZddlmZddljZdefd„ƒYZ de fd„ƒYZ de fd„ƒYZ d e fd „ƒYZ d e fd „ƒYZ d efd„ƒYZde fd„ƒYZdefd„ƒYZd„ZdS(iÿÿÿÿN(tstatstBaseProbplotMixincBsLeZd„Zejjd„ƒZejjd„ƒZejjd„ƒZejjd„ƒZ ejjd„ƒZ ejjd„ƒZ ejjd„ƒZ ejjd„ƒZ ejjd „ƒZejjd „ƒZejjd „ƒZejjd „ƒZejjd „ƒZejjd„ƒZejjd„ƒZRS(cCsyy+ddlj}|jƒ\|_|_Wntk r>nXtjjd|j j j ƒ|_ t j|j ƒ|_dS(Niÿÿÿÿtsize(tmatplotlib.pyplottpyplottsubplotstfigtaxt ImportErrortnptrandomtnormaltprbplttdatatshapet other_arraytsmtProbPlott other_prbplot(tselftplt((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pytsetup s !cCs#|jjd|jd|jƒdS(NRtline(R tqqplotRR(Rt close_figures((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyt test_qqplotscCs#|jjd|jd|jƒdS(NRR(R tppplotRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyt test_ppplotscCs#|jjd|jd|jƒdS(NRR(R tprobplotRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyt test_probplotscCs,|jjd|jd|jd|jƒdS(NRRtother(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_qqplot_other_arrayscCs,|jjd|jd|jd|jƒdS(NRRR(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_ppplot_other_array#scCs,|jjd|jd|jd|jƒdS(NRRR(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pytt_est_probplot_other_array(scCs,|jjd|jd|jd|jƒdS(NRRR(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_qqplot_other_prbplt-scCs,|jjd|jd|jd|jƒdS(NRRR(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_ppplot_other_prbplt2scCs,|jjd|jd|jd|jƒdS(NRRR(R RRRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pytt_est_probplot_other_prbplt7sc Cs/|jjd|jd|jddddƒdS(NRRtxlabelsCustom X-LabeltylabelsCustom Y-Label(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_qqplot_custom_labels<sc Cs/|jjd|jd|jddddƒdS(NRRR%sCustom X-LabelR&sCustom Y-Label(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_ppplot_custom_labelsBsc Cs/|jjd|jd|jddddƒdS(NRRR%sCustom X-LabelR&sCustom Y-Label(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_probplot_custom_labelsHsc Cs;|jjd|jd|jddddddd d ƒdS( NRRtmarkertdtmarkerfacecolortcornflowerbluetmarkeredgecolortwhitetalphagà?(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_qqplot_pltkwargsNs c Cs;|jjd|jd|jddddddd d ƒdS( NRRR*R+R,R-R.R/R0gà?(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_ppplot_pltkwargsVs c Cs;|jjd|jd|jddddddd d ƒdS( NRRR*R+R,R-R.R/R0gà?(R RRR(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_probplot_pltkwargs^s (t__name__t __module__Rtpytesttmarkt matplotlibRRRRR R!R"R#R$R'R(R)R1R2R3(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRs  tTestProbPlotLongelycBseZd„ZRS(cCs·tjjdƒtjjjdtƒ|_tj |jj dtƒ|j_ tj |jj |jj ƒj ƒ|_tj|jjtjddƒ|_d|_tt|ƒjƒdS(Nit as_pandastprependtdistargsitr(i(R R tseedRtdatasetstlongleytloadtFalseR t add_constanttexogtOLStendogtfittmod_fitRtresidRttR RtsuperR9R(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRhs!'$ (R4R5R(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR9gstTestProbPlotRandomNormalMinimalcBseZd„ZRS(cCsitjjdƒtjjddddddƒ|_tj|jƒ|_d|_ t t |ƒj ƒdS(Nitlocg€ @tscaleg @Ri%( R R R>R R RRR tNoneRRKRLR(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRss $ (R4R5R(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRLrstTestProbPlotRandomNormalWithFitcBseZd„ZRS(cCsotjjdƒtjjddddddƒ|_tj|jdtƒ|_d |_ t t |ƒj ƒdS( NiRMg€ @RNg @Ri%RGtq( R R R>R R RRtTrueR RRKRPR(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR|s $ (R4R5R(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRP{st TestProbPlotRandomNormalLocScalecBs#eZd„Zd„Zd„ZRS(cCsutjjdƒtjjddddddƒ|_tj|jddddƒ|_d|_t t |ƒj ƒdS( NiRMg€ @RNg @Ri%t45( R R R>R R RRR RRKRSR(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR…s $! cCs|jjdkst‚dS(Ng€ @(R RMtAssertionError(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyt test_loc_setŒscCs|jjdkst‚dS(Ng @(R RNRU(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_scale_sets(R4R5RRVRW(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRS„s  tTestCompareSamplesDifferentSizecBs;eZd„Zejjd„ƒZejjd„ƒZRS(cCsntjjdƒtjtjjddddddƒƒ|_tjtjjddddddƒƒ|_dS( NiRMg€ @RNg @Ri%i7(R R R>RRR tdata1tdata2(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR”s cCsF|jjd|jƒtjtƒ|jjd|jƒWdQXdS(NR(RYRRZR6traisest ValueError(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR›scCs0|jjd|jƒ|jjd|jƒdS(NR(RYRRZ(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR¡s(R4R5RR6R7R8RR(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRX“s t$TestProbPlotRandomNormalLocScaleDistcBs5eZd„Zd„Zd„Zd„Zd„ZRS(cCsutjjdƒtjjddddddƒ|_tj|jdddd ƒ|_d |_t t |ƒj ƒdS( NiRMg€ @RNg @Ri%iiRT( R R R>R R RRR RRKR]R(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR©s $! cCs|jjdkst‚dS(Ni(R RMRU(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRV°scCs|jjdkst‚dS(Ni(R RNRU(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRW³scCs"|jjjƒdkst‚dS(Ng @(R tdisttmeanRU(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_loc_set_in_dist¶scCs"|jjjƒdkst‚dS(Ng"@(R R^tvarRU(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_scale_set_in_dist¹s(R4R5RRVRWR`Rb(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR]§s     t TestTopLevelcBseeZd„Zejjd„ƒZejjd„ƒZejjd„ƒZejjd„ƒZ RS(cCsÐtjjjdtƒ|_tj|jjdtƒ|j_tj|jj |jjƒj ƒ|_ |j j |_ tj|j j tjddƒ|_tjjd|jjjƒ|_tj|jƒ|_dS(NR:R;R<iR(i(RR?R@RARBR RCRDRERFRGRHRItresRRRJR R R R RRR(R((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR¾s!'$!cCstj|jddƒdS(NRR=(RRRd(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRÇsc Cs2tj|jddddddddd d ƒdS( NRR=R*R+R,R-R.R/R0gà?(RRRd(RR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyR1ËscCs=x6ddddgD]"}tj|j|jd|ƒqWdS(NR=RQRTtsR(Rtqqplot_2samplesR R(RRR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyt$test_qqplot_2samples_ProbPlotObjectsÒscCs=x6ddddgD]"}tj|j|jd|ƒqWdS(NR=RQRTReR(RRfRdR(RRR((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_qqplot_2samples_arraysÚs( R4R5RR6R7R8RR1RgRh(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyRc½s  c Cstjjdƒtjjjdtƒ}tj|j dtƒ|_ tj |j |j ƒj ƒ}t jtddƒ!tj|jtjdd ƒWdQXdS( NiR:R;tmatchsdist\(0, 1, 4, loc=0, scale=1\)R<iii(iii(R R R>RR?R@RARBRCRDRERFRGR6R[t TypeErrorRRIRRJ(RR RH((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyttest_invalid_dist_configâs (tnumpyR R6tscipyRtstatsmodels.apitapiRtobjectRR9RLRPRSRXR]RcRk(((sGlib/python2.7/site-packages/statsmodels/graphics/tests/test_gofplots.pyts  _   %