ó šxŠ\c@s‡ddlmZddlZddlZddlZddlmZmZmZm Z m Z m Z ddl m Zd„Zd„Zejjddƒd „ƒZd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„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,dS(#iÿÿÿÿ(tdatetimeN(t DataFrametIndext MultiIndextSeriest bdate_rangetcompat(ttestingc Cs8d}tjtj|ƒddddddgdddggƒ}|jdƒ}|j|jjƒj ƒ}|j|jjƒj d „ƒ}t j ||ƒtjtj|ƒddddddgƒ}tj d d d gd tddƒ}tdddgd|ƒ}|jdƒj d„ƒ}t j||ƒdS(Ns+2011.05.16,00:00,1.40893 2011.05.16,01:00,1.40760 2011.05.16,02:00,1.40750 2011.05.16,03:00,1.40649 2011.05.17,02:00,1.40893 2011.05.17,03:00,1.40760 2011.05.17,04:00,1.40750 2011.05.17,05:00,1.40649 2011.05.18,02:00,1.40893 2011.05.18,03:00,1.40760 2011.05.18,04:00,1.40750 2011.05.18,05:00,1.40649theadertnamestdatettimetvaluet parse_datest date_timecSs |jƒS(N(tidxmax(tx((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt!ss 2011.05.16s 2011.05.17s 2011.05.18tdtypetnames00:00s02:00tindexcSs|d|djƒS(NR R (R(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR-s(tpdtread_csvRtStringIOtNonet set_indextgroupbyRR Rtapplyttmtassert_frame_equalRtobjectRtassert_series_equal(tstdftexpectedtresulttexp_idx((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_issues s"$!' csÓtjidddddgd6dddddgd 6d dd gƒ‰tjˆjd ˆjd gd d d ddgƒ}ˆjgˆjD]}t|ƒ^qŒd d ƒj‡fd†ƒ}tj ||ƒdS(Ntatbtkeygð?g@g@g@g@tdatatcolumnsitaxistkeystfloat64Rcs ˆjdS(Ni(tiloc(R(R!(s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR:s( RRtconcatR.RtdtypeststrRRR(R"RR#((R!s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_trivial1s (treasonsqGH#20066; function passed into apply returns a DataFrame with the same index as the one to create GroupBy object.csÅtjidddddgd6dddddgd 6d dd gƒ‰tjˆˆgd d d ddgƒ}ˆjgˆjD]}t|ƒ^q~d d ƒj‡fd†ƒ}tj||ƒdS(NR&R'R(gð?g@g@g@g@R)R*R+iR,R-RcsˆS(N((R(R!(s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRLs( RRR/RR0R1RRR(R"RR#((R!s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_trivial_fail?s(c Csûd}tjjddd|ƒ}tjjddd|ƒ}ti|d6|d6tjj|ƒd6d d d d g|d d6ƒ}d„}|jddgƒ}|j}|j|jd|j ƒ}|j ƒ}|j ||ƒ\} } | s÷t ‚dS(NièiiÐtsizeiR(tkey2tvalue1tfootbartbaztquxitvalue2cSsdS(Ni((tg((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytf^sR+( tnptrandomtrandintRtrandnRtgroupert _get_splittert _selected_objR+t_get_group_keyst fast_applytAssertionError( tNtlabelstlabels2R!R>R=RCtsplittert group_keystvaluestmutated((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_fast_applyQs !   cCsïtitjjdƒd6ddddddgd6ƒ}|jd„dd ƒ}tj|jƒ|jƒƒtid d ddd gd 6ƒ}|jd |d<|j dƒj ƒj ƒj }|j ddt ƒj ƒj }tj||ƒdS(Nitfoo1tonettwotthreetfoo2cSs|S(N((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRpsR+iiitc1g@tc2tas_index(RR?R@RBRRRtget_dtype_countsRVRtmeant reset_indexRWtFalse(R!R#tresult1tresult2((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_with_mixed_dtypels"c Csìtiddddddgd6ddddddgd6td ƒd 6ƒ}|jdd tƒ}|jdd tƒ}|jdƒj}|jdƒj}td ddd gƒ}tj ||ƒtj ||ƒ|j d„ƒj}|j d„ƒj}t j ddddgƒ}ddddg} t j | dddgƒ} tj || ƒtj ||ƒttdƒƒ} tddgddgdd gddgdd ggd| ƒ}|jd d tƒj d„ƒj} tj | | ƒdS(NR'R&tctitem_idiiituser_idiR RXiicSs |jdƒS(Ni(thead(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRŠscSs |jdƒS(Ni(Rc(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR‹sR tabcdeiRcSs|S(N((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR™s(ii(ii(ii(ii(ii(ii(ii(ii(RtrangeRtTrueR\RcRRRtassert_index_equalRRt from_tuplesRtlist( R!tg_astg_not_astres_ast res_not_astexpt res_as_applytres_not_as_applytexp_not_as_applyttpt exp_as_applytindtres((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_groupby_as_index_apply{s, ?$cCs©|jddgƒ}d„}d„}d„}|j|ƒ}|jjdksWt‚|j|ƒ}|jjdks~t‚|j|ƒ}|jjd ks¥t‚dS( NtAtBcSs|jƒ}d|j_|S(Ntstat(tdescribeRR(tgroupR#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytdesc s  cSs,|jƒ}d|j_|t|ƒ }|S(NRy(RzRRtlen(R{R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytdesc2¥s  cSs6|jƒ}dt|ƒ|j_|t|ƒ }|S(Nsstat_%d(RzR}RR(R{R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytdesc3¬s Ry(RwRxRy(RwRxRy(RwRxN(RRRR RHR(t three_grouptgroupedR|R~RR#R^tresult3((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt test_apply_concat_preserve_namess   cCs‰d„}tdddƒ}ttjjdƒd|ƒ}|jd„ƒ}|j|ƒ}t|tƒsot ‚t j |j |j ƒdS(NcSsQtjddƒtj|ƒ}WdQXti|d6||jƒd6|d6ƒS(NtinvalidtignoreR tdemeanedtlogged(R?terrstatetlogRRZ(tpieceR‡((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR>Ás  s1/1/2000tperiodsidRcSs|jS(N(tmonth(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRËs( RRR?R@RBRRt isinstanceRRHRRgR(R>tdrttsRR#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_series_to_frameÀs cCsB|jddgƒdjtƒ}|jjd dks>t‚dS(NRwRxtCi(RwRx(RRR}RR RH(R!R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt test_apply_series_yield_constantÒs"cCsž|jddgƒjtƒ}t|tƒs3t‚|jdksHt‚|jddgƒddgjtƒ}t|tƒs…t‚|jdksšt‚dS(NRwRxR‘tD(RRR}RRRHRR(R!R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_frame_yield_constant×s (cCsd|jddgƒ}|jtƒ}|jƒd}tj|j|jƒtj|j|jƒdS(NRwRxR‘( RRR}tcountRRgRtassert_numpy_array_equalRN(R!RR#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_frame_to_seriesâs csÌd„}‡fd†}titjjdddƒd6tjjdddƒd6tjjdƒd6ƒ‰ˆjdƒj|ƒ}ˆjdƒdj|ƒ}tj||d t ƒ|j dksÈt ‚dS( NcSs!|jdƒdjƒjƒd S(NRxR‘i(Rtsumt sort_values(R{((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttransëscs3|jˆj|jƒdƒ}|jƒjƒd S(NRxi(RtreindexRR˜R™(R{R(R!(s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttrans2îsiiièRwRxR‘t check_names( RR?R@RARBRRRRR\RRH(RšRœR#Rn((R!s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_frame_concat_seriesês cCsJ|jd„ƒ}|jd„ƒ}|jd„ƒ}tj||ƒdS(NcSs|jS(N(RŒ(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRýscSs|dS(Ni((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRþscSs|dS(Ni((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRÿs(RRt transformRR(RRR#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_transformüscCsk|jd„d„gƒ}d„}|j|ƒ}x1|D])\}}tj|j|||ƒƒq:WdS(NcSs|jS(N(tyear(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRscSs|jS(N(RŒ(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRscSs|jdƒdS(NRwiûÿÿÿ(R™(R{((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR>s(RRRRtloc(ttsframeRR>R#R(R{((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_multikey_corners  c Cs”tidddddddddg d6tjdƒd6ƒ}|jddtƒjd„ƒ}|jd ddd d d gƒ}tj||ƒdS( NiiiR(i R RMcSs|d S(Ni((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRsiiii( RRtlrangeRR\RttakeRR(R!R#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_chunk_views (!!c Csœtiddddddddddg d6ddddddddddg d6tjdƒddd…d6ƒ}|jddgƒ}|jd „ƒdS( NiiRitname2i iÿÿÿÿR cSs|jddtƒS(NR tinplace(R™Rf(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRs(RRR¥RR(R!R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt"test_apply_no_name_column_conflicts +%#cCs´tiddddddgd6tjdddgdƒd6tjddƒd 6ƒ}d „}|jdƒj|ƒ}|jƒ}tjd d d gdƒ|dlib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR>(s (ggà?iR­( RR?ttiletarangeRRtcopyRR(R!R>R#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_typecast_fail"s  cCsótjddddddgddddddggƒ}tiddddddgd6tjdd d gdƒd 6tjdd ƒd 6d |ƒ}d„}|jdƒj|ƒ}|jƒ}tjdddgdƒ|dlib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR><s (ggà?R­( Rt from_arraysRR?R°R±RRR²RR(RR!R>R#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_multiindex_fail5s9  cCs<|jd„ƒjd„ƒ}|d}tj||ƒdS(NcSs|jS(N(R¡(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRJscSs|dS(Ni((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRJsi(RRRR(R£R#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_cornerIs cCs›tiddddgd6ddddgd6d d d d gd 6ƒ}d„}d„}|jdƒj|ƒ}|jdƒj|ƒ}tj||ƒdS(NidiÈi,tid_fieldR&R'R`tcategoryiiiiR cSs2|jddkr|jƒS||jdkSdS(NiiR`(tshapeR²R¸(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytfilt1Ws cSs,|jddkr|S||jdkSdS(NiiR`(R¹R¸(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytfilt2]s(RRRRR(R)RºR»R"R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_without_copyOs  cCsŸd}tjjddd|ƒ}ti|d6tjj|ƒd6ddd d g|d d 6ƒ}|jdƒ}d „}|j|ƒ}d|ks›t‚dS(NièiidR5R(R7R8R9R:R;iR<cSs|dd|d<|S(NR7itvalue3((R=((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR>ssR½(R?R@RARRBRRRH(RIRJR!RR>R#((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_corner_caseshs ! cCsÿtjiddgd6dgdd6ddgd6ƒ}|jdgƒjd „ƒ}tj|jƒ|_|jdgƒjd „ƒ}tj|d|dƒtjid d d gd6dddgd6tjdƒgdd6ƒ}d„}|jdƒj|ƒd}|j }|j |_ tj||ƒd„}tjiddddgd6ddddgd6ddddgd6dd d!d!gd"6ƒ}|j ƒ}tj|j ƒ|_ |jdƒj|ƒj}|jdƒj|ƒj}tj||ƒdS(#NiitNumbers 2017-03-02tDateR8tinftStrcSs |jdS(Ni(R.(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR…scSs |jdS(Ni(R.(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR‡si iiRwt3t4Rxs12:31:22itTcSs|jddgS(NiRx(R.(R=((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytget_BscSsÊtjddddgdtƒ}dt|jƒkr`t||jdkjjdƒ|dlib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt predictions—s!*''tKeyRÊRËRÌts2016-09-19 05:24:33s2016-09-19 23:59:04RÎt23t36Lt36RRÍ(RRRRt to_datetimeRÀRRt TimestampRxRwRR²RÎRÇ(R!R"R#RÆRÑtdf1tdf2((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt)test_apply_numeric_coercion_when_datetime{s4      cCstjidd6gtdƒD]}tjƒ^qd6ƒ}d„}d„}|jddgƒj|ƒ}tjidd 6d dgƒ}d|j_|jddgƒj|ƒ}tjitd ddƒd6dd 6d dgƒ}d|j_t j ||ƒt j ||ƒdS( NiR&i R'cSstjidd6ƒS(NiR`(RR(tbatch((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytfunc_with_no_date´scSs'tjitdddƒd6dd6ƒS(NißiR'iR`(RRR(RÜ((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytfunc_with_date·stbyiR`Riß( RRReRtnowRRRRRR(tnnR!RÝRÞtdfg_no_conversiontdfg_no_conversion_expectedtdfg_conversiontdfg_conversion_expected((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_time_field_bug¬s<    cstiddddddddddddg d6ddddddddddddg d6dd d d d d d d d ddd g d6ddddddddd d d d g d6ƒ}|jddgƒ}|jdddgdtƒ}d„‰|j‡fd†ƒdS(NR&R'tgroup1R`R«tetgroup2gš™™™™™ñ?iiiiiiitweightgffffff@i i i i iR tleveltsortcSs tj||ƒjdƒ}|S(Ni(R?tarraytrepeat(R RêRÐ((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytnoddyÔscsˆ|j|jƒS(N(R Rê(R(Rï(s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRÞs(RRRRfR(R!t df_grouped((Rïs>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt(test_gb_apply_list_of_unequal_len_arraysÈs!+1 cCsptiddddgd6ddddgd6ƒ}d „}|jdƒj|ƒ}tƒ}tj||ƒdS( Niitgroupsiiiit random_varscSsdS(N((R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt test_funcçs(RRRRR(ttest_dfRôR#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_groupby_apply_all_noneás   c Csstiddddgd6ddddgd6ƒ}tiddddgd6ddddgd6ƒ}d„}|jdƒj|ƒ}|jdƒj|ƒ}tjddgddggdddgƒ}tjddgddggdddgƒ}tiddgd6ddgd6d |ƒ}tiddgd6ddgd6d |ƒ}tj||ƒtj||ƒdS( NiiRòiitvarscSs(|jddkrdS|jddgS(Niiiÿÿÿÿ(R¹RR.(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRôôsR R(RRRRR´RRR( ttest_df1ttest_df2RôR]R^tindex1tindex2t expected1t expected2((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_groupby_apply_none_firstïs22 # # c Csªtjtdddgdddgƒƒ}|jdƒ}|jd„ƒ}tjdgddd tjddgdggd ddgƒj dƒƒ}t j ||ƒdS( NR iiR{tfilledtemptycSs||jdkdS(NiR (R (R{((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyR sRRR ( RRtdictRRRRt from_productRtdropRR(R!RòR#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyt%test_groupby_apply_return_empty_chunks*  cCsÊtjidjƒd6dddgd6ddd gd 6ƒ}|jdƒ}|jd „ƒ}tjidd dd dgd6d ddgd 6ƒ}tj||ƒ|jd„ƒ}tj||ƒdS(Nsa a bRwiiiRxiiiR‘cSs||jƒS(N(R˜(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRsg@gš™™™™™Ù?g333333ã?gð?cSs||jƒS(N(R˜(R((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyRs(RRtsplitRRŸRRR(R!R=R#R"((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pyttest_apply_with_mixed_typess<7(-RtnumpyR?tpytesttpandasRRRRRRRt pandas.utilRRR%R2tmarktxfailR4RPR_RvRƒRR’R”R—RžR R¤R§RªR³RµR¶R¼R¾RÛRæRñRöRþRR(((s>lib/python2.7/site-packages/pandas/tests/groupby/test_apply.pytsB   . &    " #           1