ó šxŠ\c @sXdZddlmZddlmZddlZddlmZddl Z ddl Z ddl Z ddl mZmZmZmZmZmZmZddlmZddljjZddlmZd „Zd „Zd „Zd „Z d „Z!d„Z"d„Z#d„Z$d„Z%d„Z&d„Z'd„Z(d„Z)d„Z*d„Z+d„Z,d„Z-d„Z.e j/j0de1e jiid*d+6d,d-6d6ƒfe2e jiidddgd.6dddgd/6d6ƒfd „e jiid0d16d2d36d6ƒfd!„e jiidddgd46dddgd56d6ƒfgƒd"„ƒZ3e j/j0de1e jd6d7gd#ddgd$dƒfe2e jdddgdddggd#ddgd$dƒfd%„e jd8d9gd#ddgd$dƒfd&„e jdddgdddggd#ddgd$dƒfgƒd'„ƒZ4d(„Z5d)„Z6dS(:s test all other .agg behavior iÿÿÿÿ(tprint_function(t OrderedDictN(tpartial(t DataFrametIndext MultiIndext PeriodIndextSeriest date_ranget period_range(tSpecificationError(t pprint_thingcCsºtitjjdƒd6tjjdƒd6dddddgd6dddddgd 6ƒ}|jdƒ}d „}|j|gƒ}ddg|_|j|ƒ}tj||ƒdS( Nitdata1tdata2tatbtkey1tonettwotkey2cSs|jƒ|jƒS(N(tmaxtmin(tarr((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt peak_to_peak%s( Rtnptrandomtrandntgroupbytaggtcolumnsttmtassert_frame_equal(tdftgroupedRtexpectedtresult((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt test_agg_apis cCsÙdddgdddgdddgg}tig|D]}|d ^q7d 6g|D]}|d^qUd 6g|D]}|d^qsd 6ƒ}g|D]F}|d |drÍtjj|dd ƒjƒnd|dg^q—}tig|D]}|d ^qðd 6g|D]}|d^qd 6g|D]}|d^q,d 6ƒ}|d |d jƒ|d<|jd ƒjt jƒ}|d |d jƒ|d<|jd ƒjt jƒ}t |ƒt |ƒksÕt ‚dS(Nis 2012-01-01gð?is 2012-01-02g@ig@itkeytdatetvalues%Y-%m-%dtweights( tNoneRtdttdatetimetstrptimeR&tsumRt aggregateRtlentAssertionError(tdatatxtdf1trowtdf2tgb1tgb2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_datetimes_mixed.s   $$P$$cCstdddddƒ}ttjjddƒd|ƒ}|jdd ƒjƒ}t|jt ƒsit ‚td d dd ddƒ}t tjj t |ƒƒd|ƒ}t tjj t |ƒƒd|ƒ}d |fd|fg}tjt|ƒƒ}|j|jjƒ}t|ƒdS(Ns2012-1-1tfreqtMtperiodsiitindextlevelitstarts1999-01its1ts2(R RRRRRR-t isinstanceR<RR0RtrandR/t from_dictRtmonthtlist(tprngR trsR<R?R@tseriesR!((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_period_indexJs!$$c Cs¡tiddddddddgd6tdddd d ƒd 6ƒ}d|jd d ddgd f<|jd dddgjdƒ}|jdƒ}tj|jƒ|ƒtj|j dƒ|ƒtj|j idd 6ƒ|ƒtj |j jƒ|d ƒtj |j j dƒ|d ƒ|jd dddgjdƒ}|jdƒ}tj|j ƒ|ƒtj|j dƒ|ƒtj|j idd 6ƒ|ƒtj |j j ƒ|d ƒtj |j j dƒ|d ƒt jddddgdttdƒddƒdd ƒ}tj |j j tƒ|ƒtj |j jƒ|ƒt jd d d dgdttdƒddƒdd ƒ}tj |j jƒ|ƒdS(NtAtBtCtDtclasss1/1/2011R;iR9tHttimeiiiiiiitfirstitlastR<tABCDtname(RRR)tloct set_indexRRRRQRtassert_series_equalRPRRtpdRRRER/tsizetcount(R texpR!((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt*test_agg_dict_parameter_cast_result_dtypesZs4%"  "    cCs¤gtdƒD]}tjd|ddƒ^q }tdƒ}tji|d6|d6ƒ}|jdƒdjtƒ}|jdƒdj ƒ}t j ||ƒdS(Ni ißit aaabbbbbbccdtXtY( trangeR*R+RERXRRRR/RZRRW(R2tutvR R#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_cast_results_dtypes€s 2 cCs:tidddddgd6dddddgd6dddddgd6ƒ}tidd ddgd6d ddddgƒ}d|j_|jdƒdgjƒ}tj||ƒtidd ddgd6dd ddgd6d ddddgƒ}d|j_|jdƒddgjƒ}tj||ƒdS( NiiiiiRRtcg@R<(RR<RTRtmeanRR(R R"R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_aggregate_float64_no_int64Œs1 / c Csytiddddddddgd6ddddddddgd6tjjdƒdd 6tjdƒd 6ƒ}|jddgƒ}|d jƒ}|d jƒ}|d jƒ}|d jƒ}|d jd d gƒ}t j ||gd dƒ}d d g|_ t j ||dtƒ|jtjtjgƒ}t j ||||gd dƒ}tjd d gd d ggƒ|_ t j ||dtƒ|d d gjtjtjgƒ}t j ||||gd dƒ}tjd d gd d ggƒ|_ t j ||dtƒ|jid d 6d d 6ƒ}t j ||gd dƒ}t j ||dtƒ|jid d gd 6d d gd 6ƒ}t j ||||gd dƒ}tjd d gd d ggƒ|_ t jtdtƒ2|d d gjitjd6tjd6ƒ}WdQXt j ||||gd dƒ}tjddgd d ggƒ|_ t j ||dtƒdS(NtfootbarRJRRRKigð?RLRMR-Retaxisit check_liketcheck_stackleveltrtr2(RRRRtarangeRReR-RRXtconcatRRRtTrueRt from_producttassert_produces_warningt FutureWarningtFalse(R R!tc_meantc_sumtd_meantd_sumR#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_aggregate_api_consistency sP !%!!!c CsXtjidddddgd6tdƒd6tdƒd6ƒ}tjtdtƒd}|jdƒjiidd gd 6d6id d gd 6d6ƒdt |dj ƒks´t ‚WdQXtjtdtƒ,|jdƒddgjid d6ƒWdQXtjtƒF}|jdƒj jid d 6ƒdt |dj ƒksNt ‚WdQXdS(NiiRJiRKRLRkR-RRgRZRRhsusing a dict with renamingitmas(using a dict on a Series for aggregation( RXRR`RRrRsRtRRtstrtmessageR0RK(R tw((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt"test_agg_dict_renaming_deprecationÕs   &%- c Cs´tiddddddddgd6ddddddddgd6tjjdƒdd 6tjdƒd 6ƒ}|jddgƒ}tj|d jƒ|d j ƒgd d ƒ}t j ddgƒ|_ t jtdtƒ%|d jid dgd 6ƒ}WdQXt j||dtƒtj|d jƒ|d j ƒgd d ƒ}d d g|_ t jtdtƒ&|d jid d 6dd 6ƒ}WdQXt j||dtƒdS(NRgRhRJRRRKigð?RLRMRiiR-tstdRkRj(RLR-(RLR(RRRRRnRRXRoR-RRt from_tuplesRRRrRsRtRRRp(R tgR"R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_compatés$ / &/'c Csitiddddddddgd6ddddddddgd6tjjdƒdd 6tjdƒd 6ƒ}|jddgƒ}d }tjtd |ƒ:|j iid dgd 6d6id dgd 6d6ƒWdQXt j t dt ƒ<|jiid dgd6d 6id dgd6d 6ƒ}WdQXtj|d jƒ|d jƒ|d jƒ|d jƒgddƒ}tjjddddgƒ|_t j||dtƒt j t dt ƒ,|d jitjd6tjd6ƒ}WdQX|jdid d6ƒ}t j t dt ƒ,|d jitjd 6tjd6ƒ}WdQXt j||dtƒdS(NRgRhRJRRRKigð?RLRMs;cannot perform renaming for r[1-2] with a nested dictionarytmatchReR-tr1RmRkRtratrbRiiRjtresult1tresult2R(R…Re(R…R(R†Re(R†R(RRRRRnRtpytesttraisesR R.RRrRsRtRRXRoReRRR€RRRpR-trename(R RtmsgR#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_nested_dictss6      --cCs]ttjjdddƒƒ}d„}tjtddƒ|jdƒj|ƒWdQXdS( Ni RYicSs*tdƒt|jƒƒtdƒ‚dS(Ns(----------------------------------------ttest(R t to_stringt TypeError(R ((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pytraiseException,s RƒRŽi(ii ( RRRtrandintR‰RŠRRR(R R‘((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt%test_agg_item_by_item_raise_typeerror)s cCsYtjƒ}|jd„d„gƒ}|jtjƒ}|jƒ}tj||ƒdS(NcSs|jS(N(tyear(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt7scSs|jS(N(RD(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•7s(RtmakeTimeSeriesRRRR-RW(ttsR!R#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_series_agg_multikey5s   c Cstidddddddddddg d6dddddddddddg d6dddddddddddg d 6tjjd ƒd 6tjjd ƒd 6tjjd ƒd 6ƒ}d„}|jddgƒj|ƒ}|jddgƒjd„ƒ}tj||ƒdS(NRgRhRJRRRKtdulltshinyRLi RMtEtFcSs"t|jjƒdkst‚dS(NiRg(R/tvaluestbaseR0(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pytbadJscSsdS(NRg((R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•Os(RRRRRRRR(R1RŸR#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt!test_series_agg_multi_pure_python>s !c Csád„}tiddddgd6ddd d gd 6tjd ddƒtjd ddƒtjd dd ƒtjd dd ƒgd6ƒ}|jdƒ}|j|gƒ}|jjd|_|j|ƒ}tj||ƒdS(NcSs9ytj|jƒddƒSWntk r4tjSXdS(Ntqi(Rt percentiletdropnat Exceptiontnan(R((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pytP1Vs iiiitcol1i iiitcol2iÝi R&i( RR*R&RRRtlevelsRR(R¦R RR"R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_consistencySs $1cCsÇtiddgd6ddgd6ƒjtjƒ}dtfd„ƒY}ttjd „d „ttƒ|ƒg}|jdƒjtƒ}x6|D].}|jdƒj|ƒ}t j ||ƒq‘WdS( NiiRgiiRhtfn_classcBseZd„ZRS(cSs t|ƒS(N(R-(tselfR2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyt__call__ps(t__name__t __module__R­(((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR«nscSs t|ƒS(N(R-(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•uscSs |jƒS(N(R-(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•vs( RtastypeRtint64tobjectR-RRRRR(R R«tequiv_callablesR"tecallR#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_callablesjs2   cCs tjdtjdddgƒgdtjdddgƒgdtjdddgƒggd d d gƒ}|jd ƒjtƒ}tjdd d gƒgtjdddgƒgg}tjddgdd ƒ}d g}tj|d|d |ƒ}tj ||ƒdS(Nii iii(i2i<iRtcategoryt arraydataiFiZRTR<( RXRRtarrayRRR-RRR(R R#t expected_datatexpected_indextexpected_columnR"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_over_numpy_arrays€s!!6   c Csdtjdddƒ}tjidd6gtdƒD]}|tjd|ƒ^q2d6ƒ}|jdƒdjtj ƒj d }|jdƒdjd „ƒj d }|jdƒdj ƒj d }||ksÙt ‚||ksët ‚||ksýt ‚gtdd ƒD]}tjd |ddƒ^q }tjiddgd d6|d6ƒ}|jdƒ}|dj d }||j d ƒdj d ksŸt ‚||j dƒdj d ksÅt ‚||jƒdj d ksèt ‚||jd„ƒd ks t ‚|dj d }||jƒdj d ks>t ‚||jd„ƒd ks`t ‚dS(Ns2016-01-01 12:00:00ttzs US/PacificiRi tminutesRicSs tj|ƒS(N(RR(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•šsis2016-01-0%d 12:00:00iRJRKcSs |jdS(Ni(tiloc(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ªscSs |jdS(Niÿÿÿÿ(R¿(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•®s(RXt TimestampRR`R*t timedeltaRRRRR¿R0tnththeadRQtapplyRR( R—tnnR R‡Rˆtresult3titdatesR!((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_timezone_round_trip’s,3&& 2'&&#"#cCsµtjddgddgddggdtƒ}|d}tjd d d gdtjƒ}tjid d dgd6d|ƒ}d|j_|jdƒj ƒ}t j ||ƒdS(NiiiiiitdtypeIÿÿÿÿÿÿÿllllllR<i( RXRR²RRtuint64R<RTRR-RR(R R<R"R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_sum_uint64_overflow±s0        sstructure, expectediiiRLcCs t|ƒS(N(ttuple(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ÈscCs t|ƒS(N(RE(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ÊscCsžtjiddddddgd6ddddddgd6ddddddgd6ƒ}|jddgƒj|ƒ}ddg|j_tj||ƒdS(NiiRJiRKRL(RXRRR.R<tnamesRR(t structureR"R R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_structs_dataframeÅs "R<RTcCs t|ƒS(N(RÍ(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ÚscCs t|ƒS(N(RE(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ÜscCs–tjiddddddgd6ddddddgd6ddddddgd6ƒ}|jdƒdj|ƒ}d|j_tj||ƒdS(NiiRJiRKRL(RXRRR.R<RTRRW(RÏR"R R#((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_structs_series×s " c Csàdddg}tjitjdddgd|ƒd6dddgd 6ƒ}|jdd |ƒjjtjƒ}tjddd gd tj dddgd|d dƒd d ƒ}|rÌ||d k}nt j ||ƒdS(NRRRdt categoriesRJiiiRKtobservediR<RT( RXRt CategoricalRRKRRtnansumRtCategoricalIndexRRW(RÓRÒR R#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_category_nansumês $  cCstjigtdƒD]}t|ƒ^qd6gtdƒD]}t|ƒ^q<d6ƒ}|jddtdtƒ}|jid„d6ƒ}tjigtdƒD]}t|ƒ^q¥d6gtdƒD]}t|ƒg^qËd6ƒ}tj||ƒdS(NiRJRKtas_indextsortcSs t|ƒS(N(RE(R2((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyR•ÿs( RXRR`R{RRtRRR(R2R R!R#R"((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pyttest_agg_list_like_funcús/,//(iii(ii(iii(ii(ii(ii(iii(ii(iii(ii(ii(ii(iii(iii(iii(iii(7t__doc__t __future__Rt collectionsRR+R*t functoolsRtnumpyRR‰tpandasRXRRRRRRR tpandas.core.groupby.groupbyR tpandas.util.testingtutilttestingRtpandas.io.formats.printingR R$R8RIR\RcRfRyR~R‚RR“R˜R RªRµR¼RÉRÌtmarkt parametrizeRÍRERÐRÑR×RÚ(((sHlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_other.pytsZ    4    &  5   &       '9", *<*'