gL i ddlmZmZddlZddlZddlmZddlZddlm Z m Z m Z m Z m Z ddlmZddlmZdZdZdZd Zd Zej0j3d e gd gd dgdfe gdgddddgfe ddgidgfe gdeddddgfe gdgddgdfe edgdgddgdfe gdgdggd ddgfggd!"d#Zd$Zd%Zej0j3d&d'd(d)d*gd+Zd,Z d-Z!d.Z"d/Z#d0Z$d1Z%d2Z&d3Z'd4Z(d5Z)d6Z*ej0j3d7d8d9gd:Z+d;Z,d<Z-d=Z.d>Z/d?Z0ej0j3d@d8d9gdAZ1dBZ2dCZ3dDZ4dEZ5dFZ6dGZ7dHZ8dIZ9dJZ:dKZ;dLZ.stores e7DataFrameGroupBy.apply operated on the grouping columnsmatchr)rrdr) rlistrange reset_indextmassert_produces_warning FutureWarninggroupbyapplypd RangeIndexassert_frame_equal)dfrmsgexpected_valuers @r7test_apply_func_that_appends_group_to_list_without_copyr,s 1DrOb01# > J J LB F DC # #M =) 7!!%()#(sRx( ab0IN&)^4 ))s !CCcfgd}tdgdittj|d}|j |j j j}|j |j j jd}tj||y)N z2011-05-16 00:00z2011-05-16 01:00z2011-05-16 02:00z2011-05-16 03:00z2011-05-17 02:00z2011-05-17 03:00z2011-05-17 04:00z2011-05-17 05:00z2011-05-18 02:00z2011-05-18 03:00z2011-05-18 04:00z2011-05-18 05:00value /?ݓ?Q?j?r1r2r3r4r1r2r3r4 date_timenamerc"|jSridxmaxxs rz'test_apply_index_date..Ns qxxzr) rrr& to_datetimer$rrr:r%r!r()using_infer_stringtsr)expectedresults rtest_apply_index_daterC*s B    BNN2&[9# B&zz"((--(//1H ZZ & , ,-A BF&(+rcgd}t|Dcgc]}|jc}ddg}gd|d<tgdd}tgd | }d }t j t | 5|jdd jd}dddt j|ycc}w#1swY%xYw)Nr.rtimerr0r/)z 2011-05-16z 2011-05-17z 2011-05-18r6)z00:0002:00rGrrrF group_keysc4|d|djS)NrEr/r9r;s rr=z.test_apply_index_date_object..wsai' 1 1 34r) rsplitrr r!r"r#r$r%assert_series_equal)r@rowr)exp_idxrAr*rBs rtest_apply_index_date_objectrORs B 2.CCIIK.8H IB BwK>VLG1AH CC # #M = Fu5;; 4  68,-/$  sB52$B::Cc tgdgddddg|rdnd}tjjd djd dgd d |g }d }t j t | 5jjDcgc] }t|c}d }dddjfd}t j||ycc}w#1swY9xYw)NarRbrSrR?@@@g@keydatarZr[rFstrobjectr float64axiskeys+DataFrame.groupby with axis=1 is deprecatedrr`c"jddSNr ilocr<r)s rr=z$test_apply_trivial..s r) rr&concatrgr!r"r#r$dtypesr\r%r(r?dtyperAr*r<gbrBr)s @rtest_apply_trivialrn|s )3LM B(EXEyy"''!"+rwwqr{3!9eBTUH 7C # #M == ZZ3AQ3!Z <= XX+ ,F&(+4==s1C C  CCC'c tgdgddddg|rdnd}tjgd d |g }d }tjt | 5j jDcgc] }t|c}d d}dddjfd}tj||ycc}w#1swY9xYw)NrQrTrYrZr[rFr\r]r r^r_rbrT)r`rIcSrrhs rr=z)test_apply_trivial_fail..sr) rr&rir!r"r#r$rjr\r%r(rks @rtest_apply_trivial_failrrs )3LM B(EXEyy"bE0BCH 7C # #M =N ZZ3AQ3!Z MN XXl #F&(+4NNsC1C  CCCzdf, group_names)r r r )rRrRrRrScrRrSr rsrtrrr r )rr rr r rR)r r r rsrsr r rsrsr rsrtr rsrt)ry )rRtwo aaabbbcccc) rtr{rtr}r|rsr rr|r{) r{rrsrsrsr~ryr}rsry)rRBCrRrSru)rsrsrtrF)GH2936zGH7739 & GH10519GH10519GH2656GH12155GH20084GH21417)idsc g fd} fd} fd} fd} fd}|||||fD]V} dd=d}tjt|5|jdd j |ddd |k(rVJy#1swYxYw) NcZj|j|jSr)rr7copyrnamess rf_copyz/test_group_apply_once_per_group..f_copys UZZ zz|rc>j|j|Srrr7rs rf_nocopyz1test_group_apply_once_per_group..f_nocopys UZZ  rc<j|jyNrrrs rf_scalarz1test_group_apply_once_per_group..f_scalars UZZ rc<j|jyrrrs rf_nonez/test_group_apply_once_per_group..f_nones UZZ rcZj|jtdgdgdS)Nr rv)rr7rrs r f_constant_dfz6test_group_apply_once_per_group..f_constant_dfs' UZZ s!-..rrrrRFrH)r!r"r#r$r%) r) group_namesrrrrrfuncr*rs @rtest_group_apply_once_per_grouprsN E   !/ 8V]C$ !HG  ' ' S A : JJsuJ - 3 3D 9 : ### $ : :s #A>>B c>d}tgdgddgd}d}tjt|5|j dd j d ddd|j jjd }||k(sJy#1swY:xYw) Nrs)rrrrr r r r )02468101214)group_by_column test_columnrrrrFrHctdS)Nfunction_called)printr)s rr=z2test_group_apply_once_per_group2..s u./rr) rr!r"r#r$r% readouterroutcount)capsysrAr)r*rBs r test_group_apply_once_per_group2rsH 7F :  B DC # #M =  $ 7== /     $ $ * *+< =F X    s $BBctgdtdd}d}d}d}tjt|5|j dd j |}dddtjt|5|j dd j |}dddtjy#1swYfxYw#1swY,xYw) N)rrr rt)ArSc|Srrqrs rslowz,test_apply_fast_slow_identical..slow s rc"|jSrrrs rfastz,test_apply_fast_slow_identical..fast szz|rrrrFrH)rrr!r"r#r$r%r()r)rrr*fast_dfslow_dfs rtest_apply_fast_slow_identicalrs q2 3B DC # #M =@**SU*399$?@ # #M =@**SU*399$?@'7+ @@@@s#C#CCCrc|Srrqr;s rr=r=s!rc |ddSrrqr;s rr=r=s !A$rc&|jdS)NFdeeprr;s rr=r=s!&&e&$rc&|jdS)NTrrr;s rr=r=s!&&d&#rctgdgdgdd}d}tjt|5|j dd j |}dddtj |y#1swY xYw) Nr rsrsrsr rsrtr{)r|r}r~ry)grRrSrrrFrHrr!r"r#r$r%r()rr)r*rBs r5test_groupby_apply_identity_maybecopy_index_identicalrsl L|L MB CC # #M =?CE288>?&"%??s #A55A>c.ttjjdj dgdd}|j ddj }|j }tj||tdgd i}|jd z |d <|jd jjj}|jd d jj}tj||y)Nrsr}onerrthreerrfoo1foo2c|Srrqr;s rr=z-test_apply_with_mixed_dtype..8srr rcc1)r rsr}r}ryrVc2Fas_index)rnprandom default_rngstandard_normalr%rjr!rLrr$meanr r)r)rBrAresult1result2s rtest_apply_with_mixed_dtyper0s II))!,<.YsAFF1Irc$|jdSrrr;s rr=z-test_groupby_as_index_apply..[sAFF1Ir))rr)rrsr r rsr{)r rr rs)rsr )rtr{rabcder rtr{r|rr)rrIc|Srrqr;s rr=z-test_groupby_as_index_apply..jsar)rrr$rrrr!assert_index_equalr"r#r%r from_tuplesr)r)g_asg_not_asres_as res_not_asexpr* res_as_applyres_not_as_applyexp_not_as_applytp exp_as_applyindress rtest_groupby_as_index_applyrDs 5)!H  B ::i$: /Dzz)ez4H YYq\  Fq!''J  C&#&*c* CC # #M ==zz"56<< = # #M =E#>>*=>DDE "--.NO )B))"Y4EFL, 5*,<= W C QFQFQFQFQF;3 GB CC # #M =WjjUuj=CCKPVVW#s#'==EEWWs$H/H<!/I /H9<I Icp|jddg}d}d}d}d}tjt|5|j |}dddj j dk(sJd}tjt|5|j |}dddj j dk(sJd}tjt|5|j |}dddj j d k(sJy#1swYxYw#1swYxYw#1swY=xYw) NrrcH|j}d|j_|SNstat)describerr7rrBs rdescz.test_apply_concat_preserve_names..descqs!"  rcd|j}d|j_|dt|}|Sr)rrr7lenrs rdesc2z/test_apply_concat_preserve_names..desc2vs/!"  #e*% rc~|j}dt|d|j_|dt|}|S)Nstat_d)rrrr7rs rdesc3z/test_apply_concat_preserve_names..desc3}s>!$CJq>2  #e*% rrr)rrr)rrN)r$r!r"r#r%rr) three_groupgroupedrrr r*rBrresult3s r test_apply_concat_preserve_namesr ns!!3*-G  DC # #M =%t$% <<  !3 33 3 CC # #M ='--&' ==  "4 44 4 CC # #M ='--&' ==  "2 22 2%% '' ''s#D D D,D D),D5ctd}tdd}ttjj dj d|}|j dd }|j|}t|tsJt|d rJtj|j|jy) Nctjd5tj|}dddt|||j z dS#1swY)xYw)Nignore)invalid)r/demeanedlogged)rerrstatelogrr)piecers rfz%test_apply_series_to_frame..fsR [[ * #VVE]F #)= P   # #s AAz1/1/2000r)periodsrsrc|jSrmonthr;s rr=z,test_apply_series_to_frame.. 177rFrHr7)r r rrrrr$r% isinstancerhasattrr!rr)rdrr@r rBs rtest_apply_series_to_framer s  Z -B  %%a(88=R HBjj*uj=G ]]1 F fi (( (vv&& &&,,1rc|jddgdjt}|jjdddk(sJy)Nrrrrsrr)r$r%rrr)r)rBs r test_apply_series_yield_constantr#sC ZZc #C ( . .s 3F <<  bq !Z // /rcd}tjt|5|jddgj t }dddt tsJ|jJ|jddgddgj t }t |tsJ|jJy#1swYuxYw)NrrrrrD) r!r"r#r$r%rrr r7)r)r*rBs rtest_apply_frame_yield_constantr&s CC # #M =3S#J'--c23 ff %% % ;;   ZZc #S#J / 5 5c :F ff %% % ;;  33s 'B99Cc|jddg}d}tjt|5|j t }ddd|j d}tjj|jtj|j|jy#1swYqxYw)Nrrrrr) r$r!r"r#r%rrrrassert_numpy_array_equalvaluesr)r r*rBrAs rtest_apply_frame_to_seriesr+sjj#s$G CC # #M =$s#$}}s#H&,,7 x? $$s B77Cc|jddgd}d}tjt|5|j t }ddd|j jdtjijd }tjj|jtj|j|jy#1swYxYw) NrrFrrrrrFr%)r$r!r"r#r%rrrenamernandroprrr(r)r*s r)test_apply_frame_not_as_index_column_namer0sjj#sej4G CC # #M =$s#$}}%%sBFFm%<AA#ANH&,,7 x? $$s C&&C/cd}fd}ttjjdj dddtjjdj dddtjjdj ddd}t jt| 5jd j|}dddjd d j|}t j|d |jd k(sJy#1swYVxYw)Nc||jddjjjddS)Nrrrs)r$sum sort_valuesrgrs rtransz-test_apply_frame_concat_series..transs5}}S!#&**,88:??CCrc|jj|jd}|jj j ddS)Nrrs)r$reindexrr3r4rg)rr r)s rtrans2z.test_apply_frame_concat_series..trans2sE-- 5;; 7 <={{}((*//33rrsrr|rrrrrrrF) check_names) rrrrintegersrr!r"r#r$r%rLr7)r5r8r*rBrr)s @rtest_apply_frame_concat_seriesr=sD4 &&q)221a>&&q)221a>&&q)99$?  B DC # #M =.C&&u-. **S/#  $ $V ,C63E: ;;#   ..s >!D44D=c|jdd}|jd}|jd}tj||y)Nc|jSrrr;s rr=z&test_apply_transform..rrFrHc |dzSrrqr;s rr=z&test_apply_transform..s QUrc |dzSrrqr;s rr=z&test_apply_transform..s 1q5r)r$r% transformr!rL)r@r rBrAs rtest_apply_transformrCsCjj*uj=G ]]? +F  1H68,rc|jddg}d}|j|}|D].\}}tj|j|||0y)Nc|jSryearr;s rr=z,test_apply_multikey_corner..s rc|jSrrr;s rr=z,test_apply_multikey_corner..s 177rc*|jdddS)Nrr4rs rrz%test_apply_multikey_corner..fs  %bc**r)r$r%r!r(loc)tsframer rrBrZrs rtest_apply_multikey_cornerrNs[oo/1BCDG+]]1 F9 U fjjoqx89rrITFctgdtdd}d}tjt|5|j d|j d}ddd|jgd }|r,tjgd |jgddg |_ tj|y#1swYaxYw) N) r r r rsrsrsrtrtrtr)rZr/rrrZrHc |jddSrrfr;s rr=z'test_apply_chunk_view..s!&&QSRS*r)rr rtr{r}r~)r r rsrsrtrtr) rrr!r"r#r$r%taker from_arraysrr()rIr)r*rBrAs rtest_apply_chunk_viewrSs 6qJ KB CC # #M =VEj9??@TUVww)*H#//  0  &(+VVs $B;;Cc tgdgdtdddd}|jddg}d}tjt | 5|j d dddy#1swYyxYw) N) r r r r r r rsrsrsrs) rrrr r r rrr r r)r7name2r/r7rVrrc(|jddS)Nr/TinplacerKr;s rr=z4test_apply_no_name_column_conflict..s gt Dr)rrr$r!r"r#r%)r)r r*s r"test_apply_no_name_column_conflictrZ ss 231b"%  Bjj&'*+G CC # #M =F DEFFFs A++A4ctgdtjgddtjddd}d}d}t j t | 5|jd d j|}ddd|j}tjgd d|d<t j|y#1swYKxYw)NrUrUrUrVrVrVrrsrU@rruvc|d}||jz |j|jz z |d<|SNr_v2minmaxrr_s rrz#test_apply_typecast_fail..f&9 #J1557{quuw'89d  rrrrFrHg?r rb) rrtilearanger!r"r#r$r%rr()r)rr*rBrAs rtest_apply_typecast_failrls /!,3$  B DC # #M =<CE288;<wwyHWW]A.HTN&(+ <.f@rgrrrrFrHrhrb) rrRrrrjrkr!r"r#r$r%rr()rr)rr*rBrAs rtest_apply_multiindex_failro5s  " "$68J#K LE /!,3$   B DC # #M =<CE288;<wwyHWW]A.HTN&(+ <.Ps qvvrFrHc |dzSrrqr;s rr=z#test_apply_corner..Ps QRUVQVrrs)r$r%r!r()rMrBrAs rtest_apply_cornerrsOs9 __-%_ @ F F WF{H&(+rctgdgdgdd}d}d}d}tjt|5|j d j |}dddtjt|5|j d j |}dddtj y#1swYdxYw#1swY,xYw) N)rri,)rRrSrurur)id_fieldcategoryr/cj|jddk(r|jS||jdk(SNrr ru)shaperrwr;s rfilt1z&test_apply_without_copy..filt1as0 771:?668OQZZ3&' 'rcN|jddk(r|S||jdk(Sry)rzrwr;s rfilt2z&test_apply_without_copy..filt2gs* 771:?HQZZ3&' 'rrrrvr)r[r{r}r*rArBs rtest_apply_without_copyr~Us ,,!  D( ( DC # #M =9<< +11%89 # #M =7j)//67&(+ 9977s!B==!C =C C test_seriesc4tddgddgddggddggd}|rn|jdd}|jdd jd }|j }|j }t j ||yd }t jt| 5|jdd jd}dddjd}|jd}t j||y#1swYBxYw)Nr<poXY)r rsrsrrrF)levelrIc|Srrqr;s rr=z.}srrrrHc|Srrqr;s rr=z.qr) r set_indexr$r% sort_indexr!rLr"r#r4r()rr)serrBrAr*s r*test_apply_with_duplicated_non_sorted_axisrus  sc3Z#s,sCj  Bll3$17==kJ""$>># vx0G  ' ' S A JZZZ6<<[IF J##C(>>#& fh/  J Js )$DDcgd}gd}tddgdz|d|}t||d }d }|jd d jj |}t j ||y)Nrr r rsrsGroup1Group2rsrr/rr/rr7c|jtj|jj |jj dzSre)r7rrkrrdrer;s rreindex_helperz1test_apply_reindex_values..reindex_helpers5yy177;;=!''++-!2CDEErrFrH)rr r$r/r%r!rL)r)indicesr)rArrBs rtest_apply_reindex_valuesrsr FG h1A5Gw WBfG':HFZZEZ 2 8 8 > >~ NF8V,rcd}tjjdjdd|}t |tjjdj |gd|dzzd}|j d d }d }d }tjt|5|j|}ddddvsJy#1swYxYw)Nr9rsrr)size)foobarbazquxr{)rZvalue1value2rZFrHc|ddz|d<|S)Nrrsvalue3rqrs rrz"test_apply_corner_cases..fskAo( rrrr) rrrr<rrr$r!r"r#r%)Nlabelsr)r rr*rBs rtest_apply_corner_casesrs A YY " "1 % . .q#A . >F ii++A.>>qA2a1f=  Bjj5j1G DC # #M ="q!" v  ""s *C  Cctddgdgdzddgd}d}tjt|5|j d gj d }dddt j|j|_d}tjt|5|j d gj d }dddtjd d tgd gdt jdgdzd}d}d}tjt|5|j dj |d}ddd|j}|j|_ tj||d}tgdgdgdgdd}|j}t j|j|_d}tjt|5|j dj |j }dddtjt|5|j dj |j }dddtj||y#1swY?xYw#1swYxYw#1swY[xYw#1swYxYw#1swYSxYw)Nr rsz 2017-03-02rinf)NumberDateStrrrrc |jdSrrfr;s rr=z;test_apply_numeric_coercion_when_datetime..s!&&)rc |jdSrrfr;s rr=z;test_apply_numeric_coercion_when_datetime..sq rr)r)r3rz12:31:22rt)rrTc(|jddgS)Nrrrfrs rget_Bz8test_apply_numeric_coercion_when_datetime..get_Bsvvay#rrrctgdt}dt|jvr5t ||jdk(j j d|d<dt|jvrjt ||jdk(j j d|d<t ||jdk(jj d|d<|S) N)p1p2useTime)rrlstep1rrstep2rr)r r]rStater\Machiner)oTime)toolrs r predictionsz>test_apply_numeric_coercion_when_datetime..predictionss2&A d4::& &Dw!67??FFqIJCI d4::& &Dw!67??FFqIJCI djjG&;!rrL Timestamprrrrrr)r)r*rArBrrdf1df2s r)test_apply_numeric_coercion_when_datetimersg q6L>A#5uenM B DC # #M =E::xj)//0CDEnnRWW%BG CC # #M =CXJ'--.ABC6%=(5/: !2",,z:R9SVW9WX B  DC # #M =3C&&u-c23ttHTTHN68, '9K2   C ((*Csyy)CI CC # #M =<;;u%++K8;;< # #M =:U#))+699:8V,aEECC336<<::s;#J!#J"2$J/=+J< +KJ"J,/J9<KKctgdtjdgdzd}|j|jz |d<d}t j t |5|jdjd }dddtgdtjd d gdztjdgdzd jd}t j|y#1swYoxYw) Nr:z2017-02-01 00:00:00rt)clientidrtime_delta_zerorrrc~t|jj|jjdS)N) clientid_ager)r rrdr)ddfs rr=z?test_apply_aggregating_timedelta_and_datetime..s/!$!4!4!8!8!:CLLDTDTDVWrrr%)rrr) rr datetime64rr!r"r#r$r% timedelta64rr(r)r*rBrAs r-test_apply_aggregating_timedelta_and_datetimers ''<=>B  B KK"++5B CC # #M = J'--   '^^As34q8]]#89:Q>  i   &(+  s !"C11C:c"ddgddgddgddgddgg}t|d d gtjd d  }|jd j }tgdgdd}|j d dt j||y)Nrrrrrr(2NameValuez 2020-09-01z 2020-09-05rr:)rrZ)rrTrX)rr& date_ranger$r3rr!r()r[r)rBrAs r test_apply_groupby_datetimeindexrs"IRy3)c2Yb BD  vw'r}}\.func_with_no_date-ssAhrc4ttdddddS)Nr rsrSru)r rrs rfunc_with_datez+test_time_field_bug..func_with_date0sHT1a0q9::rrrrRbyrursrrnsr)rrrnowr!r"r#r$r%rr7r&ras_unitr() nnr)rrr*dfg_no_conversiondfg_no_conversion_expecteddfg_conversiondfg_conversion_expecteds rtest_time_field_bugr%sX 59!ER(,,.!EF GB ; DC # #M =JJJ3%J0667HIJ!*C8A3!?,/$$) CC # #M =Du-33NCD'll4A&..t41=aS*-!!&+-GH.*AB/"FJJ DDsE" #E'#E3'E03E<ctgdgdgdgdd}|jddg}|jddgd }d |jfd y) N) rRrRrRrSrSrSrRrRrRrSrSrS) rururrrerururrrr) g?rsrtr{r|r}rsr{r}ryr rs) gffffff@ryrr ryr~r}r|r{rt)group1group2weightr/rrT)rsortcTtj||zjd}|S)Nrt)rarrayrepeat)r/rrs rnoddyz7test_gb_apply_list_of_unequal_len_arrays..noddyRs$hhuv~&--a0 rc>|j|jSr)r/r)r<rs rr=z:test_gb_apply_list_of_unequal_len_arrays..\suQWWahh7r)rrr$r%)r) df_groupedrs @r(test_gb_apply_list_of_unequal_len_arraysrEs` RR<>   B x* +B8X"6TBJ78rctgdgdd}d}d}tjt|5|j dj |}dddt}tj |y#1swY*xYw)Nrx)ryr~r{r|)r random_varscyrrqr;s r test_funcz.test_groupby_apply_all_none..test_funcds rrrrr)test_dfrr*rBrAs rtest_groupby_apply_all_noner_su< MNG  DC # #M =<*00;<{H&(+<.test_funcss& 771:>vvq"grrrrr rrsrrtr) rr!r"r#r$r%rrRr() test_df1test_df2rr*rrindex1index2 expected1 expected2s rtest_groupby_apply_none_firstr nsGL,GHHL,GHH DC # #M =>""8,229=> # #M =>""8,229=>  # #aVaV$4Xt>>>s!E !EEEc rtddgddgd}|jd}d}tjt|5|j d }dddt dgd tjddgdggddg jd }tj|y#1swYZxYw) Nrr filledempty)r/rrrrc,||jdk7dS)Nr r/)r/rs rr=z7test_groupby_apply_return_empty_chunk..sE%++2B,CG,Lrr/r)r7r) rr$r!r"r#r%r r from_productr/rL)r)rr*rBrAs r%test_groupby_apply_return_empty_chunkrs aVx.AB CB ZZ F CC # #M =NLMN  %%x 1# &wo $w- H68,NNs B--B6c0tdjgdgdd}|jdd}|jd}tgd gd d }t j |||j d }t j ||y) Nza a brwr{r}r|r:rFrHc(||jz Srr3r;s rr=z-test_apply_with_mixed_types..s1quuw;r)gUUUUUU?gUUUUUU?r )皙?g333333?rUrrc(||jz Srrr;s rr=z-test_apply_with_mixed_types..sq1557{r)rrKr$rBr!r(r%)r)rrBrAs rtest_apply_with_mixed_typesrsy yyI JB 35 )A [[. /F4?KLH&(+ WW* +F&(+rcntdddgitddg}d}tjt|5|j dj d}dddttdgtdggtddgd}tj|y#1swYPxYw) NrRr rsrrrc|jSrrrs rr=z*test_func_returns_object..s rr6) rrr!r"r#r$r%r rLrs rtest_func_returns_objectrs C!Q=q!f 6B CC # #M =:C&&'89:uaSz5!:.eQF6MNH68, ::s "B++B4group_column_dtlikec0tdg|gd}d}tjt|5|j dj d}dddtdgt dgdd d g }tj|y#1swY=xYw) NrrvrrrRc tdgdgS)Nspam*rrr;s rr=z+test_apply_datetime_issue..s1Mrr!r\rlr7r"rF)rr!r"r#r$r%rr()rr)r*rBrAs rtest_apply_datetime_issuer$s %(;'<= >B CC # #M =OC&&'MNO&5%C#HSURVWH&(+ OOs "B  Bc *ttjdtjdtjdtjdtjddddddddddddddd}d}d}tjt | 5|j d j|d }dddtdgtjd gd d }tj|y#1swYExYw)Nz2015-02-24 00:00:00)rr rsrtr{zsome UA stringzanother UA string17661101)day userAgentuserIdc t|jDcic]#\}}||jjd%c}}Scc}}wr)r items value_countsr)r)russ rmost_common_valueszEtest_apply_series_return_dataframe_groups..most_common_valuess< K1q!..*0033KLLKs(A rrr'r)z 2015-02-24r6r) rr&rr!r"r#r$r%r DatetimeIndexrL)tdfr.r*rBrAs r)test_apply_series_return_dataframe_groupsr1s << 56<< 56<< 56<< 56<< 56 $##&#    C4M DC # #M =HU#))*<=hGH  B,,l^%HxH68, HHs !$D  Drwc ,ddgdz}|r:tj|gd}tjgdgdd}gd}ntddgd }d d g}t ||d | }t t j d|ttdttddjddg}|jddjd}tj|||jjddgk(sJy)Nr rsr|rw) categoriesr)r3r7)rrr6rr4)rr%rr)rrrr%rF)observedc"|jSrrr;s rr=z-test_apply_multi_level_name..s QUUWr)r& CategoricalCategoricalIndexrrrrkrrrr$r%r!r(rr)rwrSexpected_indexexpected_valuesrAr)rBs rtest_apply_multi_level_namer;s A A NN1 3,,Y9SVW%1vC0r(O4NH iim!$uRy/U2YP ic ZZeZ , 2 23D EF&(+ 88>>c3Z '' 'rc tjtjdddddddftjdddddd d ftjd ddd d ddftjdddd dddfggd}d}t j t |5|jdjdj}ddd|r$tjtjnt}ttjd||tj |ggd}t j"|y#1swYxYw)Nir reddarkrrgreenstormyrs9ibluebrightrtricalmr{potato) observationcolormood intensityscorerFrrrGc |jdSrrfrs rr=z;test_groupby_apply_datetime_result_dtypes.. sqvvayr)na_valuezdatetime64[ns]r)r from_recordsr&rr!r"r#r$r%rj StringDtyperr.r]r rlint64rL)r?r[r*rBrlrAs r)test_groupby_apply_datetime_result_dtypesrPs+  ! ! \\$1 %ufa = \\$1 %w!S A \\$1 %vxD A \\$1 %vvq( C  G D DC # #M =Ig&,,-@AHHI/ABNNBFF +vE " #UE288UCDH68,IIs #,EErabcrt2020r%)rfreq))rRr)rRr )rSrctgdgdd|}d}tjt|5|j dd j d }dddtj |y#1swY xYw) Nr r rs)rr rrrrrrFrHc|Srrqr;s rr=z8test_apply_index_has_complex_internals..$rrr)rr)r*rBs r&test_apply_index_has_complex_internalsrWsp Y;5 IB CC # #M =JG6<<[IJ&"%JJs $A55A>zfunction, expected_valuesc6|jjSr)rto_listr;s rr=r=+s177??$rcHt|jjSr)setrrYr;s rr=r=,s3qww()rcHt|jjSr)tuplerrYr;s rr=r=-s5*+r)rr rsrtcZtt|jjSr)dict enumeraterrYr;s rr=r=/sd9QWW__%678rc~t|jjDcgc] \}}||i c}}Scc}}wr)rarrY)r<nis rr=r=3s,Iaggoo6G,HI&1a1vIIs9c(tgddg}d}tjt|5|j dj |}dddt |tddgd }tj|y#1swY:xYw) N)rrrrrrFrrrrr6r rr!r"r#r$r%r rrL)functionr:r)r*rBrAs r1test_apply_function_returns_non_pandas_non_scalarrh(s}$ '( .fctDsSz  ((**r)rRrRrSnoner rsrtr"rrrrUrVrW)rRrSrmr6r) rrr.r!r"r#r$r%r rrL)rlr)r*rBrAs r'test_apply_function_returns_numpy_arrayrnBs+ 01a7HI JB CC # #M =,C&&s+, scURVVH%U3EC-PH68, ,,s !B22B;rgc|jSrrgrs rr=r=Rs rc&|jdzdz Srerrps rr=r=RsbhhQRlUVFVrcptgddg}d}tjt|5|j dj |}dddt tgdtgdtdd ggtgd d  }tj|y#1swY^xYw) N) r rsrsrsr rsrtr rtr idrFrr)rr{r~r)r rsrtr|r}ryrwr6rrf)rgr)r*rBrAs r test_apply_function_index_returnruRs 1D6 BB CC # #M =2D!''12 | eL15!Q=AID)H68, 22s !B,,B5ctgdgdd}|jddjd}tgdgd d}tj||y) Nr)r rsr{r{r|r}rrFrc"|jSr)rr;s rr=zAtest_apply_function_with_indexing_return_column..hs r)rrr)rWrXrX)rr$r%r!r()r)rBrAs r/test_apply_function_with_indexing_return_columnrx`s^ @&  B ZZZ / 5 56H IF+# H &(+rudfc"|jSrrr;s rr=r=ts rcB|jjdS)Nc |dzSrerq)ys rr=z..ts Ar)rr-r;s rr=r=tsaffhooo&Frctddgddgd}d}tjt|5|j d| j |}ddd|j j |j| j |}|r7jjdk(sJ|jjdk(sJyjjdk(sJ|jjdk(sJy#1swYxYw) NrRrSr rsr"rrrrH) rr!r"r#r$r%rrrnlevels)rIryr)r* df_result series_results rtest_apply_result_typerrs #s1a&1 2B CC # #M =FJJszJ:@@E FDDLL*L=CCCHM&&!+++""**a///&&!+++""**a///FFs #DD ctgdgdd}d}tjt|5|j ddj d }dddtjt|5|j ddj d }dddtj y#1swYgxYw#1swY,xYw) N)rsr rsrwr"rrrFrHc|Srrqr;s rr=z4test_result_order_group_keys_false..s1rc"|jSrrr;s rr=z4test_result_order_group_keys_false..s QVVXrrrs r"test_result_order_group_keys_falsers 3 4B CC # #M =FCE288EF # #M =O::ce:4::;MNO&(+ FFOOs$B:7$C:CCcdgdzdgdzzdgdzz}tj|}tj|d}ttt ddzt d|d }ttt ddzt d|d }d }t j t| 5|jd d jd}dddt j t| 5|jd d jd}dddt jy#1swYgxYw#1swY,xYw)Nz 2001-01-01rsz 2001-01-02z 2001-01-03UTC)tzrtr})r<r}trrr<FrHc,|ddgjSNr<r}rrs rr=z1test_apply_with_timezones_aware..r3*~**,rc,|ddgjSrrrs rr=z1test_apply_with_timezones_aware..rr) r&r/rrrr!r"r#r$r%r()dates index_no_tzindex_tzrrr*rrs rtest_apply_with_timezones_awarers/ NQ ,!!3 3|nq6H HE""5)K%0H $uQx.1,58+N OC $uQx.1,58(K LC CC # #M = ++ce+4:: ,   # #M = ++ce+4:: ,  '7+    s8$E?$EE Ecjtgdgdgdd}tddgddgd td d gd }|jd }d}tjt |d5|j td}dddtj||jd }t||}t|||}tjt |d5|j td}dddtj||y#1swYxYw#1swY,xYw)N)crrXrr)r rsrtr{r|r})rrrrr<rr}rrrrrRr6rr:The behavior of DataFrame.sum with axis=None is deprecatedF)rcheck_stacklevelinclude_groups) rrr$r!r"r#r%r3r(r getattr)reduction_funcr)rAgrpr*rBargs_s r;test_apply_is_unchanged_when_other_methods_are_called_firstrs  )#)  B1gS"I&RH3'H *** C FC # #Mu U63u56&(+ *** C ">2 6D$^$d+A # #Mu U63u56&(+6666s'D&D)D&)D2c tgdtdddtdddtdddtdddggddtgdd  }|jd d g}d}t j t |5|jd}ddd|jgd}|j}tj|gd|_ |jd }t j||jjdD]}t!|turJy#1swYxYw)N)rRrRrRrSir rrsrr:)refgidxr6rrrrrc$|jdSrerr;s rr=zRtest_apply_with_date_in_multiindex_does_not_convert_to_timestamp..sQVVAYr)rrsrt)rrrrF)rrrr$r!r"r#r%rgr r from_framerr/r(levelstype)r)rr*rBrAvals r@test_apply_with_date_in_multiindex_does_not_convert_to_timestamprs0 %T1b!T1b!T1b!T1b!   (u5 B **c3Z C CC # #M =0./0wwy!H##%H**84E+FGHN}}U}+H&(+||""1%!CyD   !00s D88Ecttjjdjddgt j ddgddgg}d}t jt| 5|jjd d }dddjd }d }t jt| 5|jdd }dddjd}t j||jt j||y#1swYxYw#1swY^xYw)Nrsr}r{rrr rFz5The 'axis' keyword in DataFrame.groupby is deprecatedrrr`rc(|jddS)Nrr droplevelr;s rr=zDtest_apply_by_cols_equals_apply_by_rows_transposed..s!1!=rrbc(|jddS)Nr rrrr;s rr=zDtest_apply_by_cols_equals_apply_by_rows_transposed..s!++1A+">r) rrrrrrr!r"r#rr$r%r()r)r*rmby_rowsgb2by_colss r2test_apply_by_cols_equals_apply_by_rows_transposedrs   a ''A/''#saV(<= B BC # #M =+ TT\\q\ *+hh=>G 7C # #M =*jjaqj)*ii>?G'799-'2&++ **s2D,D8,D58Edropnactgddtjtjddgdtd}d}t j t |5|jd |d jd }ddd|r|jn|jgd }t j|y#1swYCxYw)Nr rsrtr{r|rRrS)colrxxyxzrrrrF)rrIc|Srrqr;s rr=z5test_apply_dropna_with_indexed_same..sVWr)rrtr rsr{) rrr.rr!r"r#r$r%rrgr()rr)r*rBrAs r#test_apply_dropna_with_indexed_samers "26626634 7m  B DC # #M =YGFuEKKKXY$ryy{"''/*BH&(+YYs %CC zas_index, expected)r r r )rsrsr )rRrSN)rlzTODO(infer_string))reason)marksr)rsrsrSrrctgdgdgdd}d}tjt|5|j ddg|j d }dddtj |y#1swY xYw) Nr)r r r r rrrrRrSrcyrerqr;s rr=z5test_apply_as_index_constant_lambda..,sr)rr!r"r#r$r% assert_equal)rrAr)r*rBs r#test_apply_as_index_constant_lambdarsr( L|L MB CC # #M =NS#J:@@MNOOFH%NNs &A88Bcxtgdgdgddtd}d}tjt|5|j d j d }dddttd d tjgd d dgd}tj|y#1swYQxYw)Nr)r}r~ryrr)r r r rsrsr:r|rrrrc6|jjSr)rrr;s rr=z(test_sort_index_groups..8s1Arr r})rrrr^rrrr) rrr!r"r#r$r%r rrrLrs rtest_sort_index_groupsr0s O/JAh B DC # #M =CC&&'ABC a $$ 4S$K  H68,CCs "B00B9cttjdddtdt dd}d}t j t|5|j|j|jjjgd jd }dddt j|y#1swY xYw) Nz 2010-01-0112hr|)rSrr)rvalsletrrFrHc |jddSrrfr;s rr=z;test_positional_slice_groups_datetimelike..Ps!&&*r)rr&rrrr!r"r#r$rrdtr%r()rAr*rBs r)test_positional_slice_groups_datetimelikerCsMM,UAF!H= H DC # #M =&!! \\8==++00 1e" %$ % &&(+ &&s ACC ctgdgdgdd}|jd}|ddgjd}td d gd d gd td dgd}t j ||y)N)rRrRrSrwrr:rrrcj|jtj|jz Sr)astypefloatrerdr;s rr=z7test_groupby_apply_shape_cache_safety..Xs"AHHUO,?,?,AAEEG,KrrUrirVrrRrSr6r)rr$r%rr!r()r)rmrBrAs r%test_groupby_apply_shape_cache_safetyrTsr yyI JB CB c ^ ! !"K LFCjSz*%c 2MH&(+rctjgdgdgdd}|jdddg}|jd}t j gd gd gddg }t gd |d }tj||y)N)rRrSrRrS)aaacrad)rrrr)rRb1b2rRrrc>|jjSr)unstackr,r;s rr=z3test_groupby_apply_to_series_name..js!9!9!;r)rRrRrSrSrS)rrrrr)arraysr)rtr rsr r rr) r from_dictr$r%rrRr r!rL)r)rrB expected_idxrAs r!test_groupby_apply_to_series_namer`s   %**  B **S/4, 'C YY; .}sFOOAs,CrcF|jddjdS)NrF) ascendingr )r4rr;s rr=ztest_apply_na..sse)L)Q)QRS)Tr) rrr.r$r!r"r#r%r()rr)dfgrpr*rBrAs r test_apply_narts .sbggbkrr rrsr{)r r)rsrrr) r!r"r#rr&r>r$r%rrr()r*rBrAs r&test_apply_empty_string_nan_coerce_bugrs CC # #M =  %) 3? Wc3Z U) *    R, -2r~~ac7R/ST$$gw%7SzJH &(+#   s A C$$C- index_values)rUrVrWcLtgdgddt|}tdgditgdd }d }tjt| 5|j dj d }dddtj||y#1swY xYw) N)ra2a3rwrvrb_mean)rVrWrU)rrrrRr6rrcBt|djgdgS)NrSrr)r rrs rr=z0test_apply_index_key_error_bug..svr#w||~.xjAr)rrr!r"r#r$r%r()rrBrAr*s rtest_apply_index_key_error_bugrs# L! F o &S1 H DC # #M = $** A  &(+   s "BB#zarg,idx)皙?333333?皙?)rrr)r r{rt)rrrctd|i|}d}tjt|5|j ddj d}dddtj |y#1swY xYw)NrrrrFrHc|Srrqr;s rr=z5test_apply_nonmonotonic_float_index..s1rr)argrrAr*rBs r#test_apply_nonmonotonic_float_indexrssV%S1H CC # #M =N!!%E!:@@MN&(+NNs $A00A9z args, kwargs numeric_onlyc|jd}|jdg|i|}|jd}tj||y)Nrr3T)r)r$r%r3r!r()r)rkwargsrmrBrAs rtest_apply_str_with_argsrsK CB RXXe -d -f -Fvv4v(H&(+rr7 some_namectddg|}|jddgdjd}tddg|}tj||y)Nr rsr6rRFrHc|Srrqr;s rr=z1test_result_name_when_one_group..rr)r r$r%r!rL)r7rrBrAs rtest_result_name_when_one_grouprsU !Qd #C [[#s[ 6 < <[ IFq!f4(H68,rz method, opr%c |jdSr)r)rms rr=r=sRYYr]rc&|djdS)NrSrrfrs rr=r=sRW\\!_r)aggskew)rprod)rr3c tggd}|jdd}t|d}t|||}tgddt gdd}t j ||y) NrvrRTrHrSr^r#)r7rlr)rr$rr rr!rL)methodopempty_dfrmrrBrAs r test_empty_dfrsur+,H   #$  /B B E #WUF #B 'F IU2YS-QH68,rrc`tgdgdd}|jd}|rtnd}d}tj||5|j d|}dddtd d gd d gdt d d gd }|s|dg}tj|y#1swYIxYw)NrU)rtr{r|rvrRrrc"|jSrrr;s rr=z%test_include_groups..s AEEGrrrsr~r|r r6rrS)rr$r#r!r"r%rr()rr)rmwarnr*rBrAs rtest_include_groupsrs 3 4B CB*=D CC # #D 4L+NKL1vQF35!Qc;RSH SE?&(+ LLs B$$B-rrajimjoec tjjd}t|j dddddg}|j d|d <|j |}|j |turdnt}d }tj||d d 5|j|}dddt|j| }djd|d} |j|dfk(sJ|  fd d}tjt|5|j } dddtj | tjt|5|j fd} dddtj | |tk7rK|j# j%} | j'|dd tj || d tj(t+| dt+| dy#1swY}xYw#1swYxYw#1swYxYw)Nrsr r~)rrsrrrFrjolierF)rrraise_on_extra_warnings)subsetzinvalid frame shape: z (expected (z, 3))rtc4tt|dS)Nrrc)rr)r<fnames rr=z%test_builtins_apply..6s)wr5)!!4rrrc|Srrq)r<npfuncs rr=z%test_builtins_apply..=s vayrT)rYr/) check_dtyperrc)rrrrr<rr$__name__r3r#r!r"r%rdrop_duplicatesrzr(rr rrLr)rarrsr)rmrr*rBngroups assert_msgrAr rrs @@rtest_builtins_applyr#s   q !B 2;;q!W-u~ FB$$R(BwK D B JJEC<4]D FC # # C% !  "$$D$12G(l7)5QJ <r2sY  -85&%,P'-T,$ ,  /HI JIV L9 :QFC C!: $ 1aA BQFK +4FG H)T l+77    Iy)? CaVL!$ )<+$=<+$\2,, $#  & &-('$T&3R2&0 @ @0-9e}5 ,6 , F ,0,4, ,@u 60700-$2;-|,:,"C@94 ,..-" ,-X^^~x~~',,.0@0E0E0GH ,  ,%-PeT]3(4(,-, DK(!Q4 => && $1v1v&67 )QFQF+;< +ff-=> 8q\qQ< ( J!fq!f AA/ 0   -! - - &9;V%WX -Y -,$ FHe}506  0$ ,,*,@!@'2D%=1,2,"  I&6Ff0U ++##$6$8AU#V    A4j44ff5EcSVZX  $&%$&-&," ,-(D%=1 ,2 ,,.)_)EF,G,.             3%(R,S(R,D62,nd=S8T)UV,W,+t!45-6- *+ ,-   -  -)D%=9 ,: ,sCo.%%!89%W:/%Wr