U sVcɾ@s<dZddlZddlZddlmZddlZddlZddlZddlm Z ddl m Z ddl Z ddl mZmZmZmZmZmZddlmZddlmZdd Zd d Zd d ZddZddZddZddZddZ ddZ!ej"#dddddddggddZ$d d!Z%ej"#d"d#d$d%d&gd'ifd(d)gd*e&e'e'e'd+fd,d-gd*e&e'e'e'd+fd#d$d%d&gd.d/d'ifd(d)gd*d.e'e'e'd+fd,d-gd*d.d0d0d0d+fgd1d2Z(ej"#d3d#d4d5gd6d7gd8d9ggd'd'd:fd(d;gd4gd*d0fd,d4gd4gd*d0d0d:fgdd?Z*d@dAZ+dBdCZ,dDdEZ-dFdGZ.dHdIZ/dJdKZ0dLdMZ1dNdOZ2ej"#dPdQddRddSddTddUddVdgdWdXZ3ej"#dYdZged[gdZd\fdZd]ged[gd4ggdgdggdZd]gd^fgej"#d_d`dad'dbd0gej"#dcd`dad'dbd0gej"#dddedfdggdhdiZ4ej"#dYdZged[gdZd\fdZd]ged[gd4ggdgdggdZd]gd^fgej"#djdkd[dlgej"#dme5e&e'gej"#dddedfdggdndoZ6dpdqZ7drdsZ8ej"#dmej9ej:gej"#dtdudvdwdxdydzgd{d|Z;d}d~Zde ?dd*e j@dd*ddeAdddggddZBddZCddZDGdddZEGdddZFej"#dddejGfdd[d*gdd4gddgfdddfdddfdd[d*gd[d[gddgfe Hddxe HdejIe Hdddd[d*gddgd[d[gfgddZJddZKddZLej"#dddwgigdwgdgddZMddZNddZOddZPddZQej"#dddddddgddZRddZSej"#dddeSgddZTddZUej"jVddddZWGdddZXdd„ZYej"#ddwdwdĜd[d$gdZdgdĜfddwiddZdgifddwidd[d$gifgddȄZZej"#ddwdxgdwdĜddgfdwdwdxgdĜddgfddwdxgiddgfgddЄZ[dd҄Z\ddԄZ]ddքZ^dd؄Z_ddڄZ`dd܄Zaej"#ddddgdgfdddgdgfgddZbej"#ddyddgfd#ddgfgddZcej"#ddwdxd,gddZdej"#ddwgdydxgd]d#iddzdgddZeej"#dd9dgd9dgd6dggd6d9gdddggdfd9dgd9dgddggd9dgddgdgdfdZd[gdZd[gd]d4gd]d*ggdZd]gd[d4d*ggdfejfdZd[gdZd4gd]d*gd]d*ggdZd]gd[d4gd*gdej"jVdgddZgddZhdS)zN test .agg behavior / note that .apply is tested generally in test_groupby.py N)partial)SpecificationError)is_integer_dtype) DataFrameIndex MultiIndexSeriesconcat to_datetime)GroupingcCsBtddddgddddgd}|d d }d d }||dS) Nabc)keyvaluerrcSst|dkst|SNr)lenAssertionErrorsumxrQ/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/groupby/aggregate/test_aggregate.py dummy_func"sz3test_groupby_agg_no_extra_calls..dummy_func)rgroupbyagg)dfgbrrrrtest_groupby_agg_no_extra_callssr"cCs:|ddddg}|tj}|}t||dS)NcSs|jSNyearrrrr*z&test_agg_regression1..cSs|jSr#monthrrrrr&*r'rrnpmeantmassert_frame_equal)tsframegroupedresultexpectedrrrtest_agg_regression1)s r3c Csf|dd}d}tjt|d|ddW5QRXtjt|d|ddW5QRXdS)NACzMust produce aggregated valuematchcSs|Sr#)Zdescriberrrrr&5r'z#test_agg_must_agg..cSs|jddSNrindexrrrrr&7r')rpytestraises Exceptionr)r r0msgrrrtest_agg_must_agg0s r?cCsHdd}|j|j|jg|}|ddgd}t||dS)NcSs|Sr#rrrrrr&<r'z(test_agg_ser_multi_key..r4Br5)r5rr4rA aggregaterr-assert_series_equal)r fresultsr2rrrtest_agg_ser_multi_key:srFcCstdddtjddddgdddtjddd d gd tjd d ddddddgddgdd}tdddddddtjddddg d ddddddtjd dddg ddddtjdddddtjd g d!d"dd#tjd$d#dddtjtjg d%}|ddg}|d&d'g}t||dS)(Nrrr7M!, )v1v2)r_)rc)rrP)rrQ)bigdamp)bluedry)redrV)rVwetby1by2namesr9 XrQrVrTrrR rWrUrPrS)rNrOrXrYrNrO) rr+nanr from_tuplesrr,r-r.)r2r gr1rrr$test_groupby_aggregation_mixed_dtypeBsN rccCsddddgddtjdgddtjdgddtjdgg}t|tddddgd}|jddd }|jdd }td dddgdd ddgd }t||dS)NTF)r4r)r4r)rAr)rAr)datacolumnsrlevelaxisZ numeric_only@rrr) r+r`rrrarrr-r.)lstr r!r1r2rrr+test_groupby_aggregation_multi_level_columnss     rmcCs|j|tjdd}|jtjks$ttgtjtgtjdd}t | |t | tj |tj | tj |dd|j|dtjdd}t |jttgdtjdd}t| |t| tj |t| tj |dS) NFZ group_keysdtype)rpr:)Zcheck_index_typer4)namerp)rerpr:)rr+r`rpfloat64rrrr-rCrrapplyrrefloatr.)tsr/r0expZexp_dfrrrtest_agg_apply_cornersrwcCst}|dd}|jjdj}t|jt||jjd<| t j }| }t ||t|jt ||jjd<| t j }| }t ||dS)NcSs|jSr#r$rrrrr&r'z1test_agg_grouping_is_list_tuple..r)r-ZmakeTimeDataFramergrouperZ groupingsZgrouping_vectorr r:listrr+r,r.tuple)rur r0rxr1r2rrrtest_agg_grouping_is_list_tuples   r{cCs2|ddg}|tj}|}t||dS)Nr4rAr*)Zmframer0r1r2rrrtest_agg_python_multiindexs r| groupbyfunccCs|Sr#)weekdayrrrrr&r'r&cCs|jSr#r(rrrrr&r'cCs||}|dd}|d}t|||d}|}t|||ddddd}t|d|d|d |d  d}t||dS) Nr4stdvarr,sem)r4rAr5DrAr5r) rrrr-rCrBrr.rr,r)r/r}r0r1r2rrrtest_aggregate_str_funcs          rc CsZ|jdd}|dkrt}d}nt}d|d}tj||d|j|dd W5QRXdS) Nrrg)ZidxmaxZidxminz3reduction operation '.*' not allowed for this dtypez Operation z does not support axis=1r6rrh)r TypeError ValueErrorr;r<r)r reduction_funcr!errorr>rrr%test_agg_str_with_kwarg_axis_1_raisess  rz(func, expected, dtype, result_dtype_dictrrGrHr]int64rglv@r)ijkrg@Int64rrrc Cstddddddggdtddgd d d ggd ||d }|dkrJtnd}d}tj||d|jddd|}W5QRXt|gdd d d gd |}t ||dS)NrrrrrGr r rrrre))r r)r rr!The default value of numeric_onlyr6rf) rr from_productastype FutureWarningr-assert_produces_warningrrr.)funcr2rpresult_dtype_dictr warnr>r1rrr(test_multiindex_groupby_mixed_cols_axis1s rz&func, expected_data, result_dtype_dictrr r_)rrg;f?c Csttdddtdddgddtdd dd gd dd d dd i}|dkrRtnd}d}tj||d|j d dd |}W5QRXt|tdddgddtdd gd dd|}t ||dS)Nr_rrrryrqrrrr)r:rerprrrr6r)rdr:re) rr+arangereshaperrrr-rrrr.)r expected_datarr rr>r1r2rrrtest_groupby_mixed_cols_axis1s( rcCs|d}dd}||}|jdk}|jdk}t|j}tt|g|t ddd}t | d|tt|g|t ddd}t | d|dd }t|j|}t|tstt|d kstdS) Nr4cSs|jSr#sizeZserrrrr&r'z-test_aggregate_item_by_item..foobarZBCDr:rqcSs|jSr#rrrrraggfun)sz+test_aggregate_item_by_item..aggfunr)rrr4rrrerr+arrayryr-rCxsZassert_almost_equalr isinstancer)r r0rr1rrKrvrrrtest_aggregate_item_by_items   rc Csv|ddg}dd}tjtdd||}W5QRX|jdd|jdkf}|ddg|}t||dS)Nr4rAcSs|jtkrtn|SdSr#)rpobjectrrrrrrr4s ztest_wrap_agg_out..funczDropping invalid columnsr6r5)rr-rrrBlocrer.)Z three_groupr0rr1Z exp_groupedr2rrrtest_wrap_agg_out1srcCsPdtjfdtjfdtjfg}|dd|}tdddg}t|j |dS)Nr,maxminr4r5) r+r,rrrrrr-assert_index_equalre)r funcsr1Zexp_colsrrr*test_agg_multiple_functions_maintain_orderAsrcsttjddtjdddddddgd d dttj d d ttj d d gi}tjdd dd}t ddg}t fdddDj }t|||d}t||dS)Nr1/1/2012S)freqperiodsr4rAr5r:re3TH.?q镲 q?r)r4quantilecs"g|]}djj|djqSrrresampler4rvalues.0rr rr Wsz9test_agg_multiple_functions_same_name..rrrer:)rr+randomrandnpd date_rangerrrrrrarTr-r.)r1expected_indexexpected_columnsexpected_valuesr2rrr%test_agg_multiple_functions_same_nameJs&  rc sttjddtjdddddtddd gd d d d ddt tj ddt tj ddgi}tjdd ddd}t j ddddddgd ddgd}t fdddDj}td j|g}t|||d}t||dS)NrrrrZdti)rrrqr4rAr5alpharrrohlcrrrr)r4ropen)r4rhigh)r4rlow)r4rclose)r4rr4rZcs"g|]}djj|djqSrrrrrrrwszKtest_agg_multiple_functions_same_name_with_ohlc_present..rr)rr+rrrrrrrrrrrarrZhstackr4rr-r.)r1rrZnon_ohlc_expected_valuesrr2rrr7test_agg_multiple_functions_same_name_with_ohlc_present_s:    rc Csddg}ddg}|dd|}|dd|}t||tjtdd|d|}W5QRXtjtdd|d|}W5QRXt||dS)N)rr,r)rrr4r5z&\['B'\] did not aggregate successfullyr6)rrr-r.rr)r rZex_funcsr1r2rrr-test_multiple_functions_tuples_and_non_tupless  rc Cs@|d}|tjtjd}|tjtjd}t||gddgdd}|jddddjddd }tjtjgtjtjgd}||}t |||tjtjtjgd}|tjtjtjgd}t ||d d }d d }d} t j t | d&tjtjtjdd}||W5QRXtjg||gd}||dS)Nr4r5rr,rr)keysrhrrrfcSs t|Sr#r+r,rrrrrsz4test_more_flexible_frame_multi_function..foocSstj|ddS)Nr)Zddof)r+rrrrrrsz4test_more_flexible_frame_multi_function..barnested renamer is not supportedr6rr)rrr+r,rr Z swaplevelZ sort_indexrBr-r.r;r<r) r r0ZexmeanZexstdr2dr1rrr>rrr'test_more_flexible_frame_multi_functions&    rc Cs|d}dddddid}d}tjt|d||W5QRXddddd}tjt|d||W5QRXddddd}tjt|d||W5QRXdS) Nr4r,rrrrrr6)rr;r<rrB)r r0rr>rrr test_multi_function_flexible_mixs rcCstddddgddddgddddgd}|d}tddgdd}|dd d }td d g|dd }t|||ddd }td d g|dd }t||dS)Nrrrrr r rr rr cSs |dkSrallrrrrr&r'z1test_groupby_agg_coercing_bools..FTrrcSs |Sr#)Zisnullrrrrrr&r')rrrrBrr-rC)ZdatZgpr:r1r2rrrtest_groupby_agg_coercing_boolss(  ropcCs|Sr#r@rrrrr&r'cCs|Sr#)cumsumrrrrr&r'cCs |dSNr transformrrrrr&r'cCs |dSNrrrrrrr&r'cCs |dSrrrrrrr&r'cCs |dSrrrrrrr&r'cCsdtddgddgd}|dd}||ddj}t|sDt||dj}t|s`tdS)NrFTr r r r )r set_indexrrprr)rr sr1rrrtest_bool_agg_dtypes  rzkeys, agg_indexr rrr rZ input_dtypeboolZint32Zfloat32 result_dtypemethodrsrBrc stdgdgdgd}|d||d<t||dg|}|fdd}|dkrbtd dn|}td|djd gi|d } |d krd g| j_t || dS) NrrTrrcs|jdSrrilocrrrrr& r'z2test_callable_result_dtype_frame..rrr9rs) rrgetattrrr RangeIndexrrer[r-r.) r agg_indexrrrr rr1rr2rrr test_callable_result_dtype_frames rinputT?rpc stdgdg|gd}t||d|}|fdd}|dkrNtddn|}t|djdg|dd } t || dS) Nrrrrcs|jdSrrrrorrr&%r'z3test_callable_result_dtype_series..rrr) rrrrrrrrr-rC) rrrrprr rr1rr2rror!test_callable_result_dtype_seriess  rcCsftddddgddddgd}|dddd d d g}|jjd}tddd d d g}t||dS) Nrrrrr4rAr4rrr,rr)rrrrelevelsrr-r)r resr1r2rrr#test_order_aggregate_multiple_funcs+s  rc Cstddddddgddtjddtjgd |d }|d }tddddgtjgddgddgdgtd gd dddggtddddg|d d|d}t ||dS)NrrrrrMrrrror r rrrrrprq)rer:rp) rrNArrrrrr-r.)Zany_numeric_ea_dtyper r1r2rrrtest_ohlc_ea_dtypes7s$&r howfirstlastrrr,mediancCsttdddgd}|dd|i}|j||_|dd|i}|dkr`|jtj|_tj||dd dS) Nl&tLW Hd~rrrrrr)r,rT)Z check_exact) rrrrrr+rr-r.)rpr r r2r1rrrtest_uint64_type_handlingGsrc CsTd}tddddgddddgd}tjt|d|d d d gW5QRXdS) NzFunction namesrrrrrrr6r4r)rr;r<rrr)r>r rrrtest_func_duplicates_raisesUsrr:abcZ2020r)rr)r r)r r)r rcCs`tdddgdddgd|d}|ddtji}tddgddgdd}t||dS)Nrrr)grouprr9rr)rrrrZnuniquerr-r.)r:r r1r2rrr$test_agg_index_has_complex_internals]s rcCsrtdddddgdddddgdddddgd}|d}tddgddgd tddgdd d }t||dS) Nr r ZoneZtwoZthreesix)key1key2key3r)rrrr9)rrrrr-r.r r1r2rrrtest_agg_split_blockns   rc Csttjdddddddgdd d dgdddd gtjddddd d dgd t}|d d d d g}ttdgdgdgdgtdgdgd }t ||dS)N2000r)rr r rrrrre)r4rAr5rEFr) rrrrrrr Timestampr-r.rrrr#test_agg_split_object_part_datetimes,           r!c@sVeZdZddZddZddZddZej d e j d d d d d d ggddZ dS)TestNamedAggregationSeriescCstddddg}|ddddg}|jddd}tdd gddgdd d gddgd }t|||jddd }|d d g}t||dS)NrrrrrrrrrHr r r)r r rrrrr-r.)selfr grr1r2rrrtest_series_named_aggs  z0TestNamedAggregationSeries.test_series_named_aggc CsXtddgddg}tjtdd|W5QRX|g}t}t||dS)Nrrr Must provider6) rrr;r<rrrr-r.r$r%r1r2rrrtest_no_args_raisess  z.TestNamedAggregationSeries.test_no_args_raisescCsNtdddgdddg}|jddd}tddgddgd}t||dS)Nrrrrrrr#)r$r%r0r2rrr*test_series_named_agg_duplicates_no_raisesszETestNamedAggregationSeries.test_series_named_agg_duplicates_no_raisescCsVtdddgdddg}|jddddd}tddgddgd}t||dS) NrrrrcSsdSrrrrrrr&r'z9TestNamedAggregationSeries.test_mangled..cSsdSNrrrrrrr&r'rr#r(rrr test_mangledsz'TestNamedAggregationSeries.test_mangledinpanythingrcolumnaggfunc)r.rc CsZtddddddddg}dt|j}tjt|d||jj|dW5QRXdS) NrrrrrGzfunc is expected but received r6r ) rtype__name__r;r<rrrr)r$r-rr>rrrtest_named_agg_nametuples z3TestNamedAggregationSeries.test_named_agg_nametupleN) r4 __module__ __qualname__r&r)r*r,r;mark parametrizerNamedAggr5rrrrr"s   r"c@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)TestNamedAggregationDataFramec Cs tddddgddddgddd d gd }|d jd dd}tddgdd gdtddgd dddgd}t||tjtj dd}|d jddt fdtj fd dd|fd}tdd gddgddgddgdd gddgdtddgd ddddddd gd}t||dS)!Nr r rrrrrGrrHr\rr4rArr4rrAr)a_maxb_maxrr?r@rbrrArr4)b_mina_mina_meanr?r@a_98g?g@g\(\?gףp= @rCrDrErF) rrrrr-r. functoolsrr+ percentilerr,)r$r r1r2Zp98rrrtest_agg_relabels>"   z.TestNamedAggregationDataFrame.test_agg_relabelcCsntddddgddddgddd d gd }|d jfd di}td ddgitddgd dd}t||dS)Nr r rrrrrGrrHr\r<rzmy colr=rr9rrrrr-r.r$r r1r2rrrtest_agg_relabel_non_identifiers " z=TestNamedAggregationDataFrame.test_agg_relabel_non_identifiercCstddddgddddgd}|djddd }tddgddgd tddgdd d }t||tjtj d d }tjtj dd }d|_ d|_ tdddddgdddddgd}|djd|fd|fd}tddgddgdtddgdd d }t||dS)Nrrrrrrr4rBrrr92rFquant50quant70r r rG)col1col2rQrR)Z quantile_50Z quantile_70?@g333333?g@) rrrrr-r.rGrr+rHr4)r$r r0r2rOrPtestrrrtest_duplicate_no_raisess$& " z6TestNamedAggregationDataFrame.test_duplicate_no_raisescCstddddgddddgdtddgd d ggd }|jdd jd ddd}tddgddgddgdddgd }t||dS)Nrrrrrrr4rAr r r9rr=)r4r)rAr,)ZaaZbbccrSg @)rrrrrr-r.rKrrrtest_agg_relabel_with_levels z9TestNamedAggregationDataFrame.test_agg_relabel_with_levelc Cstdddgdddgd}|d}d}tjt|d|jdd W5QRXtjt|d|W5QRXtjt|d|jd d d W5QRXdS) Nrrrrrr4r'r6)rr>)rrrr)rrr;r<rr)r$r r0r7rrrtest_agg_relabel_other_raises"s z;TestNamedAggregationDataFrame.test_agg_relabel_other_raisesc CsPtddgddgd}td}tjt|d|djdd W5QRXdS) NrrrrzColumn(s) ['C'] do not existr6r4)r5r)r)rreescaper;r<KeyErrorrr)r$r r7rrrtest_missing_raises/s z1TestNamedAggregationDataFrame.test_missing_raisescCs`tddgddgd}|djtddtjddd d }|djd d d }t||dS) Nrrrrr4rArcountr/r r)rAr)rAr^rrrrr:r-r.rKrrrtest_agg_namedtuple5s  z1TestNamedAggregationDataFrame.test_agg_namedtuplecCsvtddgddgddgd}|djdd d fd d d fd }tddgddgd tddgddd}t||dS)Nrrrrr)r4rAr5r4rAcSsdSrrrrrrr&?r'z.r5cSsdSr+rrrrrr&?r'r_rr9rJrKrrrr,=s$&z*TestNamedAggregationDataFrame.test_mangledN) r4r6r7rIrLrVrXrYr]rar,rrrrr;s$   r;zCagg_col1, agg_col2, agg_col3, agg_result1, agg_result2, agg_result3rr4rrc)rrAr,g@@cCst|Sr#)rrrrrr&Pr'cCsdSr+rrrrrr&Qr'rdcCsdSr+rrrrrr&Zr'c Cstddddgddddgddd d gd }td d dg|_tddgd d}|d jdd}tdddgi|d} t|| |d j|||d}t|||d|d} t|| dS)Nr r rrrrrGrrHr\r<rrrcrdrrb)r?r?r9)Zcol_1Zcol_2Zcol_3) rrrarerrrr-r.) Zagg_col1Zagg_col2Zagg_col3Z agg_result1Z agg_result2Z agg_result3r idxr1r2rrr"test_agg_relabel_multiindex_columnDs$!"   rhc Csjtddddgddddgddd d gd }td d dg|_tjtdd|d jddW5QRXdS)Nr r rrrrrGrrHr\r<rfrcrdz do not existr6))Yr rr2) rrrarer;r<r\rrrrrr,test_agg_relabel_multiindex_raises_not_existxs "rjcCstddddgddddgddd d gd }td d dg|_|d jddd}tddgd d}tddgddgd|d}t||dS)Nr r rrrrrGrrHr\r<rfrcrd)rcrrrr9) rrrarerrrr-r.)r r1rgr2rrr&test_agg_relabel_multiindex_duplicatess" rkkwargsrr_cCsntdddgdddgdddgd}|d|}tddgtddgdddtd d ggd }t||dS) Nrrrrrr rr rrr)rrrrrrar-r.)rlr r1r2rrr test_groupby_aggregate_empty_keys"rmcCsbtdddgdddgdddgd}|ddgi}ttdggggggdd }t||dS) Nrrrrrr r )rcodesr)rrrrr-r.rrrr-test_groupby_aggregate_empty_key_empty_returns"rocCsbtdddgd}|jddgddjdtfd}tdgtggggggddgd d }t||dS) Nr r rrFrn)rrrZr)rrrryrr-r.rrrr2test_groupby_aggregate_empty_with_multiindex_framesrpcCsttddddddgdddddd gd }|jd d d }|jtjdddd}tdddgddd gd}t||dS)Nrrzr皙?rj@ @?)rvalrFZas_indexrvrr/min_val)rryr`r r0r1r2rrr9test_grouby_agg_loses_results_with_as_index_false_relabels r{c Cstddddddgddddddgdddd d d gd }|jd dgdd}|jtjdddd}tdddgdddgdd dgd}t||dS)Nrrr r rrrrrjrsrtru)rrrvrrFrwrvrr/rx)rrryr`rzrrrDtest_grouby_agg_loses_results_with_as_index_false_relabel_multiindexsr|rcCs|Sr#)r,rrrrr&r'cCs t|Sr#rr}rrrr&r'cCs t|Sr#)r+Znanmeanr}rrrr&r'cCsdddgdddgg}t|tjdddgdddggddgd d }|td dg|}d d dddddd dd}t|}|j|_t ||dS)NrrrrGrHrZSiskoZJanewayrZrrrg@rkrTg@rj))rr)rr)rr) rrZ from_arraysrr+rrrer-r.)rrdr r1 expected_dictr2rrrtest_multiindex_custom_funcs rcCstj|ddSNg?rr+rHr}rrrmyfuncsrcCstj|ddSrrr}rrrr&r'cCsztddddgddddgddd d gd }|d jd d|fd}tddgddggddgtddgd dd}t||dS)Ncatdog333333"@@#@A@皙@re#@h@kindheightweightr)rr,r) mean_heightperc90g"@g0' 5"@g4@g5^I @rrrrrJ)rZanimalsr1r2rrrtest_lambda_named_aggs     rcCsttdgdddtdtdd}dddg|d <|d d d }dgdgdggddgddgddggg}t|tddgd d d tdddgd dd}t ||dS)Nrr]rZXYZrrdrer:zgroup 1rgroupingcSs|Sr#)tolistrrrrr&r'z,test_aggregate_mixed_types..rr XriZror) rr+rrryrrBrr-r.)r r1rr2rrrtest_aggregate_mixed_typess&rzNot implemented;see GH 31256reasoncCsjdd}tdtdddgi}|dddg|}tdtjdtjgddiddgd }t||dS) NcSst|dkrdStjSdS)Nrr)rrr rrrrr1!s z5test_aggregate_udf_na_extension_type..aggfuncr4rrrrror9)rrrrrr r-r.)r1r r1r2rrr$test_aggregate_udf_na_extension_types $rc@sNeZdZddZddZejjddejddd Z d d Z d d Z dS)TestLambdaManglingcCsttddddgddddgd}|ddd d d d gi}tddgddgd tddgdd d}t||dS)Nrrrrrrr4rAcSsdSrrrrrrr&0r'z/TestLambdaMangling.test_basic..cSsdSr+rrrrrr&0r'))rA )rA rr9rJrKrrr test_basic.s zTestLambdaMangling.test_basiccCsZtddddgddddg}|ddddg}tddgddgd }t||dS) NrrrrrcSsdSrrrrrrr&:r'z?TestLambdaMangling.test_mangle_series_groupby..cSsdSr+rrrrrr&:r'rrr#r(rrrtest_mangle_series_groupby8sz-TestLambdaMangling.test_mangle_series_groupbyzGH-26611. kwargs for multi-agg.rz-ignore:Dropping invalid columns:FutureWarningcCsddd}ddd}tddgddg||gd}tdgdgd }t||tddgddgj||gdd d }td gd gd }t||dS)NrcSs|||Sr#r@rrr rrrr&Ar'z5TestLambdaMangling.test_with_kwargs..rcSs|||Sr#r@rrrrr&Br'rrrrr)r )r)rr#)r$f1f2r1r2rrrtest_with_kwargs>s  " &z#TestLambdaMangling.test_with_kwargscCstddddgddddgddd d gd }d d dg}tddgddgd d gdtddgdd|d}|jddjtjddddtjdddtjdddd}t|||jddjdddfddd}t||dS)Nrrrrrrrrerrrheight_sqr_min height_max weight_maxp= ׳T@B@)rrrrrrZbyrcSst|dSr8r+rrrrrr&cr'z=TestLambdaMangling.test_agg_with_one_lambda..r/rrcSst|dSr8rrrrrr&lr'rrrr)rrrrrr:r-r.r$r rer2Zresult1Zresult2rrrtest_agg_with_one_lambdaKs<          z+TestLambdaMangling.test_agg_with_one_lambdac Cstddddgddddgddd d gd }d d dddg}tddgddgd d gddgddgdtddgdd|d}|jddjdddfdddddfdddfd}t|||jddjtjdd dd!tjdd"d!tjdd"d!tjdd#dd!tjdd$dd!d}t||dS)%Nrrrrrrrrerrrrrr height_max_2 weight_minrr)rrrrrrrrrrcSst|dSr8rrrrrr&r'z=TestLambdaMangling.test_agg_multiple_lambda..rrcSs t|Sr#r+rrrrrr&r'rcSs t|Sr#rrrrrr&r'cSst|dSr8rrrrrr&r'r/rcSs t|Sr#rrrrrr&r'cSs t|Sr#rrrrrr&r')rrrrr-r.rr:rrrrtest_agg_multiple_lambdarsR            z+TestLambdaMangling.test_agg_multiple_lambdaN) r4r6r7rrr;r8xfailfilterwarningsrrrrrrrr-s    'rcCs^tdddgdddgd}|dddd i}tddgddgdd}t||dS) NrWrrjrr4rAcSs||jdS)N)getr:rrrrr&r'z+test_groupby_get_by_index..)rrrrr-r.)r rr2rrrtest_groupby_get_by_indexsrzgrp_col_dict, exp_data)nrcat_ordrrc Cstddddddddgtd td d }|d d d }|dj|d<|d|}tjddgddgddd d}t||d}d|kr|dj }|d||d<t ||dS)NrrrrrGrrHr\aabbccddaaaabbbbrrrcategoryrrrrr r F categoriesZorderedrqrp)rdr:) rryrr as_orderedrrrCategoricalIndexrpr-r.) grp_col_dictexp_datainput_df result_df cat_index expected_dfrprrr test_groupby_single_agg_cat_colss*   r)rrr )rGr\r)rr r )rGrrr)rrc Cs tddddddddgtd td d }|d d d }|dj|d<|d|}tjddgddgddd d}g}| D]:\}}t |tr|D]}| ||gqq| ||gqt t|} t|| |d} | jD].} t | trd| kr| | |dj| | <qt|| dS)NrrrrrGrrHr\rrrrrrrr r Frr)rryrrrrrrritemsrappendrrarzrerpr-r.) rrrrrZmulti_index_listrvrZ multi_indexrcolrrr#test_groupby_combined_aggs_cat_colss8   rcCs\tddddgddddgd}|d}|dg}|jd|_|d}t||dS)Nrrrr rr)rrrreZ droplevelr-r.)r rbr1r2rrrtest_nonagg_aggs    rc Cshtddgtdddddddtd ddddddgd }|d j}|d d }t||dS) Nrriirrri`=i rr4rA)rdatetimerrArrr-rCrrrrtest_aggregate_datetime_objectss rcCstdddgdddgdddgd}|jd|_|dd g}|jd d }tjd d gdd}tddg|dd}t ||dS)Nrrrrr)c0c1pOrrcSs t|dkSrrrrrrr&*r'z1test_groupby_index_object_dtype..)rrr)rrrZFTrr) rr:rrrrrrarr-rC)r r0rrr2rrrtest_groupby_index_object_dtype%s"rcCsTdd}tdgtdgd}|dd|}tdggdgd}t||dS) NcSs|rdSt|Sr#)Zisnarr+rrrrrr7s z)test_timeseries_groupby_agg..funcrz2018-01-16 00:00:00+00:00r9cSsdSr+rrrrrr&=r'z-test_timeseries_groupby_agg..r)rrr rrr-r.)rr rr2rrrtest_timeseries_groupby_agg4s rc Cs|dkr dS|dkrtnd}tddgdtjgg}tj|dd*|d|}|dd|i}W5QRX|dkrtj||ddd nt ||t|j |j dS) N)ZcorrwithZnthZmadrrzThe 'mad' method is deprecatedr6)rZngroupF)Z check_names) rrr+r`r-rrrrCr.Zdtypes)rrobjZresult_reduced_seriesZresult_reduced_framerrr test_groupby_aggregate_directoryCs$ rcCsDtdddgdd}tdgdd}|dddg}t||dS)Nz1 dayz3 daysNaTztimedelta64[ns]roz2 daysr)rrr,r-rC)rdr2r1rrrtest_group_mean_timedelta_nat[srzinput_data, expected_outputz2021-01-01T00:00rz2021-01-01T02:00z2021-01-01T01:00z2021-01-01T00:00-0100z2021-01-01T02:00-0100z2021-01-01T01:00-0100cCs<tt|}tt|}|dddg}t||dSr)r rrr,r-rC)Z input_dataZexpected_outputrdr2r1rrrtest_group_mean_datetime64_nates  rz func, outputy @2@y$@6@yD@V@yI@[@cCsNttdddddg}||jd|}t|}t ||dS)Nrrrr@) rr+rrdotrr:rr-rC)routputrdr1r2rrrtest_groupby_complex{s rc CsXttdddddg}d}tjt|d||j d |W5QRXdS)NrrrrrzNo matching signature foundr6) rr+rrrr;r<rrr:r)rrdr>rrrtest_groupby_complex_raisess rprodc CsXtdddgdddgdddgd }|jd dd }tjtd d ||W5QRXdS)NrrrrrGrrHr\rr rz"axis other than 0 is not supportedr6)rrr;r<NotImplementedErrorr)rr r!rrrtest_multi_axis_1_raisess"rztest, constantr4rAr5rkr)ZmarkscCs<t|}|dtj}t|}|d}t||dSr)rrrrmoderr-r.)rUZconstantZdf1r1r2rrrtest_agg_of_mode_lists  rc Cstdddgtddddgd}|d}d }tjt|d |tjW5QRXt j t d d |jtjd d W5QRXdS)NrrZxyzrrrGrr z$The operation rrrtest_numeric_only_warning_numpys  r)i__doc__rrGrrZZnumpyr+r;Z pandas.errorsrZpandas.core.dtypes.commonrZpandasrrrrrr r Zpandas._testingZ_testingr-Zpandas.core.groupby.grouperr r"r3r?rFrcrmrwr{r|r8r9rrintrtrrrrrrrrrrrrrrrrr rZuint64rrrryZinterval_rangeZ period_rangerarrr!r"r;rr:r,rhrjrkrmrorpr{r|rrrrrrrrrrrrrrrrrrrrparamrrrrrrs        1      !$ (  (      5v            y  )      **0