ó šxŠ\c@sIdZddlZddlZddlmZddlZddlmZm Z m Z m Z m Z ddl mZddlmZddljjZd„Zd„Zd „Zd „Zd „Zd „Zd „Zejjdd„d„d„ggƒd„ƒZd„Zd„Z d„Z!d„Z"d„Z#d„Z$d„Z%dS(sN test .agg behavior / note that .apply is tested generally in test_groupby.py iÿÿÿÿN(t OrderedDict(t DataFrametIndext MultiIndextSeriestconcat(tSpecificationError(tGroupingcCsM|jd„d„gƒ}|jtjƒ}|jƒ}tj||ƒdS(NcSs|jS(N(tyear(tx((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pytscSs|jS(N(tmonth(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR s(tgroupbytaggtnptmeanttmtassert_frame_equal(ttsframetgroupedtresulttexpected((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_regression1s cCsu|jdƒd}d}tjtd|ƒ|jd„ƒWdQXtjtd|ƒ|jd„ƒWdQXdS(NtAtCsMust produce aggregated valuetmatchcSs |jƒS(N(tdescribe(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR scSs |jd S(Ni(tindex(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR !s(R tpytesttraisest ExceptionR (tdfRtmsg((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_must_aggs cCsl|j}d„}|jj|j|jgƒj|ƒ}|jddgƒjƒd}tj||ƒdS(NcSs |jƒS(N(tsum(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR (sRtBR(RR RR#t aggregateR"Rtassert_series_equal(RtsertftresultsR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_ser_multi_key$s   'cCstidddtjddddgd6dddtjd d d d gd 6d tjddd d!d"d#d$d%gdddgƒƒ}tidddddddtjddddg d6d d dddd dtjd dddg d 6ddddtjdddddtjdg d6ddddtjdddddtjtjg d6ƒ}|jddgƒ}|dd gjƒ}tj||ƒdS(&Niiiiitv1i7iMi!i,i tv2Ri_icitbigtdamptbluetdrytredtwettnamestby1tby2ii iXi (ii_(iic(ii_(iic(R,R-(R.R/(R0R0(R0R1( RRtnanRt from_tuplesR RRR(RRtgR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt$test_groupby_aggregation_mixed_dtype.s$"% ..' !c Csv|j|tjƒ}|jtjks.t‚tgdtjdtjgdtjƒƒ}t j |j ƒ|ƒt j |j tj ƒ|ƒt j |j tj ƒ|dtƒ|j|dtjƒ}td|jdtdtjgdtjƒƒ}t j|j ƒ|dtƒt j|j tj ƒ|dtƒt j|j tj ƒ|jdd…dd…fdtƒdS(NtdtypeRtcheck_index_typeRtcolumnst check_namesi(R RR5R9tfloat64tAssertionErrorRtpdRRR%R"R tapplytFalseRR;tfloatRtiloc(ttsRRtexptexp_df((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_apply_cornerHs"4cCsÕtjƒ}|jd„ƒ}|jjdj}t|jt|ƒƒ|jjd<|jt j ƒ}|j ƒ}tj ||ƒt|jt |ƒƒ|jjd<|jt j ƒ}|j ƒ}tj ||ƒdS(NcSs|jS(N(R(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR bsi( RtmakeTimeDataFrameR tgroupert groupingsRRtlistR RRRttuple(RDRRRIRR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_grouping_is_list_tuple_s " " cCsG|jddgƒ}|jtjƒ}|jƒ}tj||ƒdS(NRR#(R R RRRR(tmframeRRR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_agg_python_multiindexqs t groupbyfunccCs |jƒS(N(tweekday(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR zscCs|jS(N(R (R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR {scCs |jƒS(N(RQ(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR {scCs|j|ƒ}|djdƒ}|djƒ}tj||ƒ|jdƒ}|jƒ}tj||ƒ|jtddgddgddgddggƒƒ}t td|djƒgd|djƒgd|dj ƒgd|dj ƒggƒƒ}tj||ƒdS( NRtstdtvarR#RRtDtsem( R R RRRR%R$RSRRRRRU(RRPRRR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_aggregate_str_funcys    cCs^|jdƒ}d„}|j|ƒ}|jdkjƒ}|jdkjƒ}t|jƒ}tjtj |g|ƒdt dƒdtj ddƒ}t j |jdƒ|ƒtjtj |g|ƒdt dƒdtj ddƒ}t j|jdƒ|ƒd „}tƒj|jƒj|ƒ}t|tƒsBt‚t|ƒd ksZt‚dS( NRcSs|jS(N(tsize(R&((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR ™stfootbarRtBCDR9tnamecSs|jS(N(RW(R&((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pytaggfun©si(R R RR"tlenR;R?RRtarrayRKR=RR%txstassert_almost_equalRt isinstanceR>(RRR\RRXRYtKRE((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_aggregate_item_by_item–s  (( cCs|jddgƒ}d„}|j|ƒ}|jdd…|jdkf}|jddgƒj|ƒ}tj||ƒdS(NRR#cSs)|jtjkrt‚n |jƒSdS(N(R9Rtobjectt TypeErrorR"(R&((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pytfunc´s R(R R$tlocR;RR(t three_groupRRfRt exp_groupedR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyttest_wrap_agg_out±s  "cCsrdtjfdtjfdtjfg}|jdƒdj|ƒ}tdddgƒ}tj|j |ƒdS(NRtmaxtminRR( RRRkRlR R RRtassert_index_equalR;(RtfuncsRtexp_cols((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt*test_agg_multiple_functions_maintain_orderÀs*cCs¤ddg}ddg}|jdƒdj|ƒ}|jdƒdj|ƒ}tj||ƒ|jdƒj|ƒ}|jdƒj|ƒ}tj||ƒdS( NRXRRRRR(RXR(RXR(RRRR(R R RR(RRntex_funcsRR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt-test_multiple_functions_tuples_and_non_tuplesÉs  cCsW|jdƒ}dd„d„g}d}tjtd|ƒ|j|ƒWdQXdS(NRRcSs |jƒS(N(R(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR ÙscSs |jƒS(N(RR(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyR Ùss<Function names must be unique, found multiple named R(R RRRR (RRRnR ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt,test_agg_multiple_functions_too_many_lambdas×s c CsQ|jdƒ}|jtdtjgdtjggƒƒ}|jtdtjgdtjggƒƒ}t||gdddgddƒ}|jd dddƒjd d ddƒ}tdtjtjggdtjtjgggƒ}|j |ƒ}t j ||ƒ|j tdtjgdtjtjgggƒƒ}|j tdtjgdtjtjgggƒƒ}t j ||ƒd „}d „}t j t d tƒStdtjgdtdtjgdtjggƒggƒ}|j |ƒ}WdQXtdtjggd||gggƒ}|j |ƒ}t j ||ƒdS(NRRRTtkeysRRRtaxisiitlevelcSs tj|ƒS(N(RR(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyRXõscSstj|ddƒS(Ntddofi(RRR(R ((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyRYøstcheck_stacklevelRXRY(R R RRRRRRt swaplevelt sort_indexR$RRtassert_produces_warningt FutureWarningRA( RRtexmeantexstdRtdRRXRY((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt'test_more_flexible_frame_multi_functionàs.--$*6!!  *cCs`|jdƒ}tdtddgddggƒgdidd6ggƒ}tjtd tƒ|j|ƒ}WdQXtdtddgddggƒgddggƒ}tjtd tƒ|j|ƒ}WdQXtj||ƒtdtddgddggƒgddgggƒ}tjtd tƒ|j|ƒ}WdQXtj||ƒdS( NRRRXRRYRRRTR"Rx(R RRR{R|RAR$R(RRRRR((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyt test_multi_function_flexible_mixs$$$(&t__doc__tnumpyRRt pandas.compatRtpandasR?RRRRRtpandas.core.baseRtpandas.core.groupby.grouperRtpandas.util.testingtutilttestingRRR!R)R8RGRMROtmarkt parametrizeRVRcRjRpRrRsR€R(((sLlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_aggregate.pyts2   (      !    (