3 Udd?Z2d@dAZ3ej+j,dBe4e5e6e7e8e9e:e;gdCdDZ.peak_to_peakrr) rnprandomrandngroupbyaggcolumnstmassert_frame_equal)dfgroupedrexpectedresultrrr test_agg_apis       r&cCsdddgdddgdddgg}td d |Dd d |Dd d |Dd }dd |D}tdd |Ddd |Ddd |Dd }|d|dj|d<|jdjtj}|d|dj|d<|jdjtj}t|t|kstdS)Nz 2012-01-01g?z 2012-01-02g@g@cSsg|] }|dqS)rr).0xrrr 8sz,test_agg_datetimes_mixed..cSsg|] }|dqS)r'r)r*r+rrrr,9scSsg|] }|dqS)r(r)r*r+rrrr,:s)keydatevaluecSs<g|]4}|d|dr,tjj|ddjnd|dgqS)rr'z%Y-%m-%dNr()dtdatetimestrptimer.)r*rowrrrr,?scSsg|] }|dqS)rr)r*r+rrrr,IscSsg|] }|dqS)r'r)r*r+rrrr,JscSsg|] }|dqS)r(r)r*r+rrrr,Ksr/weightsr.)rsumr aggregaterlenAssertionError)dataZdf1Zdf2Zgb1Zgb2rrrtest_agg_datetimes_mixed3s     r:cCstdddd}ttjjdd|d}|jddj}t|jt sDt td d dd }t tjj t ||d}t tjj t ||d}tj||d }|j|jj}t|dS) Nz2012-1-1Mr))freqperiodsr()indexr)levelz1999-01r )startr=r<)s1s2)r rrrrrr5 isinstancer>rr8rZrandr7 from_dictmonthlist)Zprngr"rsr>rArBr#rrrtest_agg_period_indexXsrHc Cstddddddddgtddddd }d|jd d d d gdf<|jd dddgjd}|jd}tj|j|tj|jd|tj|jddi|tj |j j|dtj |j jd|d|jd dddgjd}|jd}tj|j |tj|jd|tj|jddi|tj |j j |dtj |j jd|dt j d d d d gttddddd}tj |j jt|tj |j j|t j d d d d gttddddd}tj |j j|dS)NABCDz1/1/2011H)r=r<)classtimerr'r(r rPr)rOfirstlastABCD)name)r>rW)rrlocZ set_indexrr r!rSrassert_series_equalrPrUpdrrrFr7sizecount)r"expr#rrr*test_agg_dict_parameter_cast_result_dtypesgs.  $$r^cCs`ddtdD}td}tj||d}|jddjt}|jddj}tj ||dS)NcSsg|]}tjd|ddqS)ir')r0r1)r*r+rrrr,sz0test_agg_cast_results_dtypes.. Z aaabbbbbbccd)XYr`ra) rangerFrZrrrr7r\r rY)uvr"r%r$rrrtest_agg_cast_results_dtypess recCstdddddgdddddgdddddgd}tdddddgiddddgd }d |j_|jd dgj}tj||tddddgddddgd ddddgd }d |j_|jd dd gj}tj||dS) Nr'r(r)rQr )r rcr g@)r>r)r rfrf)rr>rWrmeanr r!)r"r$r%rrrtest_aggregate_float64_no_int64s.  *rhc Cstddddddddgddddddddgtjjddtjdd}|jdd g}|d j}|d j}|d j}|d j}|d jd d g}t j ||gdd}d d g|_ t j ||dd|jtjtjg}t j ||||gdd}tjd d gd d gg|_ t j ||dd|d d gjtjtjg}t j ||||gdd}tjd d gd d gg|_ t j ||dd|jd d d}t j ||gdd}t j ||dd|jd d gd d gd}t j ||||gdd}tjd d gd d gg|_ d}tjt|d"|d d gjtjtjdWdQRXdS)NfoobarrrrMg?)rIrJrKrLrIrJrKrLr5rgr')ZaxisT)Z check_like)rKrLz&Column\(s\) \['r', 'r2'\] do not exist)match)rr2)rrrrarangerrgr5rrZconcatrr r!rZ from_productpytestraisesr ) r"r#Zc_meanZc_sumZd_meanZd_sumr%r$msgrrrtest_aggregate_api_consistencys>     rscCstjdddddgtdtdd}d}tjt|d,|jdjdd d gid d d gidWdQRXd}tjt|d"|jdddgjdd iWdQRXd}tjt|d|jdjjdd iWdQRXdS)Nr'r(r )rIrJrKznested renamer is not supported)rkrIrir5rrjr\r)rJrKz!Column\(s\) \['ma'\] do not existrJrKma) rZrrbrprqr rrrJ)r"rrrrr"test_agg_dict_renaming_deprecations$ &&rucCstddddddddgddddddddgtjjddtjdd}|jdd g}d }tjt|d |d j d ddgiWdQRXtjt|d |d j dddWdQRXdS)NrirjrrrMg?)rIrJrKrLrIrJznested renamer is not supported)rkrLrKr5std)rKrL) rrrrrnrrprqr r)r"grrrrrtest_agg_compats rxc Cs2tddddddddgddddddddgtjjddtjdd}|jdd g}d }tjt|d &|j d d dgidd dgidWdQRXtjt|d &|j dd dgidd dgidWdQRXtjt|d |dj tj tj dWdQRXtjt|d |dj tj tj dWdQRXdS)NrirjrrrMg?)rIrJrKrLrIrJznested renamer is not supported)rkrKrgr5rL)Zr1rmrarvrb)rKrL)result1result2)rLr|) rrrrrnrrprqr r6rr5rg)r"rwrrrrrtest_agg_nested_dictss**"r}c CsJttjjdd d}dd}tjtdd|jdj|WdQRXdS) N )r[cSs tdt|jtddS)Nz(----------------------------------------test)r Z to_string TypeError)r"rrrraiseException!s z=test_agg_item_by_item_raise_typeerror..raiseExceptionr)rkr)rr~) rrrrandintrprqrrr)r"rrrr%test_agg_item_by_item_raise_typeerrorsrcCsBtj}|jddddg}|jtj}|j}tj||dS)NcSs|jS)N)year)r+rrr,sz*test_series_agg_multikey..cSs|jS)N)rE)r+rrrr,s)r ZmakeTimeSeriesrrrr5rY)tsr#r%r$rrrtest_series_agg_multikey*s  rcCstdddddddddddg dddddddddddg dddddddddddg tjjdtjjdtjjdd}d d }|jd d gj|}|jd d gjd d}tj||dS)NrirjrrZdullZshiny )rIrJrKrLEFcSst|jjdkstdS)Nrri)r7valuesbaser8)r+rrrbadcsz.test_series_agg_multi_pure_python..badrIrJcSsdS)Nrir)r+rrrrhsz3test_series_agg_multi_pure_python..)rrrrrrr r!)r9rr%r$rrr!test_series_agg_multi_pure_python3sR  rc Csdd}tddddgddd d gtjd ddtjd ddtjd dd tjd dd gd }|jd}|j|g}|jjd|_|j|}tj||dS)NcSstj|jddS)Nr')q)rZ percentileZdropna)r rrrP1osz test_agg_consistency..P1r'r(r)rQr~ir)Zcol1Zcol2r.r.r) rr0r.rrrZlevelsr r!)rr"rwr$r%rrrtest_agg_consistencyls        rcCstddgddgdjtj}Gddd}ttjdd d d tt|g}|jd jt}x(|D] }|jd j|}tj ||qbWdS) Nr'r(r)rQ)rirjc@seZdZddZdS)z$test_agg_callables..fn_classcSst|S)N)r5)selfr+rrr__call__sz-test_agg_callables..fn_class.__call__N)__name__ __module__ __qualname__rrrrrfn_classsrcSst|S)N)r5)r+rrrrsz$test_agg_callables..cSs|jS)N)r5)r+rrrrsri) rZastyperZint64r5rrrr r!)r"rZequiv_callablesr$Zecallr%rrrtest_agg_callabless rcCstjdtjdddggdtjdddggdtjdddgggd d gd }|jd jt}tjdd d ggtjdddggg}tjddgd d}d g}tj|||d}tj ||dS)Nr'r~r(2<r(categoryZ arraydata)rFZ)rW)r>r) rZrrarrayrrr5rr r!)r"r%Z expected_dataZexpected_indexZexpected_columnr$rrrtest_agg_over_numpy_arrayss $ rcCstjdddd}tjddddg|d}|jd}|d jd d }tj|ddd d d }d|j_tj |||d jdd }tjddgd d }d|j_tj |||d jdd }tjtj ddtj ddgd d }d|j_tj ||dS)Nz 2012-01-01rQUTC)r=tzrr')r rr rcSs |jdS)Nr)iloc)r+rrrrsz6test_agg_tzaware_non_datetime_result..r()rWcSs |jdjS)Nr)rr)r+rrrrsicSs|jd|jdS)Nr'r)r)r+rrrrs)days) rZrrrrrr>rWr rYZ Timedelta)Zdtir"gbr%r$rrr$test_agg_tzaware_non_datetime_results   "rcstjdddtjdfddtdDd}|jd d jtjjd }|jd d jd d jd }|jd d jjd }|kst |kst |kst ddtddD}tjd d gd|d}|jd}|djd |j d djd kst |j ddjd ks&t |j djd ksBt |j dd jdks`t |djd|jdjd kst |j dd jdkst dS)Nz2016-01-01 12:00:00z US/Pacific)rr'csg|]}tj|dqS))minutes)r0 timedelta)r*nn)rrrr,sz0test_agg_timezone_round_trip..r~)r rr rrcSs tj|S)N)rr)r+rrrrsz.test_agg_timezone_round_trip..cSs$g|]}tjd|ddddqS)z 2016-01-0dz 12:00:00z US/Pacific)r)rZ Timestamp)r*irrrr,sr r()rIrJrIrJcSs |jdS)Nr)r)r+rrrrscSs |jdS)Nr'r)r)r+rrrrs)rr)rr)rZrrrbrrrrrr8ZnthheadrSapplyrU)r"r{r|Zresult3datesr#r)rrtest_agg_timezone_round_trips(    rcCs|tjddgddgddggtd}|d}tjd d d gtjd}tjdd d dgi|d}d|j_|jdj }t j ||dS)Nr'r(r)rQr rR)Zdtypelllllll)r>r) rZrobjectrrZuint64r>rWrr5r r!)r"r>r$r%rrrtest_sum_uint64_overflows  rzstructure, expectedrKr'r)rQcCst|S)N)tuple)r+rrrr srcCst|S)N)rF)r+rrrrsc Csftjddddddgddddddgddddddgd}|jddgj|}ddg|j_tj||dS)Nr'r)rQ)rIrJrKrIrJ)rZrrr6r>namesr r!) structurer$r"r%rrrtest_agg_structs_dataframes 2 r)r>rWcCst|S)N)r)r+rrrr#scCst|S)N)rF)r+rrrr$sc Csbtjddddddgddddddgddddddgd}|jddj|}d|j_tj||dS)Nr'r)rQ)rIrJrKrIrK)rZrrr6r>rWr rY)rr$r"r%rrrtest_agg_structs_seriess 2rcCsdddg}tjtjdddg|ddddgd}|jd |d jjtj}tjddd gtj dddg|d d d d}|r||d k}t j ||dS)Nr rrf) categoriesr'r(r))rIrJrI)observedr)rrWrJ)r>rW) rZrZ CategoricalrrJrrZnansumrZCategoricalIndexr rY)rrr"r%r$rrrtest_agg_category_nansum2s " rcCstjddtdDddtdDd}|jdddd}|jdd d i}tjd dtdDd dtdDd}tj||dS) NcSsg|] }t|qSr)str)r*r+rrrr,Esz+test_agg_list_like_func..r))rIrJrIF)Zas_indexsortrJcSst|S)N)rF)r+rrrrHsz)test_agg_list_like_func..cSsg|] }t|qSr)r)r*r+rrrr,JscSsg|]}t|gqSr)r)r*r+rrrr,Js)rZrrbrrr r!)r"r#r%r$rrrtest_agg_list_like_funcBs((rcCs~tjddgtjdddtjdddgd}|jdjdd d i}tjtjdddgtjdgdd dgd }tj||dS) Nr'z 2018-01-01r)rz 2018-01-02)tagr.rr.cSs |jdS)Nr')r)errrrZsz/test_agg_lambda_with_timezone..)rW)r>r)rZrrrrrr r!)r"r%r$rrrtest_agg_lambda_with_timezoneOs  rerr_clsc sddlm}m}m}|dd}tjdddddgdddddg||d}tj||d|dg}fdd}|d j|d j|}t j ||d d dS) Nr) DecimalArray make_data to_decimalr r')id1Zid2decimalsr)cst|dkr|jdS)Nr)r7r)r+)rrr weird_func~s z1test_groupby_agg_err_catching..weird_funcrrF)Z check_names) Z$pandas.tests.extension.decimal.arrayrrrrZrrrrr rY) rrrrr9r"r$rr%r)rrtest_groupby_agg_err_catchingcs& r)r'r')r'r'r')r)rQ)r)rQrQ)r'r')r)rQ)r'r')r'r'r')r)rQ)r)rQrQ)r'r')r)rQ)r'r'r')r)rQrQ)r'r'r')r)rQrQ)=__doc__r1r0 functoolsrZnumpyrrpZpandasrZrrrrrrr Zpandas._testingZ_testingr Zpandas.core.baser Zpandas.io.formats.printingr r&r:rHr^rerhrsrurxr}rrrrrrrrrmarkZ parametrizerrFrrrrrNotImplementedError RuntimeErrorKeyError IndexErrorOSError ValueErrorArithmeticErrorAttributeErrorrrrrrsn $   %& /  9$$0&8