a Pкf╒Wу@s<ddlZddlZddlZddlmZmZmZmZm Z m Z mZddlm ZddlmZejddgdНddДГZd d ДZddДZd dДZejаdddgбddДГZddДZejаddidfdddidfdidfdddidfgбdd ДГZd!d"ДZd#d$ДZd%d&ДZd'd(ДZd)d*ДZd+d,ДZ d-d.ДZ!d/d0ДZ"ejаd1d2d2ej#gd2d3ej#ggбd4d5ДГZ$d6d7ДZ%d8d9ДZ&d:d;ДZ'ejаdДГZ(d?d@ДZ)dAdBДZ*ejаdCe+dDdEДgбdFdGДГZ,dHdIДZ-ejаdJdKdLgбdMdNДГZ.dOdPДZ/dQdRДZ0ejаdSdKifdLddifdLddifgбdTdUДГZ1dVdWДZ2ejаdXeddgeа3ddYgбdZНeej4ddggddd[Нd\d]Нfeej5d^ej6d]Нe d_d^d`НdadbНeej4ddggd^dd[Нd\d]НfgбejаdcddgбdddeДГГZ7ejаdfdeeа8d^бd\d]НfdggбdhdiДГZ9djdkДZ:ejаdlej;gdmgfej;ejejаdrdsej;iej;ej.frcSs|абSr)rrrrr┌ єz/test_series_map_box_timedelta..)gї@g∙ї@g┌Ў@)rr┌apply┌map┌tm┌assert_series_equal)r┌serr┌result┌expectedrrr┌test_series_map_box_timedeltasr%cCs|jtj|dН}tjddНПtа|б}WdГn1s<0Ytа||б|jtj|dН}tа|б}tа||бttdt gddНdН}|jdd Д|dН}tа||б||us╕JВ|j |j us╚JВ|j|jks╪JВ|j|jksшJВtgd вtj dН}|jdd Д|dН}tа||бdS) Nr┌ignoreй┌all┌foo┌barй┌name)┌dtyper,┌indexcSs|Srrrrrrr4rztest_apply..йщщrйr.r-cSs|Srrrrrrr?r)r┌np┌sqrt┌errstater r!┌expr┌objectrr.r-r,┌float64)┌datetime_seriesrr#r$┌s┌rsrrr┌ test_apply's"( r<cCs<tddgГ}ddД}|j|ddН}|а|б}tа||бdS)Nr0r1cSs||dfSйNr0rrrrrrFsz3test_apply_map_same_length_inference_bug..fr r)rrrr r!)r:rr#r$rrr┌(test_apply_map_same_length_inference_bugCs r>┌ convert_dtypeTcCs\ttjаdбаdбГ}ddД}tаtбП |j||ddНWdГn1sN0YdS)Nr1щ cSs|dkr|StjS)Nr)r3┌nanrrrr┌funcRsz1test_apply_convert_dtype_deprecated..funcr )r?r) rr3┌randomZdefault_rngZstandard_normalr ┌assert_produces_warning┌ FutureWarningr)r?r"rBrrr┌#test_apply_convert_dtype_deprecatedNsrFcCsDtdgГ}|jtjddН}|dddgks.JВt|dtГs@JВdS)Nzfoo,bar)·,й┌argsrr)r*)rr┌str┌split┌ isinstance┌list)r:r#rrr┌test_apply_argsYs rNzargs, kwargs, incrementr┌ar0)r1rщ )r0┌cr1щ╔cCszdddД}tddgГ}d}tjt|dНП*|j|dg|вRi|дО}WdГn1sX0Y||}tа||бdS) NrcSs||d|d|S)Nr@щdr)rrO┌brQrrrrgsztest_agg_args..fr0r1zhin Series.agg cannot aggregate and has been deprecated. Use Series.transform to keep behavior unchanged.й┌match)rrr)rr rDrE┌aggr!)rI┌kwargs┌ incrementrr:┌msgr#r$rrr┌ test_agg_argsas 8r[cCsЄtgdвГ}dddД}dddД}d }tjt|d НП"|j|dddd НWdГn1sZ0Ytjt|d НП&|j||gdddd НWdГn1sЮ0Ytjt|d НП(|j||dЬdddd НWdГn1sф0YdS)Nr/r0rcSs|||Srr)rrOrQrrr┌foo1ysz.test_agg_mapping_func_deprecated..foo1r1cSs|||Srr)rrTrQrrr┌foo2|sz.test_agg_mapping_func_deprecated..foo2z+using .+ in Series.agg cannot aggregate andrUrщ)rQ)rOrT)r0r)r1r)rr rDrErW)r:r\r]rZrrr┌ test_agg_mapping_func_deprecatedus 04r_cCsИttdddНГ}ddД}|s`d}tjt|dНП|j||dНWdГn1sR0YdS|j||dН}|а|б}tа||бdS) Nz1/1/2000r@йrcSs|j|j|jfSr)┌hour┌day┌monthrrrrrBМsz2test_series_apply_map_box_timestamps..funcz&Series' object has no attribute 'hour'rUr) rr┌pytest┌raises┌AttributeErrorrrr r!)rr"rBrZr#r$rrr┌$test_series_apply_map_box_timestampsИs, rgcCs\tаdбtаdбg}t|ddН}|jdks.JВ|jddДdd Н}td dgГ}tа||бdS)N· 2011-01-01· 2011-01-02zM8[ns]йr-zdatetime64[ns]cSst|ГjЫd|jЫd|jЫЭSйN┌_й┌type┌__name__rb┌tzrrrrrаrz%test_apply_box_dt64..r rZTimestamp_1_NoneZTimestamp_2_Noneй┌pd┌ Timestamprr-rr r!й┌valsr"┌resr6rrr┌test_apply_box_dt64ЪsrwcCsdtjdddНtjdddНg}t|ddН}|jdks6JВ|jdd Дd dН}tdd gГ}tа||бdS)Nrhz US/Eastern)rprizM8[ns, US/Eastern]rjzdatetime64[ns, US/Eastern]cSst|ГjЫd|jЫd|jЫЭSrkrmrrrrrмrz'test_apply_box_dt64tz..r rzTimestamp_1_US/EasternzTimestamp_2_US/Easternrqrtrrr┌test_apply_box_dt64tzеs■rxcCsXtаdбtаdбg}t|Г}|jdks*JВ|jddДddН}tdd gГ}tа||бdS) Nz1 daysz2 daysztimedelta64[ns]cSst|ГjЫd|jЫЭSrk)rnro┌daysrrrrr╢rz%test_apply_box_td64..r rZTimedelta_1ZTimedelta_2)rrZ Timedeltarr-rr r!rtrrr┌test_apply_box_td64▒srzcCs`tjdddНtjdddНg}t|Г}|jdks2JВ|jddДdd Н}td d gГ}tа||бdS)Nrh┌Mйrriz Period[M]cSst|ГjЫd|jЫЭSrk)rnroZfreqstrrrrrr└rz'test_apply_box_period..r rZPeriod_M)rrZPeriodrr-rr r!rtrrr┌test_apply_box_period╗sr}csЄtddddНаdб}t|ddН}|jdd ДИd Н}tddddНаdб}t|ddН}tа||б|jЗfdd ДИd Н}tttd ГГdgdИrРdnddН}tа||бЗfddД}|j|Иd Н}ИrтtdgdddН}tа||бn|dksюJВdS)Nrhrirr|z Asia/Tokyo┌XXr+cSs|tjабSr)rr┌offsetsZDayrrrrr╔rz'test_apply_datetimetz..rz 2011-01-03csИr |jS|jjSr)rarrrrrr╨rщr┌int64Zint32йr,r-csИrt|jГSt|jjГSr)rJrprrrrrr╒sz test_apply_datetimetz..fщ)r┌tz_localizerrr r!rM┌range)r┌valuesr:r#Z exp_valuesr6rrrr┌test_apply_datetimetz┼s" $rЗcCs<tjtdГtdГddН}t|dtdГdН}|sРd}tjt|d НП"|jd dД|dНWdГn1sh0Y|jd dД|dНdksМJВdS|jddД|dН}tjtdГtdГddН}t|dtdГdН}tа ||бtа |j|jб|аddДб}tdgddtdГdН}tа ||б|Рs.|jt kРs8ndРs8JВdS)NZABBABCDZDCBAT)┌ categories┌orderedr~Zabcdefg)r,r.z&Series' object has no attribute 'lowerrUcSs|абSrй┌lowerrrrrrчrz(test_apply_categorical..rcSsdSйN┌ArrrrrrшrrНcSs|абSrrКrrrrrыrZabbabcdZdcbacSsdSrМrrrrrrЇrщzstring[pyarrow_numpy])rrZCategoricalrMrrdrerfrr r!Zassert_categorical_equalrЖr-r7)rZusing_infer_stringrЖr"rZr#r6rrr┌test_apply_categoricalрs"0rП┌seriesz1-1z1-2cCsвt|ddН}|sXd}tjt|dНП"|jddД|dНWdГn1sJ0YdS|jddД|dН}|аtб}td d tjgddН}|аtб}t а ||бdS) N┌categoryrjz('Series' object has no attribute 'split'rUcSs|аdбdSйN·-rйrKrrrrrrz8test_apply_categorical_with_nan_values..rcSs|аdбdSrТrФrrrrrr┌1)rrdrerfrZastyper7r3rAr r!)rРrr:rZr#r$rrr┌&test_apply_categorical_with_nan_values·s0 rЦcCs8tgtdddНtdН}|jddД|dН}tа||бdS)Nz 2018-01-01r)┌startrr2cSs|SrrrrrrrrzEtest_apply_empty_integer_series_with_datetime_index..r)rr┌intrr r!)rr:r#rrr┌3test_apply_empty_integer_series_with_datetime_indexsrЩcCshttаgdвбdgdН}tgdвdgdН}ddД}|dj||fdН}tgd вdd dН}tа||бdS)N)r0r1rr^щ┌Numbers)┌columns)r1rr1r0r1ZIndicescSs|dj|}|S)NrЫ)Ziloc)Z targetRowZtargetDF┌valrrr┌retrievesz+test_apply_dataframe_iloc..retrieverH)rr^rr1r┌uint64rВ)rr3rЯrrr r!)ZuintDFZindexDFrЮr#r$rrr┌test_apply_dataframe_ilocsrаcCsHtjddНРП$tа|б}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а|а|б|бWdГn1Рs:0YdS)Nr&r'rr4r0йZaxis┌absoluter)r*rР)r)r*)r3r5r4┌absr┌copyr r!Zto_framerЬ┌assert_frame_equalrZunstack┌renameZreindex_like)┌ string_seriesrZf_sqrtZf_absr#r$rrr┌test_transforms* rи┌opcCs\|dvr"|аtjj|ЫdЭdНбtdtgГ}|dvrBt}d}nt}dаdd gб}tj ||d НП|а |dgбWdГn1sЖ0Ytj ||d НП |а |ddЬбWdГn1s─0Ytj ||d НП$|а |gdgdЬбWdГn1Рs0Ytj ||d НП&|а |dg|gdЬбWdГn1РsN0YdS) N)ZffillZbfill┌padZbackfill┌shiftz is successful on any dtype)┌reasonr)ZfillnaZngroup·Transform function failed·|z4not supported between instances of 'type' and 'type'zunsupported operand typerUrлйrН┌B)Zapplymarkerrd┌markZxfailrr7┌ ValueError┌ TypeError┌joinre┌ transform)rйrr"┌errorrZrrr┌test_transform_partial_failureEs, ■ ,.4r╖cCs(ddД}ddД}tdtgГ}d}tjt|dНП|а||gбWdГn1sT0Ytjt|dНП |а||dЬбWdГn1sТ0Ytjt|dНП$|а|g|gdЬбWdГn1s╘0Ytjt|dНП&|а||g|gdЬбWdГn1Рs0YdS) NcSs|Srrrrrr┌noopksz7test_transform_partial_failure_valueerror..noopcSstВdSr)r▓)rlrrr┌ raising_opnsz=test_transform_partial_failure_valueerror..raising_oprrнrUrп)rr7rdrer▓r╡)r╕r╣r"rZrrr┌)test_transform_partial_failure_valueerroris,.2r║cCsrttdГdddН}|аddgб}tddgddgdd Н}tа||б|аd diб}tdgd gdd Н}tа||бdS)NщrБrР)r-r,┌min┌maxrrЪйr.r,r))rrЕrWr r!йr:r#r$rrr┌ test_demoБsr└rBcCst|ГSr)rJrrrrrОrrcCs>|j||dН}|r*|а|б}tа||бn|t|Гks:JВdS)Nr)rrr r!rJ)rзrBrr#r$rrr┌(test_apply_map_evaluate_lambdas_the_sameОs r┴cCsмtаtбП|аddДб}WdГn1s.0Yt|tГrRt|Гt|ГksVJВtаtбП|аtб}WdГn1sА0Yt|tГrдt|Гt|ГksиJВdS)NcSst|ГSr)rnrrrrrаrz+test_agg_evaluate_lambdas..)r rDrErWrLr┌lenrn)rзr#rrr┌test_agg_evaluate_lambdasЫs,(r├┌op_namerWrcCs╛d}|dkrtnd}tj||dНП"t||ГddДГ}WdГn1sJ0Yt||ddЬГ}tа||бtjt|dНП|аddДб}WdГn1sд0Ytа||бdS) Nzcannot aggregaterWrUcSst||dgddgdНSйNr1r·x^2йr.йrrrrrr▒rz)test_with_nested_series..r1)rr╞cSst||dgddgdНSr┼r╚rrrrr╖r)rEr rD┌getattrrrеrW)r9r─rZ┌warningr#r$rrr┌test_with_nested_seriesиs ",r╦cCs<|аб}|аddddddДdddДd d Ьб}tа||бdS)N┌count┌mean┌stdr╝cSs |аdбS)Ng╨?йZquantilerrrrr─rz)test_replicate_describe..ZmediancSs |аdбS)Ngш?r╧rrrrr╞rr╜)r╠r═r╬r╝z25%z50%z75%r╜)Zdescriberr r!)rзr$r#rrr┌test_replicate_describe╗s° r╨cCs>|аddgб}t|аб|абgddg|jdН}tа||бdS)N┌sumr═r+)rWrr╤r═r,r r!)rзr#r$rrr┌test_reduce═s¤r╥z how, kwdscCsЦtgdвГ}t||Гd i|дО}|j}||ks2JВt||Гgdвfi|дО}tddddЬГ}tа||бt||ГddddЬfi|дО}tа||бdS) N)r0r1N┌size)r╙r╠r═ч@ч@g°?r╠r═)r╙)rr╔r╙r r!)┌how┌kwdsr:r#r$rrr┌test_non_callable_aggregates╪sr╪cCsNtdgdГ}|jdddДddДg|dН}tgdвgdвd Н}tа||бdS) Nr^rr╤cSs|абSrйr╤rrrrrЄrz3test_series_apply_no_suffix_index..r)щr┌r┌)r╤rrr╟)rrr r!)rr:r#r$rrr┌!test_series_apply_no_suffix_indexяsr█zdti,expl,b/r╟rбrБrjr@· 2020-01-01r`┌tsr╛┌awarecCs:|r|аdбj}n|j}t|ГаddДб}tа||бdS)N┌UTCcSstddgГS)Nr0r1r╚rrrrrrzCtest_apply_series_on_date_time_index_aware_series..)rДr.rrr rе)Zdtir6r▐r.r#rrr┌1test_apply_series_on_date_time_index_aware_series°s rрzby_row, expected)Fr0cCsJttjdtjdНtddddНdН}t|jГjddД|d Н}tа||бdS) Nr@rjr▄r▀)rrpr╟cSsdSr=rrrrrrrzCtest_apply_scalar_on_date_time_index_aware_series..r) rr3┌aranger8rr.rr ┌assert_equal)rr$rРr#rrr┌1test_apply_scalar_on_date_time_index_aware_seriess■rуcCs~ddg}tа|б}t|Гjtj|dН}tаt|Г|бdtjtjtjg}tа|б}t|Г}|jtj|dН}tаt|Г|бdS)Nz00:00:01z00:00:02r) rrZto_timedeltarrr r!r3rAZNaT)rZlist_of_valid_stringsrOrTZlist_of_stringsr"rrr┌test_apply_to_timedelta#s rфz ops, namesr╤r═zhow, kwargsc sШtЗfddДt||ГDГГ}d|_|dkr.tnd}ddа|бЫdЭ}tj||dНП&tИ|Г|fi|дО}WdГn1s~0Ytа||бdS) Ncsi|]\}}||ИГУqSrrй┌.0r,rййrзrr┌ @rz/test_apply_listlike_reducer..rРrWzusing Series.[rо·]rU) r┌zipr,rEr┤r rDr╔r!) rз┌ops┌namesr╓rXr$┌warnrZr#rrчr┌test_apply_listlike_reducer1s4rюrыrНrпc sКtЗfddД|абDГГ}Иj|_|dkr.tnd}d}tj||dНП&tИ|Г|fi|дО}WdГn1sp0Ytа||бdS)Ncsi|]\}}||ИГУqSrrrхrчrrrшXrz/test_apply_dictlike_reducer..rWzusing Series.[sum|mean]rU)r┌itemsr,rEr rDr╔r!) rзrыr╓rXrr$rэrZr#rrчr┌test_apply_dictlike_reducerIs4rЁr4rвcsjtjddНПJtЗfddД|DГddН}||_Иj||dН}tа||бWdГn1s\0YdS)Nr&r'csg|]}|ИГСqSrr)rцrйrчrr┌ mrz3test_apply_listlike_transformer..r0rбr)r3r5rrЬrr rе)rзrыrьrr$r#rrчr┌test_apply_listlike_transformeras rЄz ops, expectedcCs|Srrrrrrrvrr/cCs|абSrr┘rrrrrwrr╗cCs*tgdвГ}|j||dН}tа||бdSйNr/rйrrr rт)rыr$rr"r#rrr┌test_apply_listlike_lambdass rїcsltjddНПLtЗfddД|абDГГ}Иj|_Иj||dН}tа||бWdГn1s^0YdS)Nr&r'csi|]\}}||ИГУqSrrrхrчrrrшНrz3test_apply_dictlike_transformer..r)r3r5rrяr,rr r!)rзrыrr$r#rrчr┌test_apply_dictlike_transformerБs rЎcCs|SrrrrrrrЧrrcCs|абSrr┘rrrrrЪrcCs*tgdвГ}|j||dН}tа||бdSrєrЇ)rыrr$r"r#rrr┌test_apply_dictlike_lambdaУsrўcCsАtdtdГittdГddНГ}|jаddДб}tdtjtjgddtjggdвgttdГd dНttdГddНd Н}tа||бdS)Nrrr+cSs"tt|dГtt|dГddНГS)Nr0┌yr+)rrЕrrrrrrзrz0test_apply_retains_column_name..ччЁ?)r∙r·r╒r°)rЬr.) rrЕrrrr3rAr rе)rZdfr#r$rrr┌test_apply_retains_column_nameдs¤r√cCsFtddtggdвdН}|аtб}ttttggdвdН}tа||бdS)Nr┌string)rOrTrQr╟)r┌floatrrnrШrJr r!r┐rrr┌test_apply_type░s r■cCsVtgdвgdвgГ}|аddДб}tddgddgd d gtjdgdЬГ}tа||бdS) Nr/)r^rЪr╗rОcSst|ГSrr╚rrrrr╗rz6test_series_apply_unpack_nested_data..r·g@r╒g@r╘g@rО)rr0r1r)rrrr3rAr rе)r"r#r$rrr┌$test_series_apply_unpack_nested_data╕s$r )L┌numpyr3rdZpandasrrrrrrrrrZpandas._testingZ_testingr Zpandas.tests.apply.commonr Zfixturerr%r<r>r▒ZparametrizerFrNr[r_rgrwrxrzr}rЗrПrArЦrЩrаrиr╖r║r└rJr┴r├r╦r╨r╥r╪r█Z DatetimeIndex┌repeatrсr8rрZonesrуrфr╤r═┌arrayrюrЁr4rгrЄrїr6rЎZfrom_arraysrЕrўr√r■r rrrr┌s$ *■ & # ■ ■ ¤·√■ №■ ■№■ ■№■ ■■ №■ "■√■