ó áp7]c@s=ddlmZddlZddlmZddlZddlm Z ddl m Z ddl mZmZmZmZmZddlmZddlmZmZddlZd efd „ƒYZd efd „ƒYZd efd„ƒYZd„Zd„Zejj d„ƒZ!ejj d„ƒZ"d„Z#dS(iÿÿÿÿ(tPY37N(t assert_equal(ttesting(tsparse(t dummy_sparsetGroupingtGrouptcombine_indicest group_sums(t categorical(tgrunfeldtanes96t CheckGroupingcBsžeZejjd„ƒZd„Zd„Zejjde ddƒd„ƒZ ejjde ddƒd„ƒZ d„Z ejjd „ƒZ d „ZRS( cCs|jj|jjƒdS(N(tgroupingtreindextindex(tself((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyt test_reindexscCs3|jjddƒtjj|jj|jƒdS(Ntleveli(R tcount_categoriestnpRRtcountstexpected_counts(R((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_count_categoriesscCsë|jj|jƒ\}}|jjƒ}tj||ƒtjjt |t j ƒƒtjj|j |jj ƒ ƒt|dƒr¥tjj|j |jƒ ƒn|jj|jjƒ\}}tjj||jƒtjjt |tjƒƒ|j|jjd}|jj|ƒ\}}|jƒ}tj||ƒtjjt |t jƒƒt|dƒrˆtjj|j |ƒ ƒn|j}|jj|ƒ\}}|jƒj}tjj||ƒtjjt |tjƒƒdS(Ntequalsi(R tsorttdatat sort_indexttmtassert_frame_equalRRtassert_t isinstancetpdt DataFrameRRthasattrtvaluestassert_array_equaltndarraytcolumnstassert_series_equaltSeries(Rt sorted_dataRtexpected_sorted_datatseriestarray((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyt test_sorts.      t conditiontreasons/Unexplained conversion to complex on Python 3.7cCs|jjj}|jj|jd„ddƒ}|jjƒj|dƒ}|jd„ƒ|jj}t j j ||j ƒt |ƒdkr|jj|jd„ddƒ}|jjƒj|dƒ}|jd„ƒ|jj}t j j ||j ƒndS(NcSs |jƒS(N(tmean(tx((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pytFtRicSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2IR3icSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2OR3cSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2RR3(RRtnamesR ttransform_dataframet reset_indextgroupbytapplyR&RRR$R#tlen(RR4ttransformed_dataframetgroupedtexpected((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_transform_dataframe@s"        cCs |jjj}|jj|jjd„ddƒ}|jjƒj|dƒ}|jd„ƒ|jj }t j j ||jƒt |ƒdkr|jj|jjd„ddƒ}|jjƒj|dƒ}|jd„ƒ|jj }t j j ||jƒndS(NcSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2\R3RicSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2_R3icSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2fR3cSs |jƒS(N(R0(R1((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2hR3(RRR4R ttransform_arrayR#R6R7R8R&RRR$R9(RR4ttransformed_arrayR;R<((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_transform_arrayVs"         cCs |jjj}|jj|jjd„ddƒ}|jjƒj|dƒjƒ|jj }t j j ||jddddƒt |ƒdkr|jj|jjd „ddƒ}|jjƒj|dƒjƒ|jj }t j j ||jddddƒndS( NcSs |jdƒS(Ni(R0(R1tidx((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2qR3Ritrtolgê-™—q=tatolgÙ}ÚõÐò¾:icSs |jdƒS(Ni(R0(R1RA((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR2{R3(RRR4R ttransform_slicesR#R6R7R0R&RRtassert_allcloseR9(RR4ttransformed_slicesR<((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_transform_slicesms$           cCs?|jjƒt|jjƒdkr;|jjddƒndS(NiR(R tdummies_groupsR9t group_names(R((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_dummies_groupsƒs cCsÈ|j}|jjƒt|jjdƒjdtƒ}tj j |jj j ƒ|ƒt |jjƒdkrÄ|jjddƒt|jjdƒjdtƒ}tj j |jj j ƒ|ƒndS(NitdropiR(RR RR Rtget_level_valuesR#tTrueRRRt_dummiesttoarrayR9RI(RRR<((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_dummy_sparse‹s    (t__name__t __module__tpytesttmarktsmokeRRR-txfailRR=R@RGRJRP(((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR s  % tTestMultiIndexGroupingcBseZed„ƒZRS(cCsYtjƒj}|jddgƒ}|j}t|ƒ|_||_dgd|_dS(Ntfirmtyearii (R t load_pandasRt set_indexRRR R(tclst grun_datatmulti_index_datatmulti_index_panel((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyt setup_class›s   (RQRRt classmethodR`(((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyRWšstTestIndexGroupingcBseZed„ƒZRS(cCsVtjƒj}|jdgƒ}|j}t|ƒ|_||_dgd|_dS(NRXii (R RZRR[RRR R(R\R]t index_datat index_group((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyR`§s   (RQRRRaR`(((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyRb¦sc ßCsÃtjƒj}|jddgƒj}t|ƒ}tjj|j ddgƒtjj|j d"ƒtjj|j ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d d ddddddddddddddddddddddddddddddddddddddddgÜdd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d ddddddddddd ddddd d dd d dddddddddgÜgƒt|dddgƒ}tjj|j ddgƒt jƒj}|jdddgƒj}t|ƒ}tjj|j dddgƒtjj|j d#ƒ|j ƒ}t|dddgƒ}tjj|j ddgƒtjj|j d$ƒ|j ƒ}t|ddddgƒ}tjj|j dddgƒtjj|j d%ƒ|jdƒ}t|ƒ}tjj|j dgƒtjj|j d&ƒ|jdƒj ƒ}t|ƒ}tjj|j dgƒtjj|j d'ƒt|ƒ}tjj|j dd d!gƒdS((NRXRYi iiiiiiii i iiii i iiiiiiR4tfirmsteductincometTVnewsiiÜtgroup0tgroup1tgroup2(i i(iii(i i(iii(iÜiÜ(R RZRR[RRRRR$RIt index_shapetlabelsR ttolistRL( R]R_R t anes_datatmulti_index_groupst list_panelt list_groupsRdt list_group((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyt test_init_api²sz 3333333333333333333333333:       c Cs|tjjdƒtjjddddƒ}t|dtƒ\}}}}t|ddd d dtƒ\}}}}tjd d gƒ|dd…df}tjd dgƒ|dd…df}t||fddd d dtƒ\}}}}t||fddd ddtƒ\}}}}tj|ƒ|}tjddddddddddg ddƒ}t||ƒdS(Ni iitsizei t return_labelstprefixsg1,g2=tsept,tsector0tsector1tregion0tregion1issector,region=R3t.ssector1.region0ssector0.region1ssector0.region0ssector1.region1tdtypes|U15(i i(RtrandomtseedtrandintRRMR,R( tgroupstuvtuxtutlabelRiRjt group_jointtgroup_joint_expected((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_combine_indicess*))   c Cstjddddddddgƒ}ttjt|ƒddƒjt|ƒddƒ|dtƒjttjt|ƒddƒjt|ƒddƒdd…dd…df|ƒttjt|ƒddƒjt|ƒddƒdd…dd…df|ƒdS(Niiiit use_bincount(RR,RtarangeR9treshapetFalsetT(tg((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_group_sumss '8 Rc Cs†tjddddddddgƒ}tjt|ƒdƒjt|ƒdddƒ}t|ƒ}|j|j|ƒ|jƒdS(NiiiitordertF( RR,RŒR9RRt group_intRRm(RR1tmygroup((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyttest_group_class)s '1  c Csztjdddddddgƒ}t|ƒ}t|tjjƒsKt‚|jƒ}tj dddgdddgdddgdddgdddgdddgdddggdtj ƒ}t ||ƒtjddddddgƒ}t|ƒ}|jƒ}tj dddgdddgdddgdddgdddgdddggdtj ƒ}t ||ƒdS(NiiiR( RR,RRRtcsrt csr_matrixtAssertionErrorttodensetmatrixtint8R(RtinditresultR<((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyRP6s,$        !      ($tstatsmodels.compat.pythonRtnumpyRt numpy.testingRtpandasR t pandas.utilRRtscipyRtstatsmodels.tools.grouputilsRRRRRtstatsmodels.tools.toolsR tstatsmodels.datasetsR R RStobjectR RWRbRtRŠRTRUR‘R–RP(((sFlib/python2.7/site-packages/statsmodels/tools/tests/test_grouputils.pyts"  ( ‰  R