a щ┤лj├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)┌ categoriesZorderedr~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щrJ)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$ *■ & # ■ ■ ¤·√■ №■ ■№■ ■№■ ■■ №■ "■√■