ó šxŠ\c @sbdZddlmZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlmZddljjZejjdddd d d d ejd dejjdƒgƒdddg ƒd„ƒZd„Zd„Zd„Zd„Zd„Zd„Zejjdd ej fd ej!fd ej"fdej#fdej$fdej%fdej&fdd„fdd„fg ƒd „ƒZ'ejjdd ej fd d!„fd d"„fdej%fdej&fgƒd#„ƒZ(d$„Z)ejjd%ddddgƒejjd&e d'ƒe d(ƒgƒd)„ƒƒZ*dS(*s test cython .agg behavior iÿÿÿÿ(tprint_functionN(t DataFrametIndextNaTtSeriest Timedeltat Timestampt bdate_range(t DataErrortop_nametcounttsumtstdtvartsemtmeantmediantmarkssignore::RuntimeWarningtprodtmintmaxc sÈiddddddddddtjtjg d6ddgdd6tjjdƒd6}t|ƒ}tj|jd d d …df<‡fd †‰|jdgd dƒjdƒ}‡fd †|Dƒ}ti|d6ƒ}d|j_ ˆ|ƒ}t j ||ƒ|jddgƒ}i}x:|D]2\\}}} ˆ| dƒ|j |iƒ|+staxiscs)i|]\}}ˆ|dƒ|“qS(R((t.0tcattgroup(top(sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pys /s tdropnaR R(R R(tnptnantrandomtrandnRtloctdroptgroupbytindextnamettmtassert_frame_equalt setdefaulttTtstacktFalsetnamestassert_series_equal( R tdatatdftgroupedtexptresulttexpdtcat1tcat2R((RR sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cythonized_aggerss,4 !  $  cCs“titjjdddƒd6tjjdddƒjdƒd6ƒ}|jdƒdjƒ}|jdƒdjtjƒ}tj ||ƒdS(Niii2taitbooltb( RR!R#trandinttastypeR'RtaggR*R1(tframeR6texpected((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_agg_booleanCs (cCsãtitjjdddƒd6ddgdd6ƒ}d }tjtd |ƒ|jdƒdjƒWdQXtitjjdddƒd6ddgdd6ƒ}tjtd |ƒ#|dgj|dƒjƒWdQXdS( Niii2R;tfootbariR=sNo numeric types to aggregatetmatch( RR!R#R>tpytesttraisesRR'R(RAtmsg((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_agg_nothing_to_aggLscCsŽtitjjdddƒd6ddgdd6tjd d dd d ƒd 6ƒ}d}tjtd|ƒ|j dƒj j ƒWdQXdS(Niii2R;RDREiR=tnowtperiodstfreqR-tdatessNo numeric types to aggregateRF( RR!R#R>tpdt date_rangeRGRHRR'RNR(RARI((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyt)test_cython_agg_nothing_to_agg_with_datesZs "cCs tidddgd6dddgd6ƒ}|jdd d d ƒjƒ|jdd d d ƒjƒ|jdd d d ƒjƒ|jdd d d ƒjƒdS( NiiiRiitytleveliRtcolumns(RR'R(R3((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_agg_frame_columnscs ,c Csýtiddddddddgd6ddddddddgd6tjjdƒd 6tjjdƒd 6ƒ}|jdƒdjd „ƒ}tid d6d d6d d6id d6d d6d d6gdtddgddƒddƒ}tj ||ƒdS(NRDRERtonettwotthreeRiRtDcSs|jƒjƒS(N(t value_countstto_dict(R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRusiiR(R)( RR!R#R$R'R@RRR*R1(R3ttsRB((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_agg_return_dictms" cCs{tdddƒ}tdddddgd d |ƒ}|jd „ƒ}|jƒ}|jtjƒ}tj||ƒdS( Ns1/1/2000RLi2RRRRYtEi R(cSs|jS(N(tmonth(R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRs(RRR'R R@R!R*R1(tdrR\R4tsummedRB((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_fail_agg}s % s op, targoptaddtfirstcCs |jdS(Ni(tiloc(R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRstlastcCs |jdS(Niÿÿÿÿ(Re(R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRscCs€ttjjdƒƒ}tjjddddƒjtƒ}|j|ƒj|ƒ}|j|ƒj |ƒ}t j ||ƒdS(Nièii2tsize( RR!R#R$R>R?tfloatR't_cython_agg_generalR@R*R+(RttargopR3tlabelsR6RB((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest__cython_agg_general‡s $cCs&t|ƒdkrtj|ƒStjS(Ni(tlenR!RR"(R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRscCstj|ddƒS(Ntddofi(R!R (R((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyRžscs¯tjdddgƒ}tdddƒ}|jtj|d|ƒd|ƒ}|j|ƒ}|jtj|d|ƒd|ƒ}|j‡fd†ƒ}tj||ƒdS( Ni i i ii7itobservedcs ˆ|ƒS(N((R(Rj(sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyR¬s( RORtrangeR'tcutRiR@R*R+(RRjRoR3tgrpstgR6RB((RjsIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyttest_cython_agg_empty_buckets›s%%c Cs}tjdddgddgƒ}tdddƒ}|jtj|d|ƒd |ƒjd ƒ}tjdd d dƒ}tjiddd dgd6dtj|dddtƒƒ}|rÒ||j dk}nt j ||ƒ|jtj|d|ƒd |ƒjdƒ}tjiddddgd6dtj|dddtƒƒ}|ri||j dk}nt j ||ƒdS(Ni i i RTR;iiiRoRciRMi$R(R)torderedRii´( RORRpR'RqRitinterval_rangetCategoricalIndextTrueR;R*R+(RoR3RrR6t intervalsRB((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyt$test_cython_agg_empty_buckets_nanops°s&!RR2s2016-10-14 21:00:44.557s17088 days 21:00:44.557cCs‰tiddgd6|tgd6ƒ}tddgddƒ}ti|tgd6d|ƒ}|jdƒj|ƒ}tj||ƒdS(NiiR;R=R)R((RRRR't aggregateR*R+(RR2R3R(RBR6((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyt"test_cython_with_timestamp_and_natÍs &(+t__doc__t __future__RtnumpyR!RGtpandasRORRRRRRRtpandas.core.groupby.groupbyRtpandas.util.testingtutilttestingR*tmarkt parametrizetparamtfilterwarningsR:RCRJRQRUR]RbRRR R RRRRlRtRzR|(((sIlib/python2.7/site-packages/pandas/tests/groupby/aggregate/test_cython.pyts^   4    !