p7]c@sddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z ejdddd gddd d gdd d d gdddd gddddgddddgdddd gdddd gdd ddgddddgdddd gdd!d"d#gdd$d%d gg d&d'd(d)d*gZd+Zd,Zejjd-Zd.Zd/ZdS(0iN(tassert_almost_equalt assert_raises(tMANOVA(tMultivariateTestResults(t add_constants Minas Graesg%C@g(\@gHzG?gd;O@goʡ?gQ@g+?g-@gX9v@gV-@g@gS㥛?gQ@g^I +@gL7A`?s Matto Grossog\(\@g;On@gS㥛@gNbX9@g}?5^I?gS@gx@s Santa CruzgbX9@g r?g@g +@g+?gE@g+@tcolumnstLoctBasaltOcctMaxcCstjddt}|j}t|ddjd$ddd t|ddjd%d dd t|ddjd&d dd t|ddjd'ddd t|ddjd(dddt|ddjd)dddt|ddjd*dddt|ddjd+dddt|ddjd,dddt|ddjd-dddt|ddjd.dddt|ddjd/dddt|ddjd0dddt|ddjd1dddt|ddjd2dddt|ddjd3dddt|ddjd4d ddt|ddjd5d!ddt|ddjd6d"ddt|ddjd7d#dddS(8NsBasal + Occ + Max ~ LoctdataRtstats Wilks' lambdatValueg%R>?tdecimalisPillai's traceg!?sHotelling-Lawley traceg/z?sRoy's greatest rootg3^a?sF Valuegp= ף?igQ?g?gQ?sNum DFiisDen DFiigiq."@ii sPr > FgvOjM?g48E?gn?g#J{/L?(s Wilks' lambdaR (sPillai's traceR (sHotelling-Lawley traceR (sRoy's greatest rootR (s Wilks' lambdasF Value(sPillai's tracesF Value(sHotelling-Lawley tracesF Value(sRoy's greatest rootsF Value(s Wilks' lambdasNum DF(sPillai's tracesNum DF(sHotelling-Lawley tracesNum DF(sRoy's greatest rootsNum DF(s Wilks' lambdasDen DF(sPillai's tracesDen DF(sHotelling-Lawley tracesDen DF(sRoy's greatest rootsDen DF(s Wilks' lambdasPr > F(sPillai's tracesPr > F(sHotelling-Lawley tracesPr > F(sRoy's greatest rootsPr > F(Rt from_formulatXtmv_testRtloc(tmodtr((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyttest_manova_sas_examplesT                    cCsQttjtdgdt}tdddg}t||}tjd)}d|d*d%dd"t |dd j d?d&dd"t |dd j d@d'dd"t |dd j dAd(dd"dS(BNRt drop_firstRRR iiiit InterceptR s Wilks' lambdaR g%R>?R isPillai's traceg!?sHotelling-Lawley traceg/z?sRoy's greatest rootg3^a?sF Valuegp= ף?gQ?g?gQ?sNum DFisDen DFiigiq."@ii sPr > FgvOjM?g48E?gn?g#J{/L?(ii(ii(ii(ii(ii(s Wilks' lambdaR (sPillai's traceR (sHotelling-Lawley traceR (sRoy's greatest rootR (s Wilks' lambdasF Value(sPillai's tracesF Value(sHotelling-Lawley tracesF Value(sRoy's greatest rootsF Value(s Wilks' lambdasNum DF(sPillai's tracesNum DF(sHotelling-Lawley tracesNum DF(sRoy's greatest rootsNum DF(s Wilks' lambdasDen DF(sPillai's tracesDen DF(sHotelling-Lawley tracesDen DF(sRoy's greatest rootsDen DF(s Wilks' lambdasPr > F(sPillai's tracesPr > F(sHotelling-Lawley tracesPr > F(sRoy's greatest rootsPr > F( Rtpdt get_dummiesRtTrueRtnptzerosRRR(texogtendogRt interceptRt hypothesesR((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyttest_manova_no_formulaOsl"                    cCsittjtdgdt}tdddg}t||}|j}t|tset dS(NRRRRR ( RRRRRRRt isinstanceRtAssertionError(RRRR((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyt$test_manova_no_formula_no_hypothesiss " cCs"tjddt}dtjdddggdfg}|j|dtjddggdfg}tt|j|dtjdddggtjdgdgdggfg}|j|dtjdddggtjdgdggfg}tt|j|dS(NsBasal + Occ + Max ~ LocR ttesti( RRRRtarraytNoneRRt ValueError(Rt hypothesis((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyt!test_manova_test_input_validations' $B <cCstttjdtdS(Ns Basal ~ Loc(RR'RRR(((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyttest_endog_1D_arrays(tnumpyRtpandasRtpytestt numpy.testingRRtstatsmodels.multivariate.manovaRt)statsmodels.multivariate.multivariate_olsRtstatsmodels.toolsRt DataFrameRRR tmarktsmokeR#R)R*(((sIlib/python2.7/site-packages/statsmodels/multivariate/tests/test_manova.pyts2    0 :