o ۭjW @s<ddlZddlZddlZddlmZmZmZmZm Z m Z m Z ddl m ZddlmZejddgdddZd d Zd d Zd dZejdddgddZddZejddidfdddidfdidfdddidfgdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Z d-d.Z!d/d0Z"ejd1d2d2ej#gd2d3ej#ggd4d5Z$d6d7Z%d8d9Z&d:d;Z'ejdZ(d?d@Z)dAdBZ*ejdCe+dDdEgdFdGZ,dHdIZ-ejdJdKdLgdMdNZ.dOdPZ/dQdRZ0ejdSdKifdLddifdLddifgdTdUZ1dVdWZ2ejdXeddge3ddYgdZeej4ddggddd[d\d]feej5d^ej6d]e d_d^d`dadbeej4ddggd^dd[d\d]fgejdcddgdddeZ7ejdfdee8d^d\d]fdggdhdiZ9djdkZ:ejdlej;gdmgfej;ejejdrdsej;iej;ej.frcS|Sr )rrrrr z/test_series_map_box_timedelta..)g@g@g@)rrapplymaptmassert_series_equal)rserrresultexpectedrrrtest_series_map_box_timedeltas   r&cCs|jtj|d}tjdd t|}Wdn1swYt|||jtj|d}t|}t||ttdt gddd}|jdd |d}t||||us\J|j |j usdJ|j |j kslJ|j |j kstJtgd tj d }|jd d |d}t||dS) Nrignoreallfoobarname)dtyper-indexcS|Sr rrrrrr4ztest_apply..rr/r.cSr0r rrrrrr?r1)rnpsqrterrstater!r"exprobjectrr/r.r-float64)datetime_seriesrr$r%srsrrr test_apply's$      r?cCs<tddg}dd}|j|dd}||}t||dS)Nr3r4cSs ||dfSNr3rrrrrrF z3test_apply_map_same_length_inference_bug..fr r)rrr r!r")r=rr$r%rrr(test_apply_map_same_length_inference_bugCs  rB convert_dtypeTcCs^ttjdd}dd}tt|j||ddWddS1s(wYdS)Nr4 cSs|dkr|StjS)Nr)r6nanrrrrfuncRsz1test_apply_convert_dtype_deprecated..funcr )rCr) rr6random default_rngstandard_normalr!assert_produces_warning FutureWarningr)rCr#rFrrr#test_apply_convert_dtype_deprecatedNs  "rLcCsDtdg}|jtjdd}|dddgksJt|dts JdS)Nzfoo,bar),argsrr*r+)rrstrsplit isinstancelist)r=r$rrrtest_apply_argsYs rTzargs, kwargs, incrementrar3)r4r )r3cr4cCszddd}tddg}d}tjt|d|j|dg|Ri|}Wdn1s,wY||}t||dS) NrcSs||d|d|S)NrDdr)rrUbrWrrrrgsztest_agg_args..fr3r4zhin Series.agg cannot aggregate and has been deprecated. Use Series.transform to keep behavior unchanged.match)rrr)rr!rJrKaggr")rOkwargs incrementrr=msgr$r%rrr test_agg_argsas  racCstgd}ddd}ddd}d }tjt|d |j|dd d d Wdn1s-wYtjt|d |j||gdd d d Wdn1sOwYtjt|d |j||ddd d d WddS1sswYdS)Nr2r3rcS |||Sr r)rrUrWrrrfoo1yrAz.test_agg_mapping_func_deprecated..foo1r4cSrbr r)rrZrWrrrfoo2|rAz.test_agg_mapping_func_deprecated..foo2z+using .+ in Series.agg cannot aggregate andr[r)rW)rUrZ)r3r)r4r)rr!rJrKr])r=rcrdr`rrr test_agg_mapping_func_deprecatedus   "rfcCsttddd}dd}|s1d}tjt|d|j||dWddS1s*wYdS|j||d}||}t||dS) Nz1/1/2000rDrcSs|j|j|jfSr )hourdaymonthrrrrrFsz2test_series_apply_map_box_timestamps..funcz&Series' object has no attribute 'hour'r[r) rrpytestraisesAttributeErrorrr r!r")rr#rFr`r$r%rrr$test_series_apply_map_box_timestampss  rncCs\tdtdg}t|dd}|jdksJ|jdddd }td d g}t||dS) N 2011-01-01 2011-01-02zM8[ns]r.zdatetime64[ns]cSt|jd|jd|jSN_type__name__ritzrrrrrz%test_apply_box_dt64..r rTimestamp_1_NoneTimestamp_2_Nonepd Timestamprr.rr!r"valsr#resr9rrrtest_apply_box_dt64s   rcCsdtjdddtjdddg}t|dd}|jdksJ|jdd d d }td d g}t||dS)Nroz US/Eastern)rxrpzM8[ns, US/Eastern]rqzdatetime64[ns, US/Eastern]cSrrrsrurrrrrryz'test_apply_box_dt64tz..r rzTimestamp_1_US/EasternzTimestamp_2_US/Easternr|rrrrtest_apply_box_dt64tzs    rcCsXtdtdg}t|}|jdksJ|jdddd}tdd g}t||dS) Nz1 daysz2 daysztimedelta64[ns]cSt|jd|jSrs)rvrwdaysrrrrrz%test_apply_box_td64..r r Timedelta_1 Timedelta_2)r} Timedeltarr.rr!r"rrrrtest_apply_box_td64s  rcCs`tjdddtjdddg}t|}|jdksJ|jdddd }td d g}t||dS) NroMrrpz Period[M]cSrrs)rvrwfreqstrrrrrrrz'test_apply_box_period..r rPeriod_M)r}Periodrr.rr!r"rrrrtest_apply_box_periods  rcstddddd}t|dd}|jdd d }tdd ddd}t|dd}t|||jfd d d }tttd dgdrHdndd}t||fdd}|j|d }rrtdgddd}t||dS|dksxJdS)Nrorprrz Asia/TokyoXXr,cSs|tjSr )r}offsetsDayrrrrrz'test_apply_datetimetz..rz 2011-01-03csr|jS|jjSr )rhrrrrrrsrint64int32r-r.csrt|jSt|jjSr )rPrxrrrrrrsz test_apply_datetimetz..f)r tz_localizerrr!r"rSrange)rvaluesr=r$ exp_valuesr9rrrrtest_apply_datetimetzs"   $  rcCs4tjtdtddd}t|dtdd}|sHd}tjt|d |jd d |d Wdn1s4wY|jd d |d dksFJdS|jdd |d }tjtdtddd}t|dtdd}t ||t |j |j |dd }tdgddtdd}t |||s|j t ksJdSdS)NABBABCDDCBAT) categoriesorderedrabcdefg)r-r/z&Series' object has no attribute 'lowerr[cSrr lowerrrrrrrz(test_apply_categorical..rcSdSNArrrrrrr1rcSrr rrrrrrrabbabcddcbacSrrrrrrrrr1)r} CategoricalrSrrkrlrmrr!r"assert_categorical_equalrr.r:)rusing_infer_stringrr#r`r$r9rrrtest_apply_categoricals$  rseriesz1-1z1-2cCst|dd}|s-d}tjt|d|jdd|dWddS1s&wYdS|jdd|d}|t}td d tjgdd}|t}t ||dS) Ncategoryrqz('Series' object has no attribute 'split'r[cS|ddSN-rrQrrrrrrz8test_apply_categorical_with_nan_values..rcSrrrrrrrrr1) rrkrlrmrastyper:r6rEr!r")rrr=r`r$r%rrr&test_apply_categorical_with_nan_valuess    rcCs8tgtdddtd}|jdd|d}t||dS)Nz 2018-01-01r)startrr5cSr0r rrrrrrr1zEtest_apply_empty_integer_series_with_datetime_index..r)rrintrr!r")rr=r$rrr3test_apply_empty_integer_series_with_datetime_index srcCshttgddgd}tgddgd}dd}|dj||fd}tgd dd d }t||dS) N)r3r4rreNumbers)columns)r4rr4r3r4IndicescSs|dj|}|S)Nr)iloc) targetRowtargetDFvalrrrretrievesz+test_apply_dataframe_iloc..retrieverN)rrerr4ruint64r)rr6rrrr!r")uintDFindexDFrr$r%rrrtest_apply_dataframe_ilocs rcCsFtjddt|}t|}|jtj|d}|}t|||jtjg|d}|}dg|_ t |||jdg|d}t ||t ||gdd}ddg|_ |jtjtjg|d}t ||t ||gdd}dd g|_ | d }|jtjtjd |d}t|||WddS1swYdS) Nr'r(rr7r3axisabsoluter*r+r)r*r+)r6r8r7absrcopyr!r"to_framerassert_frame_equalrunstackrename reindex_like) string_seriesrf_sqrtf_absr$r%rrrtest_transforms,         "ropcCsZ|dvr|tjj|ddtdtg}|dvr!t}d}n t}ddd g}tj ||d | |d gWdn1sCwYtj ||d | |d d Wdn1sbwYtj ||d | |gd gd Wdn1swYtj ||d | |d g|gd WddS1swYdS) N)ffillbfillpadbackfillshiftz is successful on any dtype)reasonr)fillnangroupTransform function failed|z4not supported between instances of 'type' and 'type'zunsupported operand typer[rrB) applymarkerrkmarkxfailrr: ValueError TypeErrorjoinrl transform)rrr#errorr`rrrtest_transform_partial_failureEs4"rcCs(dd}dd}tdtg}d}tjt|d|||gWdn1s*wYtjt|d|||dWdn1sIwYtjt|d||g|gdWdn1sjwYtjt|d|||g|gdWddS1swYdS) NcSr0r rrrrrnoopkz7test_transform_partial_failure_valueerror..noopcSstr )r)rtrrr raising_opnrz=test_transform_partial_failure_valueerror..raising_oprrr[r)rr:rkrlrr)rrr#r`rrr)test_transform_partial_failure_valueerroris "rcCsrttdddd}|ddg}tddgddgdd }t|||d di}tdgd gdd }t||dS) Nrr)r.r-minmaxrrr/r-r*)rrr]r!r"r=r$r%rrr test_demos rrFcCt|Sr )rPrrrrrrrcCs@|j||d}|r||}t||dS|t|ksJdS)Nr)rr r!r"rP)rrFrr$r%rrr(test_apply_map_evaluate_lambdas_the_sames  rcCstt|dd}Wdn1swYt|tr)t|t|ks+Jtt |t}Wdn1s@wYt|trRt|t|ksTJdS)NcSrr )rvrrrrrrz+test_agg_evaluate_lambdas..)r!rJrKr]rRrlenrv)rr$rrrtest_agg_evaluate_lambdass   "rop_namer]rcCsd}|dkrtnd}tj||dt||dd}Wdn1s%wYt||dd}t||tjt|d|dd}Wdn1sRwYt||dS) Nzcannot aggregater]r[cSt||dgddgdSNr4rx^2r/rrrrrrz)test_with_nested_series..r4)rrcSrrrrrrrrr)rKr!rJgetattrrrr])r<rr`warningr$r%rrrtest_with_nested_seriess rc Cs<|}|dddddddddd d }t||dS) NcountmeanstdrcS |dS)Ng?quantilerrrrr z)test_replicate_describe..mediancSr)Ng?r rrrrrr r)rrrrz25%z50%z75%r)describerr!r")rr%r$rrrtest_replicate_describes rcCs>|ddg}t||gddg|jd}t||dS)Nsumrr,)r]rrrr-r!r")rr$r%rrr test_reducesrz how, kwdscCstgd}t||d i|}|j}||ksJt||gdfi|}tdddd}t||t||ddddfi|}t||dS) N)r3r4Nsize)rrr@@g?rr)r)rrrr!r")howkwdsr=r$r%rrrtest_non_callable_aggregatess   rcCsNtdgd}|jdddddg|d}tgdgdd }t||dS) NrerrcSrr rrrrrrrz3test_series_apply_no_suffix_index..r) rr)rrrr)rrr!r")rr=r$r%rrr!test_series_apply_no_suffix_indexsrzdti,expl,b/rrrrqrD 2020-01-01rgtsrawarecCs:|r |dj}n|j}t|dd}t||dS)NUTCcSs tddgS)Nr3r4rrrrrrs zCtest_apply_series_on_date_time_index_aware_series..)rr/rrr!r)dtir9rr/r$rrr1test_apply_series_on_date_time_index_aware_seriess rzby_row, expected)Fr3cCsJttjdtjdtddddd}t|jjdd|d }t||dS) NrDrqrr)rrxrcSrr@rrrrrrr1zCtest_apply_scalar_on_date_time_index_aware_series..r) rr6aranger;rr/rr! assert_equal)rr%rr$rrr1test_apply_scalar_on_date_time_index_aware_seriess  r"cCs~ddg}t|}t|jtj|d}tt||dtjtjtjg}t|}t|}|jtj|d}tt||dS)Nz00:00:01z00:00:02r) r} to_timedeltarrr!r"r6rENaT)rlist_of_valid_stringsrUrZlist_of_stringsr#rrrtest_apply_to_timedelta#s  r'z ops, namesrrz how, kwargsc stfddt||D}d|_|dkrtnd}dd|d}tj||dt||fi|}Wdn1s?wYt||dS) Nci|] \}}||qSrr.0r-rrrr @z/test_apply_listlike_reducer..rr]zusing Series.[r]r[) rzipr-rKrr!rJrr") ropsnamesrr^r%warnr`r$rr+rtest_apply_listlike_reducer1sr3r0rrc stfdd|D}j|_|dkrtnd}d}tj||dt||fi|}Wdn1s8wYt||dS)Ncr(rrr)r+rrr,Xr-z/test_apply_dictlike_reducer..r]zusing Series.[sum|mean]r[)ritemsr-rKr!rJrr") rr0rr^rr%r2r`r$rr+rtest_apply_dictlike_reducerIsr5r7rcsltjdd&tfdd|Ddd}||_j||d}t||WddS1s/wYdS)Nr'r(csg|]}|qSrr)r*rr+rr msz3test_apply_listlike_transformer..r3rr)r6r8rrrr!r)rr0r1rr%r$rr+rtest_apply_listlike_transformeras  "r7z ops, expectedcCr0r rrrrrrvr1r2cCrr rrrrrrwrrcCs*tgd}|j||d}t||dSNr2rrrr!r!)r0r%rr#r$rrrtest_apply_listlike_lambdass r:csntjdd'tfdd|D}j|_j||d}t||WddS1s0wYdS)Nr'r(cr(rrr)r+rrr,r-z3test_apply_dictlike_transformer..r)r6r8rr4r-rr!r")rr0rr%r$rr+rtest_apply_dictlike_transformers  "r;cCr0r rrrrrrr1rcCrr rrrrrrrcCs*tgd}|j||d}t||dSr8r9)r0rr%r#r$rrrtest_apply_dictlike_lambdas r<cCstdtdittddd}|jdd}tdtjtjgddtjggdgttdd dttdddd }t||dS) Nrrr,cSs"tt|dtt|dddS)Nr3yr,)rrrrrrrrs"z0test_apply_retains_column_name..?)r>r?rr=)rr/) rrrrrr6rEr!r)rdfr$r%rrrtest_apply_retains_column_namesrAcCsFtddtggdd}|t}ttttggdd}t||dS)Nrstring)rUrZrWr)rfloatrrvrrPr!r"rrrrtest_apply_types rDcCsVtgdgdg}|dd}tddgddgd d gtjd gd }t||dS) Nr2)rerrrcSrr rrrrrrrz6test_series_apply_unpack_nested_data..r?g@rg@rg@r)rr3r4r)rrrr6rEr!r)r#r$r%rrr$test_series_apply_unpack_nested_datas$rE)Lnumpyr6rkpandasr}rrrrrrrpandas._testing_testingr!pandas.tests.apply.commonr fixturerr&r?rBr parametrizerLrTrarfrnrrrrrrrErrrrrrrrPrrrrrrr DatetimeIndexrepeatr r;ronesr"r'rrarrayr3r5r7rr7r:r9r; from_arraysrr<rArDrErrrrs $    *        & #                             "