uj~ddlmZddlZddlmZddlZddlZddlZddlmZm Z ddl Z ddl m ZddlmZddlmcmZdZdZdZd Zd Zd Zd Zd ZdZej<j?ddgddggdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ej<j?dddgdZ*d Z+d!Z,d"Z-d#Z.d$Z/d%Z0d&Z1d'Z2d(Z3d)Z4d*Z5ej<j?d+d,d-gd.Z6ej<j?d+gd/ej<j?d0d1d2gd3Z7d4Z8d5Z9d6Z:d7Z;d8Zd;Z?d<Z@d=ZAd>ZBd?ZCd@ZDdAZEdBZFdCZGdDZHdEZIdFZJdGZKdHZLdIZMdJZNdKZOdLZPdMZQdNZRdOZSdPZTdQZUdRZVdSZWdTZXdUZYdVZZdWZ[dXZ\dYZ]dZZ^ej<j?d[d\d\d]ggd^Z_d_Z`d`ZadaZbdbZcdcZdddZedeZfdfZgej<j?dgdhdidjdhdigdhdjggej<j?dkdldmdldmggdnZhdoZiej<j?dpdqdqggej<j?dd\gd\d]ggej<j?drd1gdgdsgdge jdge jdge jddtdtduve jdwdxdyze jdgd{|e jdgd}|e jd2gd~|g gdej<j?dgdej<j?dgddZoej<j?dddgej<j?ddtgdtdggdZpdZqdZrdZsdZtdZudZvdZwdZxej<j?ddddde jde jdddgifdde jde jde jde jde jde jdgifdde jde jde jde jde jde jdgifgdZzdZ{dZ|dZ}dZ~dZej<j?dgddZdZej<j?de jddfe jdd2fe jdd2fe jdd2fe jdd1fe jdd2fe jddfe jdd2fe jdd2fe jdd2fe jdd1fe jdd2fg dZej<j?dgddZdZdZej<j?ddddgfdddgfgdZdZdZdZdZdZej<j?d0d1d2gdZdZdZd„ZdÄZdĄZdńZdƄZdDŽZdȄZej<j?ddddggdʄZd˄Zej<j?ddgd2fddgd1fdgdτZej<j?dgdТdфZej<j?dddgdՄZej<j?dddgdلZej<j?ddd2e j<fgd܄Zej<j?dgdޢgdߢgdfgdgdgdfgdZej<j?dgddZej<j?dgddZdZdZej<j?d0d1d2gdZdZej<j?ddtdgdZej<j?dgddZdZdZdZdZej<j?dd1d2gej<j?ddddgddfdddgddfgdZdZdZej<j?dd1d2gdZdZdZdZdZdZy()datetimeN)Decimal)Pandas4WarningSpecificationError) BooleanArraycVttjdd}d}||k(sJy)NAB)keylevelz4Grouper(key='A', level='B', sort=False, dropna=True))reprpdGrouper)resultexpecteds l/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/tests/groupby/test_groupby.py test_reprrs* "**C0 1FEH X  c|jjd}|j|}|j}|j|j dj}|jj t jk(sJ|jj t jk(sJtj||dy)NrOF)check_index_type) indexcodesgroupbysumastypedtypenpint8int64tmassert_frame_equal) multiindex_dataframe_random_datar groupedrrs rtest_groupby_nonobject_dtyper%s * 0 0 6 6q 9C.66s;G [[]F/77 3HLLNH <<   (( ( >>  288 ++ +&(UCrc  tjgdgdtjj dj dtj tjj dj ddd}tt||d<d }|jd j|}|j}|jd j}tj||y) Nfoobarr(r)r(r)r(r()oner*twothreer+r+r*r,float32rr r CDvaluecB|j|djS)Nr4)locidxmaxgroups r max_valuez5test_groupby_nonobject_dtype_mixed..max_value2syyw..011rr columns)r DataFramerrandom default_rngstandard_normalarrayrangelenrapplydtypesdropr!assert_series_equal)dfr:appliedrrs r"test_groupby_nonobject_dtype_mixedrJ&s IM&&q)99!<"))//2BB1EYW   BB.BwK2jjo##I.G ^^Fwwsw#**H68,rctjdgdi}tjj gd|d<|j dD]/\}}t |djtjur/Jy)Ny)arMb)皙?皙?333333?x) rr=rmarArtypevaluesndarray)rH_r9s r7test_groupby_masked_array_column_values_type_consistentrX;si sO, -Beekk/*BsGJJsO55E#J%%&"**4445rcd d}d}|jd}|jtjdd}|j tjdd}|j tjdd}|j d}|j |}tj||tj||tj|||j|d}|j |d}|j |d}tj||tj||tj||y) Nrc2tj|||SNaxisr percentilerRqr]s rfz test_pass_args_kwargs..fF}}Q--rc2tj|ddS)NPrr\r^rRs rz'test_pass_args_kwargs..Is"--2A.rc|jSNmonthrfs rrgz'test_pass_args_kwargs..Ls aggrrer\皙?raNr) raggrr_rD transformquantiler!rG) tsrbg ts_grouped agg_result apply_result trans_result agg_expectedtrans_expecteds rtest_pass_args_kwargsrzEs). /A-.J r:J##BMM2A#>L'' r'BL&&s+L))!,N<6:|4<8R(J##A#,L''R'0L:|4<6<8rcLdd}|jd|}|jtjdd}|j t j jd}|jd}tj||d tj|||j t j jd dg}|jd dg}|s.farcrc|jSrirjrfs rrgz1test_pass_args_kwargs_dataframe..ds 177ras_indexrer\rlF check_names皙?level_0rr-r-)r6columnr4rlevel_1)rrlrrlrmrn) rrorr_rDrr=rqr!r"rBrinsert)tsframerrb df_groupedrurvr expected_seqs rtest_pass_args_kwargs_dataframer`sU.!2XFJ r:J##BLL$9$93?L""3'H,eD*h/##BLL$9$9C:FL&&Sz2L "1X )<H);OP, %HR(J##BLL$9$9S#AL*h/,eDrc "tjtjj dj dtj tdttjddd }|jd d d g}t|t|k(sJ|jd dg}t|jDchc]}|j|jfc}}t||k(sJycc}w)Nr- rABCDr0 2000-01-01rr periodsfreqr<rc|jSriyearrfs rrgztest_len.. AFFrc|jSrirjrfs rrgztest_len.. aggrc|jSridayrfs rrgztest_len.. rc|jSrirrfs rrgztest_len..rrc|jSrirjrfs rrgztest_len..rr)rr=rr>r?r@Indexlistobject date_rangerrCrrrk)rHr$rRrs rtest_lenrzs  a 009fV4mmL"3? B jj*,=OPG w<3r7 "" "jj*,=>?Grxx8!QVVQWW%89H w<8 ## #9sD ctjtjgdzgdd}t |j ddk(sJt |j ddk(sJt |j ddgdk(sJy)Nrr-rrMrNrMrrN)rr=rnanrCrrHs rtest_len_nan_grouprsu RVVHqLy9 :B rzz# 1 $$ $ rzz# 1 $$ $ rzz3*% &! ++ +rcNtjtjddgd}tjddgtjtjdgd}|j dd}|j }tj||d y) N1Dr(zm8[us])datarr)label timedeltarrFr) rSeries Timedeltar=NaTrmedianr!rG)rrHgbactuals rtest_groupby_timedelta_medianrs}yybll40xPH %. T8J/KL B G [ )B YY[F68?rkeysrMrNc tjtjdddtjggdtjdddtjggddd}|j |||}t |}|r|rd}n"|r|sd}nt |dk(r|rdnd}n|rd nd }||k(s J|d |y) Nrr-r categoriesrMrNc)observeddropnarr z vs )rr= CategoricalrrrrC)rrrrHrrrs rtest_len_categoricalrs Aq"&& 1iHAq"&& 1iH  B D8F ;B WFF & Ta1A1B X 8&hZ88 rcTtjttdddzDcgc]}d|z c}}tj j dj ddz}tj|}|j|}|jycc}w)Nrr?r-iLg$@) rrrrBrr>r?rmean)rRrr groupingsr$s rtest_basic_regressionrs} YYeArl);b)@AAaA BF 99  # * *4 04 7D $InnY'G LLN Bs B%cd}d}d}tjgdtdtddd}tjgd tdtddd}|jd j |}|jd j |}t j ||d }tjt| 5|jd j |dddtjt| 5|jd j |dddtjt| 5|jd j |dddtjt| 5|jd j |dddy#1swYxYw#1swYxYw#1swYfxYw#1swYyxYw) Nc||jdzdk(dz}|jr.f1sy qssQw1n  " 77taxtaxSzJ,,uJ?CJ S#J'AHrc||jdzdk(dz}|jrtjS|j ddg}|S)Nr-rrNr)rNrrr=r)rRrLs rf2z,test_indices_concatenation_order..f2sG qssQw1n  " 77<<> ! S#J'AHrc||jdzdk(dz}|jr=tjggdzggdzddg}tjddg|}|S|S) Nr-rr(r)rrMrNr)rNrrrr=rs rf3z,test_indices_concatenation_order..f3si qssQw1n  " 77taxtaxu~J,,SzDCJHr)rr-r-r-rrr)rr-r-r-rMz@Cannot concat indices that do not have the same number of levelsmatch) rr=rBrrDr!r"pytestraises ValueError)rrrrHdf2result1result2msgs r test_indices_concatenation_orderrs   LuQxeAqkJ KB ,,\auQ{K LCjjo##B'Gkk#$$R(G'7+ MC z -" 3b!" z -# Cr"# z -" 3b!" z -# Cr"##""##""##s0!G!G!!G'%!G3GG$'G03G<cX|jd}|j}|jd}tj|||j }|Dcic]\}}||j }}}t j|j}tj|||j}|jd}tj||d}tjt|5|jdddycc}}w#1swYyxYw)Nc"|jSriweekdayrfs rrgz#test_attr_wrapper..s 199;rc0tj|dS)Nr)ddof)rstdrfs rrgz#test_attr_wrapper..sRVVAA%6rc|jSrir0rfs rrgz#test_attr_wrapper..s QWWrz-'SeriesGroupBy' object has no attribute 'foo'r)rrror!rGdescriberr=Tr"rrrAttributeErrorr()rrr$rrnamegprs rtest_attr_wrapperrsjj./G [[]F{{67H68,   F4;<bbkkm#.s  rrrrrFrc"|jSrirrfs rrgz$test_frame_groupby..s rc(||jz Srirrfs rrgz$test_frame_groupby..sa!&&(lrc"|jSrirrfs rrgz$test_frame_groupby..#s affhrr)r aggregaterCr<copyrrr!r"headrprrGxsgroupsindicesitemstakeall)rr$ aggregatedtscopyrRstragged transformedrr9ridxrrrkv samethings rtest_frame_groupbyr soo34G""6*J z?a   z!! "a '' '\\^F.4ll;;F9~~i(226:H(JEBll2&&'<=G##$:;K { r !! ! {"" #q (( (##$67KQ ezz|;; QC  " ";>>##6% P QQ "3{{1~%%'72223^^FooG &1MM&&wqz2 Q##%%%&3r?r@rAro)rHcol1col2r$aggedrs rtest_multi_funcr&Qs! c7D c7Djj$((DHH-.G LLdL +Ezz3*%**,H !c3Z- (,,q3*}"=5 ))''*::1=))''*::1=((9:((9:  = Bjj$&G KKrc|jddgd}|jddg}tj|jd|jdd}t j ||y)Nr r r2rr)rr)rrorr=r!r")rHr$r%rs r!test_multi_key_multiple_functionsr(ms`jj#s$S)G KK (E||W[[%8UASTUH%*rc tjgdgdtjj dj dtjj dj dtjj dj dd}|j ddg}dd g}|j|}tj|d j||d j||d j|ggd d}t|jtjsJt|jtjsJtj||y)N r(r(r(r(r)r)r)r)r(r(r( r*r*r*r+r*r*r*r+r+r+r*r- )r r r3EFr r rrr3r-r.)r3r-r.r)rr])rr=rr>r?r@rroconcat isinstancerrr!r")rr$funcsr%rs r"test_frame_multi_key_function_listr2us1 <<  &&q)99"=&&q)99"=&&q)99"=;  DDllC:&G UOE KK Eyy   % '#,"2"25"973<;K;KE;RS  H ekk2== 11 1 hnnbmm 44 4%*rc 2tjgdgdgdtjj dj dtjj dj dtjj dj dd}|j ddg}d d g}tjd }|rd }tjt| 5|j|dddy#1swYyxYw)Nr*r+) dullr4shinyr4r4r5r5r4r5r5r5r-r,)r r r2r3r-r.r r rr&agg function failed [how->mean,dtype->-dtype 'str' does not support operation 'mean'r) rr=rr>r?r@rreescaperr TypeErrorro)using_infer_stringrr$r1rs r2test_frame_multi_key_function_list_partial_failurer<s <<   &&q)99"=&&q)99"=&&q)99"=U+ - D^llC:&G UOE ))< =C= y , Es 2D  Dopc"|jSrirrfs rrgrgs !%%'rc"|jSrirrfs rrgrgs QVVXrc |}|jddg}||}g}g}|jdD][\}}|jdD]B\} } |j|| f|j|| jddddgfD]tjj |ddg} tj |dj} | | _dD]C} ||| }|| }| | }tj||tj||E|dj|d|dgj}|jddgjd} tj|| y) Nr r r2r3rrr\r) rappendr6rr from_tuplesr/rrr!rGr)rHr=rr$rrrUn1gp1n2gp2mircol result_colpivotedexprs rtest_groupby_multiple_columnsrNs DllC:&GkG D F<<$6C{{3' 6GB KKR ! MM"SWWQc ]34 5 66  " "4Sz " :Byya(**HHN- % #,sm z3/ w, -#Y  S 495 6 ; ; =F||S#J',,.s3H68,rctjddgddgddggddg}|jdd djd}tjddgd }t j |||jdd d djd}tjgddtd}t j ||y)Nrr-rrrr r r;Fr~rT)r group_keysc"|jSri)cumsumrfs rrgz-test_as_index_select_column..s !((*r)r-rrrrr) rr=r get_grouprr!rGrDrBrHrrs rtest_as_index_select_columnrWs 1v1v1v.c CB ZZeZ ,S 1 ; ;A >Fyy!Qc*H68, ZZeZ =c B H HFyyE!H=H68,rcTtjtjgdd}|jdddj d }tj|j dd t d  }dg|j _tj||y) Nr r r2alpharPr;r Fbyrr rr-rr) rr=rrrr<rBrr!r")rHleftrs r0test_groupby_as_index_select_column_sum_empty_dfr^ s~ bhhWE FB ::u: -c 2 6 6E 6 JD||BJJrN%(CH"VH$)rc|dvrtjd|tjtj j djddddd g }t|jd|}|d k(r|jd }|j}|d k7r$|dj|dj|d<|jdd }t||}tj|||j!|}tj||t|d |}tj|||d j!|}tj||y)N)corrwithnthngroupz!GH 5755: Test not applicable for r-rr)dr-)sizerMrNr;rdFr~)rskiprr=rr>r?integersgetattrrrename reset_indexrrr!r"ro)reduction_funcrHrrsrs rtest_ops_not_as_indexrksY66 77GHI  a ))!QX)>c  B8wrzz#79H??6*##%H  ,,RW]];  3 'A 'WQ ' )F&(+ UU> "F&(+ ,WQsV^ , .F&(+ sVZZ 'F&(+rc|jdd}|jddgd}|djd}|jdjddddgf}t|tj sJt j|||djd}|jdjddgdf}t|tj sJt j|||dj}|jjddddgf}t|tj sJt j|||dj}|jjddgdf}t|tj sJt j||y)Nr Fr~r r2rrY) rror6r0rr=r!r"r)rHr$grouped2rrr expected2s r!test_as_index_series_return_framero:s~jjuj-Gzz3*uz5H S\  e $F{{5!%%a#sm4H fbll ++ +&(+sm&G U#''?(:;I gr|| ,, ,'9- S\   F{{}  S#J/H fbll ++ +&(+sm!G ""1o#56I gr|| ,, ,'9-rc|jdd}|jddgd}|dj}|jddjj}tj|||dj}|jddgdjj}tj||t j gdgdd}tjtd 5|jd dd jdddy#1swYyxYw) Nr Fr~r r2)r r r r )r@@@)openpricez"cannot insert open, already existsrrtru) rohlcrir!r"rr=rrr)rHr$rmrrrrn df_conflicts rtest_as_index_series_ohlcrxSsjjuj-Gzz3*uz5H S\   Fzz#s#((*668H&(+sm  "G C:&s+002>>@I'9-,,%0DEK z)M NDFU3G<AACDDDs %EE c|jdd}d}tjt|5|dj ddddy#1swYyxYw)Nr Fr~zColumn\(s\) C already selectedrr2r3)rrr IndexError __getitem__)rHr$rs r(test_as_index_series_column_slice_raisesr|gsNjjuj-G +C z -&   %&&&s AAc|}|jdd}|jd}|jdgjd}|jdd|jt j t ||_tj|||jddgd}|j}|jddgj}tt|jjddi}|jdd|d|jd d|d t j t ||_tj||y) Nr Fr~Trrr strictr) rrrrr RangeIndexrCr!r"rziprU)rHrr$rrarrayss rtest_groupby_as_index_cythonrps) Dll3l/G \\t\ ,F||SE"''T':H OOAsHNN+]]3x=1HN&(+llC:l6G \\^F||S#J',,.H #x~~,,:T: ;F OOAsF1I& OOAsF1I&]]3x=1HN&(+rc|jddgd}|djt}|jtjddgdf}t j ||y)Nr r Fr~r2rY)rrorCr6r!r")rHr$rrs r#test_groupby_as_index_series_scalarrs\jj#sej4GS\  c "F{{3##A$67H&(+rc tjtjj dj dtj tdttjddd }|jd d d g}|j}tj|j|jy) Nr-rrr0rrr rrc|jSrirrfs rrgz+test_groupby_multiple_key..rrc|jSrirjrfs rrgz+test_groupby_multiple_key..rrc|jSrirrfs rrgz+test_groupby_multiple_key..rr)rr=rr>r?r@rrrrrrr!assert_almost_equalrU)rHr$r%s rtest_groupby_multiple_keyrs  a 009fV4mmL"3? B jj*,=OPG KKME299ell3rc |j}tj|d<|jddgj }|jddgj }tj|d<t j ||y)Nbadr r )rrrrrr!r")rHr%rs rtest_groupby_multi_cornerrsm BBuI JJSz " ' ' )Ezz3*%**,HffHUO%*rc|jd}tjd}|rd}tjt |5|j ddddtjt |5|jddd|jddgdf}tddd|d <|jd}d }tjt |5|j d dddtjt |5|jdddy#1swYxYw#1swYxYw#1swYVxYw#1swYyxYw) Nr r6r7rrr r2r3rr-z0datetime64 type does not support operation 'sum'r) rr8r9rrr:rorr6rr)rHr;r$rs rtest_raises_on_nuisancersjjoG ))< =C= y , F y ,  ?" #BtQ"BsGjjoG .C'DlSUHLLSwrvvbg.>.C'DlSUHLL&(+r)rrprodrvarsemrrTFc|jd}d}||vrq|so|r d|d}t}n,|dvr t}d}nt}tjd|d}t j || 5t||| dddyt||| }|s |d k(rgd } ngd } t|jdd| fjd|| } tj|| y#1swYyxYw)Nr )rrrrrr(dtype 'str' does not support operation '')rrz(could not convert string to float: 'one'zagg function failed [how->z,dtype->rrrr1r) rr:rr8r9rrrgr6r!r") rHrrr;r$no_drop_nuisancerklassrr<rs rtest_omit_nuisance_aggrsjjoGF''  <\N!LCE ^ +ExYw#1swYyxYw)Nr r r6r7rr)rr8r9rrr:ror) three_groupr;r$rs r&test_raise_on_nuisance_python_multiplers!!3*-G ))< =C= y , F y , sB=B#B #B,c tjtjgdtjgdddgdztjj dj dtjj dj dd}|jd d g}|d d gjd }|jd}tj|||ddjd}|jd}|djd }tj|d||jjdk(sJy)Nrrr(r)rr-r)rrk3rrrrrrrTrrrr c"|jSrirrfs rrgz*test_empty_groups_corner..s AFFHrr first)rr=rrAr>r?r@rrorr!r"rDrGrr)r#rHr$rrr%agged_As rtest_empty_groups_cornerrs+ ((9:((9:%.1$))''*::1=))''*::1=   Bjj$&G dD\ " & &v .F|||.H&(+.q3;;!;DG MM, -Ecl  (G5:w/ ;;  w && &rctjdg}d}tjt|5|j ddddy#1swYyxYw)Nrz5unsupported operand type\(s\) for \+: 'int' and 'str'rc |dzS)Nr(rfs rrgz$test_nonsense_func..#s QYr)rr=rrr:r)rHrs rtest_nonsense_funcrsG qc B BC y ,( &'(((s AAc|j}tjgdtjgdg}|j|j d}t |j tjsJd}tjtd5|j|j|dddy#1swYyxYw)NrrrrcV|jdk(r td|jS)N)r(r*Test error message)rr:r)sers raggfunz5test_wrap_aggregated_output_multindex..aggfun-s& 88~ %01 1wwyrrr) rrrArror0r<rrrrr:r)r#rHrr%rs r%test_wrap_aggregated_output_multindexr&s ) + +B HHY )!4 5D JJt   (E emmR]] 33 3 y(< =+ 4""6*+++s !C  Ccl|jdj}|jjdk(sJ|jdj}|jjdk(sJ|djdj}|jjdk(sJy)Nrrrrsecondr )rcountrr)r#rs rtest_groupby_level_applyr6s - 5 5A 5 > D D FF <<   '' ' - 5 5A 5 > D D FF <<   (( ( -c 2 : : : C I I KF <<   '' 'rc|j}ddddd}dddd}|j|dj}|j|dj}tj|dDcgc]}|j |c}tj }tj|dDcgc]}|j |c}tj }|j|j} |j|j} d \| j_| j_tj|| tj|| ycc}wcc}w) Nrr)r(r)bazqux)r*r+r,rrr0r)rr) rirrrrAr"r rrr!r") r# deleveledmapper0mapper1result0rrR mapped_level0 mapped_level1 expected0 expected1s rtest_groupby_level_mapperr@s<0<<>I!A6GA.G.66wa6HLLNG.66wa6HLLNGHH!*7!34AQ4BHHMHH!*8!45AQ5RXXM188GKKMI088GKKMI1B.IOO)//.'9-'9- 5 6s 6E8;E=c tjgdtjgdd}tjgdtjtt ddd}|j d j }tj|||j dg j }tj|||j d j }tj|||j d g j }tj||d }tjt| 5|j d dddtjt| 5|j d dddd}tjt| 5|j g dddd}tjt| 5|j ddg dddtjt| 5|j ddg dddd }tjt| 5|j dg dddy#1swY$xYw#1swYxYw#1swYxYw#1swYxYw#1swYtxYw#1swYyxYw)N)rr-rrrrr)rr-rrrrr-rr(rP)r,rrrrrrrz2level > 0 or level < -1 only valid with MultiIndexrzNo group keys passed!z*multiple levels only valid with MultiIndex) rrrrrBrrr!rGrrr)srrrs rtest_groupby_level_nonmultirWs< "BHH-EE$R Ayy-rxxU1a[8IPU/VWH YYQY  # # %F68, YYaSY ! % % 'F68, YYRY $ $ &F68, YYbTY " & & (F68, >C z -   z -   !C z -   6C z -  A  z -  A  >C z -      sH'JJ)J5 KK ?KJ&)J25J>K  KK"c&tjtjddzgd}tjdtjddg}|j d j }tj||y) Nr?@)rrrrrr)ry@$@rrr!r) rrrarangerrrr!rG)rMrrs rtest_groupby_complexrysf ryy|v.lCAyy*"((Aq62BCH YYQY  # # %F68,rc>tjdddddddddg}|jdj}tjdgdggtjddgd tjd g }t j ||y) Nr-rrr??rNr?rPrMrr<)rr=rrrr!r"rVs rtest_groupby_complex_meanrs & !& !& !  BZZ_ ! ! #F|| hh6+#6#H &(+rc.tjdddddddddg}tjtjgdtjtj gdd tj d g }|j d d j}tj||tj gdd |_ |j d dj}tj||y)Nrrrrr)rrrr0)rr?rNrPrMrFsort)rrrT) rr=rrAr rrrr!r"rrHrrs rtest_groupby_complex_numbersrs & !& !!   B|| "((+hh5C@#H ZZ%Z ( . . 0F&(+XX<3GHN ZZ$Z ' - - /F&(+rctjgdtjgd}tjgdtjgd}|j|}|j }|j|j |j jj }tj||y)N)@g"rsgY@ggK@g@)rMrNrderbrsr!)rrrsrrg@)rMrNrrbrsh) rrrrrreindexrr"r!rG)s1s2r$r%rMs r'test_groupby_series_indexed_differentlyrs 0hh:; B &bhh7U.V BjjnG LLNE **RZZ)-- . 3 3 5C5#&rcttgdgdgddi}tjj |}tjj gd}tj t jjdjd||}|jd j}tj|j||jd jd }tj|j||jd j!d }tj|j||j#d \}}d |d<|jd jd}tj|j|jddy)N)r)r)rrr(r(rr)r*r+r*r+r*r+r*r+r~T))r cat)r dog)r r)r rr-)r.rrrrrc"|jSrirrfs rrgz0test_groupby_with_hier_columns..s rr))r r(rr)rrrrrDr=rr>r?r@rrr!assert_index_equalr<rorD sortlevel)tuplesrr<rHrsorted_columnsrWs rtest_groupby_with_hier_columnsrsq  HH   F MM % %f -Emm''@G  a 008w BZZaZ % % 'F&..'2 ZZaZ $ $V ,F&..'2 ZZaZ & &'9 :F&..'2 ))!,NABzN ZZaZ % %4 % 8F&.."**Sb/:rc|j|dj}|j|djd}|j}|j}t j ||y)Nr )rrUrhrr!r")rHr$rmrrs rtest_grouping_ndarrayrsYjjC(Gzz"S'.../H [[]F||~H&(+rctjgdd}tjgdgdgdgdgdd | }|jgd }|j d }|j }t j||y) Nrrr-rrrrP)foo1rfoo2rfoo3)bar1bar2rrr)baz1rrbaz2r)spam2spam3rspam1r)rr(2<)r(r)rspamrr!)r(r)rrr)rrr=rrorr!r")rrr$rrs rtest_groupby_wrong_multi_labelsrss HH_7 3E <<;;;A(   Dll89G [[ F||~H&(+rc|j|djd}|j|ddjd}|jjdk(sJd|vsJ|j|d|dgj}|j|d|dgdj}|jjdk(sJd|vsJd|vsJy)Nr TrFr~r r r )rrrrrrHrrs rtest_groupby_series_with_namer s ZZ3 % %4 % 8FjjC5j166D6IG <<   ## # '>> ZZC"S'* + 0 0 2Fjj"S'2c7+ej<AACG <<   ++ + '>> '>>rc|jdd}|jjdk(sJ|jjdk(sJd}|j |jdk(sJy)Nr r2c2tj|dzSNr-)rrrfs rrgz.test_seriesgroupby_name_attr.. sQr)rrrrro)rHrtestFuncs rtest_seriesgroupby_name_attrrsj ZZ_S !F <<>  # %% % ;;=   $$ $&H ::h  $ $ ++ +rctjgdgdtjj dj ddztj dd}|jdgjj}|jj|jj}tj||y)Nr')r*r*r+r+r+r+r*r+r-r.rr1r ) rr=rr>r?r@rrr rr r!rGrs rtest_consistency_namers II&&q)99!.summarize#syy1aa@tLLrcXtjdddd|jddS)Nrr-rrrr2rP)rrilocrs rsummarize_random_namez.summarize_random_name&s)yy1aa@rwwqzRUWWrr metricsri)rrDr<r)rHrrrs rtest_groupby_name_propagationr!sMX jjo##I.G ??   '' 'jjo##Iy9G ??  9 ,, ,jjo##$9:G ??   '' 'rc>tjtdDcgc]}tjdc}}|j d}|j }|j |dj }tj||ycc}w)Nrr) rr=rBrrrrr!r")rRrHr$rrs rtest_groupby_nonstring_columnsr4so eBi8ryy}8 9BjjmG \\^Fzz"Q% %%'H&(+ 9sBcvtjgdggd}tjddggddgtjdgd  }|jdj }t j |||jdj}t j ||y) Nrrr-)r r rr;rr-r rr rPr)rr=rrrr!r"rrs rtest_groupby_mixed_type_columnsr<s yk= 9B||aVHsAhbhhsQT>UVH ZZ_ " " $F&(+ ZZ_ "F&(+rctjd}|jtjt j |dddf}tj tdd}|j|jtj j}|jjsJy)N)rcrcrr) rrfillrrrtilerBrrorisnar)arrobjindsrs r(test_cython_grouper_series_bug_noncontigr'Hs ((: CHHRVV ))C1I C 7759b !D [[  " "299#3#3 4F ;;=    rcrtjdgdz}tjtjj dj d|ddd}tjj djddd}|j|}d}|j|y) N aaaaaaaaaarcr-rr!rrcZtttt|jSri)rCsetmapidrrfs rrgz5test_series_grouper_noncontig_index..\s#c#b!''*+,r) rrrrr>r?r@rfrro)rrUlabelsr$rbs r#test_series_grouper_noncontig_indexr/Rs HHhZ#% &E YYryy,,Q/??C5QTSTQT: VF YY " "1 % . .q!R 8FnnV$G -A KKNrctjtd}tjgdd}d}d}|j |}|j |}|jtjk(sJt|jdtsJ|j |}|jtjk(sJt|jdtsJy)Nr)rMrNrrrrr0cFtt|jSri)rstrrrfs r convert_fastz>test_convert_objects_leave_decimal_alone..convert_fastdss1668}%%rct|jjdkDsJtt |j Srn)rCrUbaserr2rrfs rconvert_force_purezDtest_convert_objects_leave_decimal_alone..convert_force_puregs2188==!A%%%s1668}%%rr) rrrBrrArrorobject_r0rr)rr.r3r6r$rs r(test_convert_objects_leave_decimal_aloner8`s %(A XX/s ;F&& iiG [[ &F <<2:: %% % fkk!ng .. . [[+ ,F <<2:: %% % fkk!ng .. .rctjgtjddd}|djtj k(sJ|j dj}tjgdtj }tjdtjg|di}tj||y) Nrr r0)rRrBrRrrrBrr) rr=rrrfloat64rrrrr!r")rHr exp_indexrs r"test_groupby_dtype_inference_emptyr>ws B1G)DE FB c7==BJJ && & ZZ_ " " $F#RZZ8I||Wbii)7&STUH&(+rc,tjdgdgdgd}|jddgdj}tjdgtj j dgdggddgd}tj||y) Nrl4;PU)rrr4rrr4rBrP) rr=rrrr from_productr!rGrVs r$test_groupby_unit64_float_conversionrAs s>R=ST UB ZZ(+ ,W 5 9 9 ;Fyy  ""QC!:gx5H"I H 68,rc|jt|djd}|j|djd}tj||dt j td5|jt|ddddddtjd d gd d gtjjd jd d}|jddgj}|j|d|dgjdg}y#1swYxYw)Nr TrFrz^'foo'$rrrrrrr-)r(r)valr(r)rC)rrrr!r"rrKeyErrorrr=rr>r?r@rVs r"test_groupby_list_infer_array_likerEs ZZRW & + + + >Fzz"S'"''T':H&(> xz 2' 43 %&' q6q699((+;;A>  BZZ ' , , .Fzz2e9bi01668%AH''s !EE cd}tjtjdd|}tjddgddggd d g| }|j tj d | d gj }|j|jd gjt}tj||y)Nrz2015-09-29T11:34:44-0700r-)startrrr rr metricrUrr)r r) rr Timestampr=rrrrrrfloatr!r")rrrHrrs r$test_groupby_keys_same_size_as_indexrLs D MMll56 E Ry3),x6JRW XB ZZ!$7B C H H JF||RXXx0188?H&(+rc<d}tjtjj dj dt d}tjt|5|jddddtjtjj dj dt d}tjt|5|jddddy#1swYxYw#1swYyxYw) Nz^'Z'$r-)rrrr;rZ)r-r) rr=rr>r?r@rrrrDr)rdf1rs rtest_groupby_one_rowrPs C ,, a 008$v, C xs + C ,, a 008$v, C xs + C  s*D+DDDc , tjtjj dj dtj tjdtj tjdtj tjdtj tjdgtj dtj dtj dtj dgd}|jd}tjd d d tjd dd g}t|jj}t|dk(sJt||dD](\}}tj |j||*tj"|j$j&dj(||j*dk(sJtjdtj,d dgtj.tjdtj,d dgtj.i}|j0D](}tj2|j0|||*tj"|j5tjd|j6d dgtj"|j5tjd|j6d dgt9j:t<d5|j5tj>dddtjtj tj tj gtj>tj>tj>gd}|dj@dk(sJ|dj@dk(sJdD]}|j|}|jik(sJ|j*dk(sJ|j0ik(sJt9j:t<d5|j5tj dddt9j:t<d5|j5tj>dddy#1swYvxYw#1swY\xYw#1swYxYw)Nr-r.z 2013-01-01z 2013-02-01rMrN)rUdtr2rRr r)rGstopsteprrTr~rz2013-01-01 00:00:00r0z2013-02-01 00:00:00rz^NaT$r)rnatrr<rWz datetime64[s]z^nan$)!rr=rr>r?r@rrJrrsortedrrrCrr!rr"_grouperrr%ngroupsrAintprassert_numpy_array_equalrUrrrrDrr)rHr$rrr rnan_dfr s rtest_groupby_nat_excluder^s ii++A.>>qA \* \* \* \* FFCbffc2663G  B jjG ABQ/ AAA.H '..%%' (D t9>>D(4041 gnnQ/34 '**44Q7;;R@ ??a   *+RXXq!fBGG-L *+RXXq!fBGG-LH __E ##GOOA$6 DE",,|45rww1v",,|45rww1v xx 0""&&!"\\("&&"&&"&&1IJF %=  ) ++ + %=  / 11 1&..%~~###!###"$$$ ]]88 4 &   bff % & ]]88 4 &   bff % & &&"" & & & &s$ S1 S> T 1S;>T T ctjtjtjgtjtjgddgd}|j ddgj }|ik(sJy)Nrr-rrMrN)rr=rrrrrHrs r#test_groupby_two_group_keys_all_nanrasY RVVRVV,BFFBFF3C1a&Q RB ZZc # + +F R<r?r@rrrC)r r rHr]rights rtest_int32_overflowrps  %("))E*:BIIdOLMA %A &&q)99%@   B ::* + / / 1D JJ+ , 0 0 2E t9E "" "rctjgdgdgdtjj dj dd}|gdj Dcgc] }t|}}tj|}|jgdd j}tj|jj |gd |gd j Dcgc] }t|}}tj|}|jgd d j}tj|jj ||gd j Dcgc] }t|}}tj|}|jgd d j}tj|jj |gd tjgdgdtjj dj dd}|jddgd}|j}dfd}|||ddgdycc}wcc}wcc}w)Nr(r)r)rr-rrr-rrMrNrrrTr)rr-r)rrMrN)rNrrM)r-rrrrr-rrr-)rrrrrrr)rMrNrrMrNrc"|jSrir?rfs rrgz)test_groupby_sort_multi..Ms AEEGrc||jDcgc] }t|}}tj|}||j ||}|j D]\}} ||| k(rJycc}wri)rUtuplecomasarray_tuplesaferr) rHrrfieldrbrowtupsrr r s r_check_groupbyz/test_groupby_sort_multi.._check_groupbyMsx&(hoo6sc 66$$T*RZZ%e,-NN$ "DAq!9> !> "7sA8)rr=rr>r?r@rUrwrxryrrr!r\r)rHr{r|rrRr$r}s rtest_groupby_sort_multir~*s &&&q)99!<   B#%_"5"<"< =3E#J =D =   &D ZZdZ 3 7 7 9F 3 3T)_E"$_"5"<"< =3E#J =D =   &D ZZdZ 3 7 7 9F 3 3T: 188 9E!H 9D 9   &D ZZdZ 3 7 7 9F 3 3T)_E ##&&q)99!<  Bjj#s$S)G [[]F2C"2vSz3/A > > :s I5.I:7I?ctjgdgddzd}|jddjd}t j ||d gy) N)rMrMrMrNrNrNrrr-)r rr FrQc|Srirrfs rrgz/test_dont_clobber_name_column..\srr)rr=rrDr!r"r`s rtest_dont_clobber_name_columnrWsT .8MPQ8QR BZZ%Z 0 6 6{ CF&"fX,/rc tjtjj dj dtj tdttjddd }|jd d }|jd }|Dcgc]\}}|jddd}}}tj|}tj|||djdd }|jd}|Dcgc]\}}|jdd}}}tj|}tj ||ycc}}wcc}}w)Nr-rrr0rrr rrc|jSrirjrfs rrgz&test_skip_group_keys..gs AGGrFrc,|jdddS)Nr r\r sort_valuesrfs rrgz&test_skip_group_keys..hsQ]]c]%:2A%>rr rrc|jSrirjrfs rrgz&test_skip_group_keys..os rc(|jddSNrrrfs rrgz&test_skip_group_keys..psQ]]_Ra%8r)rr=rr>r?r@rrrrrrDrr/r!r"rG)tsfr$rr r9piecesrs rtest_skip_group_keysr`s7 ,, a 009fV4mmL"3? C kk+k>G ]]> ?F>E F Ue3'+ FF Fyy H&(+#h0UCG ]]8 9F8? @*#ue!"1% @F @yy H68,GAs (E/#E5c|dj}d|_|j|djd}|jJy)Nr2r r)rrrro) float_framerrs rtest_no_nonsense_namerxsJCA AF YY{3' ( , ,U 3F ;;  rctjtjdj dd}d|d<gd|d<|j d}|j ddd }|djtjk(sJy) Nrrrtest)g?rg?flrrd)rr-) rr=rrreshaperrorr<)rRr$rs rtest_multifunc_sum_bugrst RYYq\))!Q/0AAfIAdGiiG [[&1 2F $<   ++ +rcd}d}|jddj|}|jddj|}t|tjsJt j ||y)NcD|j|jdSNrrrr8s rrbz(test_handle_dict_return_value..fsyy{599;77rcjtj|j|jdSr)rrrrr8s rrsz(test_handle_dict_return_value..gs"yyUYY[ABBrr r2)rrDr0rrr!rG)rHrbrsrrs rtest_handle_dict_return_valuersm8CZZ_S ! ' ' *Fzz#s#))!,H fbii (( (68,rgrouperr r ctd}dfd}|j|d}|j||j|jd|j||dj||dj|dj|g|dj|y)Nc"|jJ|SrirPr8s rrbztest_set_group_name..fszz%%% rc>|jJ|jSri)rrr8s rfreducez$test_set_group_name..freduceszz%%%yy{rc|Srir)rRrs rfreducexz%test_set_group_name..freducexs qzrFrrr2)rrDrrp)rHrrbrr$rs @rtest_set_group_namersjjUj3G MM! g G'23 a CLq CL7# CLGX./ CL1rctjgdtjdd}gfd}|j dddj |gd}|k(sJy) N)rrrrr-r-rrcZj|j|jSri)rCrr)r9rs rrbz6test_group_name_available_in_inference_pass..fs UZZ zz|rrMF)rrQr)rr=rrrrD)rHrbexpected_namesrs @r+test_group_name_available_in_inference_passrs[ .RYYq\B CB EJJs5J177:N N "" "rc8|j|djj}|jjJ|j|dj|djgj}|jj dk(sJy)Nr r )NN)rrUrrrrrs rtest_no_dummy_key_namesrs{ ZZ3 ' + + -F <<   $$ $jj"S'.."S'..9:>>@G ==  , .. .rctjddgddgggdgdgddg}tjgd| }tjddgddgggd gd gddg}tjgd | }|jddgd j }t j |||jddgdj }t j ||jy)Nrr-)rrrrrr)rrrrrrrMrNr)rrr-rrrr!r)rrrrr-rFr rT)rrrrrr!rG sort_index)rmseriesmseries_resultrs r#test_groupby_sort_multiindex_seriesrs MMAA!#56Cj E ii*%8G MMAA 9'=c3Z EYYy6N __C:E_ : @ @ BF6>2 __C:D_ 9 ? ? AF6>#<#<#>?rcd}tjdd|}tjtj|tj|d|}d d}|j d}|j d |tji}|j d |tjd i}tj||y) Nz2012/1/15min)rGrr)highlowr!cfd}|S)z> Run an aggregate func on the subset of data. c|j|jjdj}r||jdt |dk(ry|S)Nc |jdkS)Nr,)hourrfs rrgzYtest_groupby_reindex_inside_function..agg_before.._func..s!&&2+rr)r6rr,rrC)rrfixfuncs r_funczGtest_groupby_reindex_inside_function..agg_before.._funcsR(=>?FFHATZZ]#1v{7Nrr)rrrs`` r agg_beforez8test_groupby_reindex_inside_function..agg_befores   rcXt|j|j|jSri)rrrkrrfs rrgz6test_groupby_reindex_inside_function..s8AFFAGGQUU#CrrT)F) rrr=rrrrorr!r")rindrHrr$ closure_bad closure_goods r$test_groupby_reindex_inside_functionrsG --jvw GC 7#BIIg,>?s B jjCDG++vz"&&'9:;K;; 2664(@ABL+|4rchtjgdgdgdd}|jddg}|jddgd}|j d }tj j gd ddg }tjd gd gdgg|dg}tj||y)N)rMrMrMrN)rrrr)rrrr)group1group2r4rrTrr))rMr)rMrrNrrBr-rrr4r) rr=rrrorrDr!r")rHrrr rMs r$test_groupby_multiindex_missing_pairrs **!  B x* +B8X"6TBJ .. C -- # #,Xx4H $ C ,,aS1#cG9 EC#s#rctjjgdddg}tjgdg|}|jj sJtjgdgdgd g }|j d ddgd }|j}|jj rJ|jd j}d}tj||5|jd j}dddtj|tjgdgdgddjddg}|jj rJddddgfD]}dD]} |j|| djtjj }|}tj|||j#j|| djtjj }|j#}tj||y#1swYBxYw)N))rM)b1c1)b2c2rNrrBrrrr;rs)rrrr)rrrr)r<rrMr)rr<rUzAdropping on a non-lexsorted multi-index without a level parameterr)rMrMrNrMrrr-rrrRrLzrRrLrr)FTF)r rrQ)rrrDr=r< _is_lexsorted pivot_tablerirrr!assert_produces_warningr"rrrDdrop_duplicatesr) performance_warning lexsorted_mi lexsorted_dfnot_lexsorted_dfrrrrHr rs r%test_groupby_multiindex_not_lexsortedrs'==,,/Sz-L<< \BL    - - // /||$,>@R+S(33C:c4(335''5577 7##C(--/H MC # #$7s C6!))#.3356(F+ "LI ic xx%%'' 'QA4! 4DZZe$5ZIOO ,,FH  ! !(F 3 u4EBr||334  }}H  ! !(F 3 4466s 5 IIcDtjtdgd}|jtd}|j d}|j gd}t j|||d}|jtd}|j d}|jgd}t j|||jjt|_ |jtd}|j d }|j gd}t j|||d}|jtd}|j d }|jgd}t j||y) NABCDE)r-rr-rrr!ababbct|dkDSr rCrfs rrgz4test_index_label_overlaps_location..KA rrrct|dkDSr rrfs rrgz4test_index_label_overlaps_location..Qrrct|dkDSr rrfs rrgz4test_index_label_overlaps_location..Xrrct|dkDSr rrfs rrgz4test_index_label_overlaps_location..^rr) rr=rrfilterrr!r"rrGrrrK)rHrsrrrs r"test_index_label_overlaps_locationrFs< d7m? ;B 4=!A XX* +Fwwy!H&(+ Q%C DM"A XX* +Fxx "H68,xxu%BH 4=!A XX* +Fwwy!H&(+ Q%C DM"A XX* +Fxx "H68,rctd}tj|}tj|dzd|zd|zd}tj|dzdzd|zd|zd}|j d}|j d}|j d}|j d}t j||y) Nrr-rqrrrrrMr)rrrr=rrpr!r")nrRrHrrrgb2rs r"test_transform_doesnt_clobber_intsrcs A ! A AFqsQw? @B ,,Q!Vc\acAgF GC CB \\& !F ++c C}}V$H&(+r sort_columnintsfloatsstrings group_column int_groups string_groupsctjgdgdgdgdgdd}|j}|j|}fd}|j |y) N)rrrrrrrr)rrMrrMrMrsrsrs)r.rrrr-rrr)gffffff@g333333@g@g333333g@皙?rr)rrrMrwordword24247)rrrrrrcRtj||jy)Nr)r!r"r)rRrs r test_sortz.test_groupby_preserves_sort..test_sorts a+!>?r)rr=rrrD)rrrHrsrs` rtest_groupby_preserves_sortrrs[ 2E,=H   B ; 'B < A@GGIrc tjtjtdddddj t dd}|j djj|d<|j djj|d <tjtd 5|jjdd d d dddy#1swYyxYw)NrrrMEr) eventDatethenamerrrkz 'badname'rbadnamer)rr<rUaggfunc)rr=rrtolistrBrrrrkrrrDrirrs r!test_pivot_table_values_key_errorrs q!$btfhRy   Bk*0055BvJ,,{+1177BwK x{ 3  $$')W %    s <$C))C2r<r2rUrirz US/Eastern)tz 2016-01-01rr3rInt64r0Float64boolean) boolintrKr2rdt64dt64tzperiodrrr)idsmethod)attrrorD)r7idxminrrrrrkurtcTd}ttrdvrd}tdtrdvrd}tjdt d}t d r)|jjk(jsJ|jdd}|jd |d |fd } fd } t|jjdtj} |jjdjdk(} ttj} t|jjdtj}ttjrj sdvrdvr dd}t"}n dd}t$}t'j(||5| ddddvr6t|t r&| d}| }t+j,||ydvr| s | s | s|rdk7r| rd}n| rd}n |rdd}nd}dvrd j/|d!dg}t'j(t"|5| dddt|t sydvry| d}|j1g}| r| }t+j,||y| }|j1|}d"vr%|j3|j4j}||j3|}t7d#k(rDtj8g|djd$|_d|j4_t+j,||y#1swYxYw#1swY*xYw)%N)rrrr)rrr rYABCr;rF)rQrrcXdk(rtdi|Stfi|S)Nrr)rg)kwargsrrr=s r get_resultz&test_empty_groupby..get_results: V "72r?,V, ,&72v&r4V4 4rcLtjdgj}tdk7r$tjj ||g}ntj |d}rtj gd}ng}tjg||}|S)Nrr0rrBrPr2r)rrrrCrr@rr=)levr r<rrr;rUs r get_categorical_invalid_expectedz.get_categorical_invalid_expecteds nnaS 5 t9>--,,c3Zt,DC((3T!W-C hhr/GG<<G3?rM)rrrr7)rrzCannot perform z with non-ordered Categoricalz Can't get z/ of an empty group due to unobserved categoriesrTr)rrrrrz datetime64 type does not supportzPeriod type does not supportrrzcategory type does not support)rr|zdoes not support operation ')r7rrrr)r0rrrr=rhasattrrErrrr PeriodDtypekindr StringDtypeorderedr:rrrr! assert_equaljoinrrrrCrr)r<rrUrr=rr;override_dtyperHrris_peris_dt64is_catis_strrrrrrs ```` ` @rtest_empty_groupbyr)sNJN&,'B/,A &)T"r_'<  Ff=tE{ SBvw V\\)..000 !B DU6E J7 SB5 ( q)2>> :FiinnQ$$+G  /F  q)2>> :F 62>>* 4 4  #B4'DECErd"QRCE ]]5 ,  L  3 3 7D8QT2F79H OOFH - ,, f6bEk84@AF6%%hh'CB4q%IJKy4   gt,'' $6<<-b1?AH1 \F||D!'*H !!??288>>2!??>2 4yA~"BtAwK,=,=DGL"1gOOFH%u  2  sN-NNN' method_namerorDr-c0tjtjj dj d}|dj tj|d<gd|_|j|d}t||}|d}tjdtd  }|dj d |d<gd|_|d k(r|j|}n|j| }tj||y)Nr-)rrr)rrr-rrFrc|Srirrfs rrgz.Osarr<r)rr<r ror;)rr=rr>r?r@rr r<rrgrBrrFr!r")r*rrHrrrrs r*test_empty_groupby_apply_nonunique_columnsr-Fs bii++A.>>vF GB qELL "BqE BJ DU +B R %F K F||)U1X>H1+$$W-HQK&He%%d+===.&(+rctjgdgdgdgdd}tjtd5|gdj dddd|j ddj }tjd gdtjd gd  }tj||y#1swYqxYw) N)rrrr)r-r-r-r-)rrMrNrz ('a', 'b')rrrrrrrPrT) rr=rrrDrrrrr!rGrVs rtest_tuple_as_groupingr/Zs $   B x} 50 ?##J/0ZZ #C ( , , .Fyy!3bhhs.LMH68, 00s CCc *tjdtdtjj ddgddgg}t j td5|jdjdddy#1swYyxYw) Nrrr-rrz ^\(7, 8\)$r)rr.) rr=rBrr@rrrDrrrs rtest_tuple_correct_keyerrorr1msq  q2==#=#=1v1v>N#O B x} 5" 6!"""s B  Bc tjdgdggtjdgdtjdddd  }tjgd gd gtjj d gd tjdddd  }|j tjdjddg}tj||y)Nrr(mycolsrP 2018-01-01r-r3dti)rrrr)rrrrr))r(rr()r(rvrt)r(rvr)r(rvr)r(rvclose)r3NNrB)rrrv) rr=rrrrDrrror!r"rs rtest_groupby_agg_ohlc_non_firstr7vs  qc %x0mmL!#EJ B || /* )) )*  mmL!#EJ HZZ , - 1 15&/ BF&(+rctjdftddddftddddftddddfg}tjj |ddg}tj gd | }|j d j}tj d d gddg }tj||y)NrMirr-rNrdaterB)rr-@rr!rrrr:) rrrrrDrrrr!rG)rUrIrrrs rtest_groupby_multiindex_natr;s  $1 s# $1 s# $1 s# F  " "6&$ " @B ))N" -C [[q[ ! & & (Fyy#sC:6H68,rcttdtdd}tj|ddg}d}t j t |5|jggdddy#1swYyxYw) NrTrVapplerNr;z$Grouper and axis must be same lengthr)rrBrr=rrrr)rUrHrs rtest_groupby_empty_list_raisesr>sb rE"Id 3F fwn 5B 0C z - B4s A33A<cddgddgddgg}gd}tjj||}tjddg| }|j d d gj }dgdgg}d d g}tjj||}tjd g| }t j||y)NrRrMrNr )rrthirdrBrr-rrr@rr!)rr from_arraysrrrr!rG) index_array index_namesrirreirs r8test_groupby_multiindex_series_keys_len_equal_group_axisrFs:SzC:6K.K  " ";k " BB 1vR(A YY) * . . 0F53%.KG$K  " ";k " BByy!B'H68,rctjjddgddggddg}tjgdgd d | }|j tj d dg}|j ddg}|j |j k(sJ|j dtj d g}|j ddg}|j |j k(sJy) Nr r r2r3rZbetarB)rr-rr-rrr!r)rrr@r=rrr)rIrHrrs r"test_groupby_groups_in_BaseGrouperrIs  # #c3Z#s$B  z 2D a a aF   Ai Ai2783S 6#;&8H8 QFf 5B YYF$' x}}t$T> LXXj) xx -  ##J = > ), *4)<$ n lN;<9sF z op, expectedshifttime2019-01-01 12:00:002019-01-01 12:30:00bfill2019-01-01 14:00:002019-01-01 14:30:00ffillc |gdtjdtjdddtjdtjdgd}tj|jfd}|j d }t ||}tj|jfd }t j||y) N)r r r r r r rSrTrVrW)r-rRcN|jjjSrirRrR tz_localizerRrs rrgz+test_shift_bfill_ffill_tz.. s!&&))2G2G2Kr)rRr-cN|jjjSrir[r]s rrgz+test_shift_bfill_ffill_tz..$sAFFII >?R SF'{""G9-778KLQQRTSTUH//??3'OOFH%rc tjddd}tjddgddgddgd}d |j_||_|j }|jd }|j}tjddgddgd | }d |j_ |j }tj|||gjd }|g}|j}tj||y)Nrr-r()rrrrr)r)r r2)r3r )r)rr)r r3r!r)) rrr=r<rrrrnuniquerr!r")r5rHrrrrrMrs rtest_groupby_crash_on_nuniquerrSs -- ae r?r@roption_contextrr__repr__rArM) max_seq_itemsrrHrs rtest_groups_repr_truncatesryss bii++A.>>vF GBhhBsG  2M B"C''002!!!BHHRTTN+22;;=!!! """s 'A>C//C8cRtjddddddddg}|jddg}|jddg}|j}dt j d dgt j i}t|dk(sJd}||||k(jsJy) Nrr-crXrMrN)rr-rr0) rr=rrrrrAr rCr)rHgrprrr s r6test_group_on_two_row_multiindex_returns_one_tuple_keyr~s QQR02KL MB sCj !B **c3Z C [[F!Qrxx89H v;!   C 3K8C= ( - - // /rzklass, attr, valuer rrrQrrctjdgdgdgd}|dk7r|jd}|jdi||i}|tjur|dgn|d}t ||t ||k(sJy) Nrr-rrr]rMrN)rM)rr=rrrg)rrr4rHrrs r#test_subsetting_columns_keeps_attrsrs& QCqc4 5B v~ \\# rzz/$/H % 5Xse_8C=F 64 GHd$; ;; ;rr)ranyrQctjdgd}tjdgg|}|jdg}t ||j }t j||y)NrMr rPrr;)rrr=rrgr<r!r)rrrHrrs r#test_groupby_column_index_name_lostrs`xxE*H seX .BQCJ &WZ & ( 0 0F&(+rcTtjgdgdgddjt}gd|_|j gdj }tjgdgtjdggdt }tj||y) N)rbrrsrrsrrY)r r r )rrrr)rrMrr)rr<r) rr=rrr<rrrrAr!r"rVs rtest_groupby_duplicate_columnsrs| ")=LQ  fVn!BJ ZZ % ) ) +F|| rxx}oVH&(+rc$tjgdgdd}d|j_|j dj }tjdd gd dgd}d|j_t j||y) Nrr)rMrMrr)rNrNrrr-rr)rrrrrlastr!rGrrrs r#test_groupby_series_with_tuple_namersn ))L : FCCIIN [[q[ ! & & (Fyy!Q1vJ?H$HNN68,rz func, valuesr@X@X@rg@8@g8@cgd}tjddgdz||d}t|jd|}tj||dtjddgd }t j ||y) N7yACrab7y1rrrrr-rr9rMrNr9rrPr!)rr=rgrrr!r")rrUrrHrrs r)test_groupby_numerical_stability_sum_meanrsy r?r@rr=rBrrrrSr!r")r$rHirrrs r test_groupby_cumsum_skipna_falsers ))   " 2 26 :C c B 1X1 BsG CB ))5) !C/")))7H#x(rc tjddd}tj||dz }tj|d<tjd|d}|j d }|j d d }tjd |d|dtj|d|ddzgi}tj|||j d d }tjd |d|dtjtjtjgi}tj||y)Nrrns)runitrr-rrr FT)rrr r) rrrrr=rrSr!r")r5rrHrrrMs rtest_groupby_cumsum_timedelta64rs -- ad ;C ))C.3q6 !C VVCF AC( )B CB ))t) 4C ,,c!fc!fbffc!fc!fqjIJ KC#s# ))u) 5C ,,c!fc!fbffbffbffEF GC#s#rc|}|jdj}|j|jj}tj||y)Nrr)rrrr!rG)(rand_series_with_duplicate_datetimeindexdupsrrs r!test_groupby_mean_duplicate_indexr sK 3D \\\ " ' ' )F||DJJ',,.H68,rctjgdtjtjtjg}|j |j j }tjgtjgtjtj}tj||y)Nrr0r;) rrrrrrrrr<r r!rGrrrs r test_groupby_all_nan_groups_dropr su )bffbffbff56A YYqww  # # %Fyy288Bbjj#ARH68,rctjggd}|jddg|}|j|}|r;tjggggggddg}|sdgntj gd }ntj d }|sgdnddg}tjg|| }tj||y) NrYrkr r r~rrBr2r2r0rr) rr=rrrrrr!r")rrrHrrrr<rs rtest_groupby_empty_multi_columnr s 2 7B S#J 2B VVV .F r2hRc C+3%"E1J a )5/C:||Bu=H&(+rc4tjddggddggddggddgggddg}tjdddgddggitjddgd  }|jdg }|j }t j ||y) NrrWrrMWr r;rPr!r)rr=rrrr!r")rHrrrs r*test_groupby_aggregation_non_numeric_dtyper. s  scA3Z#tsRDk:T3K B|| 1a&2r(# hhSz- H v B VVXF&(+rc tjgdtddDcgc]}tj|dc}tddDcgc]}tj|dzdc}d}tjtddDcgc]}tj|dc}tddDcgc]}tj|dzdc}d tjd dgd d }|j d g}|j }tj||ycc}wcc}wcc}wcc}w)NrrrrrrrdaysrrrrrLrrr rRrr!r) rr=rBrrrrr!r"rrHrrrs r0test_groupby_aggregation_multi_non_numeric_dtyper@ s  38A;?a",,q&)?8=a D1",,q2vv.D  B||38A;?a",,q&)?8=a D1",,q2vv.D hh1vW37 H u B VVXF&(+@D @DsD>  E E E c tjgdtddDcgc]}tj|dc}t tddd}tjtjddtjddgddgdtj d dgd d  }|j d g}|j}tj||ycc}w)Nrrrrrrr.rrr rRrr!r) rr=rBrrrrrr!r"rs r7test_groupby_aggregation_numeric_with_non_numeric_dtyperW s  38A;?a",,q&)?eAqk"  B||||Av& Q(?@1vNhh1vW37H u B VVXF&(+@sC3 c@dddddddddddddddg}tj|}||ddk(}|jd}|j}tjgd ggd tjdgd  }t j ||y) NFTg%@) filter_col groupby_colbool_col float_colg4@g>@rr)ririg$+ H@)rrrrPr)rr=rrrr!r")dictsrH df_filterdfgbrrs rtest_groupby_filtered_df_stdrk sTtRVWDdQUVDdQUV E e B2l#t+,I   ] +D XXZF|| 7hhvM2H &(+rc Ztjtjtdtjtjgddtj j gdddgd }|jd d gd j}d tjdftjdgd tjdftjdgdtjdftjdgi}||k(sJy)Nabc)r4z 2018-02-01z 2018-03-01categoryr0)rrrrrrrrMrNFrz2018-01-01 00:00:00z2018-02-01 00:00:00rz2018-03-01 00:00:00r-) rr=rr to_datetimer from_codesrrrJrrArVs r2test_datetime_categorical_multikey_groupby_indicesr s 4;'IJ **:1a&*I   BZZc UZ 3 ; ;F bll012BHHaSM bll012BHHaSM bll012BHHaSMH X  rc dgdzdgdzz}tjtjdddtjddddg }tj||gj}dd g|_d }t jt| 5|jdd jdd jdddy#1swYyxYw) NAlicerBobrr-rrrrCz@^[a-zA-Z._]*\(\) got an unexpected keyword argument 'min_period'r)window min_period) rrrr=rr<rrr:rrollingr)name_lval_ltest_dfresult_error_msgs r#test_rolling_wrong_param_min_periodr sY]eWq[ (F VVRVVQ1bffaAq 9EllFE?+--GuoGO L y(8 9M&..aA.FJJLMMMs 4CCc<|}tjgdgdtjgd|d}tjjdd}d}t j t | 5|jd gj|}dddtjd d gd gd dgddgdjd }t|dddk(r tjd d g|}|d g|d<t j|y#1swYxYw)N)Thomasr Thomas John)ii)sadhappyrr0)NameCreditMoodr)rrz6Converting a Series or array of length 1 into a scalarrrrri rrr)rrrstoragepyarrowr) rr=rmoder!rrrrorrgrAr")any_string_dtyperrHaggregate_detailsrrr mood_valuess r.test_by_column_values_with_same_starting_valuer s E 7'II7uE  B"$5A BC # #N# >=VH%))*;<=||u%w/Sk}-  i  ui&)3hh/u= '1&(+==s 1"DDcjgdgdg}tjdtjj|ddg}|j ddg j }tjdd gtjj d d gddg}tj||y) N)Nrrr)r-rr-rrrMrNrBr!rrr-)rir-)rr) rrrrArrrDr!rG)r$rrrs r#test_groupby_none_in_first_mi_levelr s L )C ))AR]]66s3*6M NC [[1v[ & * * ,Fyy A ))8X*>sCj)QH68,rcNtjgdgdgdd}|rtjgndg}|j |}|j }tjddgdd gd tj d dg|d  }tj||y)Nr)rrr-r)rrrr)NrNrrr-rrrSrrrrPr!) rr=rrrrrr!r")r;rHr\rrrs rtest_groupby_none_column_namer s \ <P QB'"&&dVB r B VVXF||!fAr7#288QFA+GH&(+r selectionctjddgtjdgtjdgdddg}||j dgn|j dg|}|Dcgc]\}}| }}}d d g}||k(sJycc}}w) Nrr-rrrRrLr!rMrr-)rr=rrr)rrHr$r rWrrs r!test_single_element_list_groupingr s !fBFFA;bffa[9#s B$-#4bjj#"**cU:KI:VG ' (fc1c (F (d|H X  )s. B cBtjgdgdd}|djd|d<tjgdgdd}|djd|d<|jdd }|j }t j ||y) N)rMrNrrM)rr-rr-)str_colnum_colrstringr)rrqrrFr~)rr=rrrr!r")rHrr$rs rtest_groupby_string_dtyper s "6<P QByM((2ByM|| '   H#9-44X>HYjjUj3G \\^F&(+rzlevel_arg, multiindex)rF)rTctjddgddgddgddd g }|r|jd d g}|j| Dcgc]\}}| }}}|rddgnddg}||k(sJycc}}w)Nrr-rrrrrrRrLr!rMrNrrrrf)rL)rr=rr) level_argrrHr rWrrs r+test_single_element_listlike_level_groupingr s QF!Qq!f=c3Z PB \\3* % "  ; C5A  fk2 ]] H  0 HH,H / 0 0 0s B66B?c~tjgdgdd}tj|dtjddd}|j |d gd dj d }tjtjd ddtjdddtjdddtjdddgd}gd}tjj||gdd g}tjtjdtjdtjdtjdtjtjdtjtjg |d}tj||y)N)rOrPrQ皙ٿ?)rMrNrMrrM)rRwrRrrrrmr$FrrgDlٿrOro)closedrPrQr#T)r!rrBr"r)rr=qcutrlinspacerrorIntervalrr@rrArr!rG)rHqqrcategorical_index_level_1 index_level_2rIrs r/test_grouping_with_categorical_interval_columnsr, s` 6=VW XB CBKK1a0 1B ZZS EZ 23 7 ; ;F CF " KKG 4 KKS 1 KKS 1 KKS 1  !$M  # # "M23* $ Byy    %H(68,rbug_varchtjd|||tjgi}t |t r|j t}|jd}|jd}tj|||dgdg|dj}tj||y)Nr c|Srirrfs rrgz;test_groupby_sum_on_nan_should_return_nan.. srr) min_count)r<r) rr=rrr0r2rrrrrr!r")r-rHrrrs r)test_groupby_sum_on_nan_should_return_nanr1 s sWgw?@ AB'3 YYv  ::k "D XXX "F,, '7D)C53 K&+.r)rcorrcummaxcumminrrrankrqrlrQrrrr7rXrU pct_changectjdt|}||_|j dgdg}|dgj |d}t ||}t ||}t j||y)N2014rr r2)rrrCrrrgr!r")rHrrngrsg_exprrMs r#test_groupby_selection_with_methodsr< s0 --B 0CBH C53% A uI  bg &E '!V  C '% "C#s#rctjdt|}d|j_||_|j dgdg}|dgj |d}tj|jd|jdtj|jdj|jdjtj|jdj|jdjtj|jd |jd y) Nr8r9r(r r2c"|jSrir?rfs rrgz6test_groupby_selection_other_methods.." s AEEGrc"|jSrir?rfs rrgz6test_groupby_selection_other_methods.." s AEEGrr3ct|dk(Srrrfs rrgz6test_groupby_selection_other_methods..( s3q6Q;rct|dk(Srrrfs rrgz6test_groupby_selection_other_methods..( sA! r)rrrCr<rrrr!r"rDresamplerrvr)rHr:rsr;s r$test_groupby_selection_other_methodsrC s  --B 0CBJJOBH C53% A uI  bg &E!''"34ekkBS6TU!**S/..0%..2E2J2J2LM!**S/..0%..2E2J2J2LM &'6K)Lrctjgdj|}tjgdgd|d}tjddd|}tj|gdgdd }|j tj d d }|jj}tj||y) N)z2016-08-31 22:08:12.000z2016-08-31 22:09:12.200z2016-08-31 22:20:12.400)rrrr)quantquant2time2z2016-08-31 22:08:00rS1min)rrr) rrrrrrrrrrrrr)rGrErFrG)r r) rras_unitr=rrrrrir!r")ridx2 test_datarGexpected_outputrrs rtest_groupby_with_Time_GrouperrM, s >>   gdm   !_tLI MM/&t TEll<= O   2::'? @B XXZ # # %F&/2rc tjgdtjddd}|j|j }tjddgd }d|j _tj||y) Nrrrz 2022-01-01rr9janrr-rrP rrrrrrrr!rGrs r1test_groupby_series_with_datetimeindex_month_namerRH sf )2==q#IPUVA YYq\   !Fyy!Qe,HHNN68,rc"tjgdtjdddd}|j|j }tjdd gd }d|j _tj||y) NrOz 2013-06-07r15minr09rr-rrPrQrs r3test_groupby_series_with_datetimeindex_numeric_namerVQ sq ))mmL!'B  C [[  # # %Fyy!Qd+HHNN68,r test_serieszkwarg, value, name, warn)r\rMrNr\r)r rrNctjdgditjgdd}|r|d}|jd i||i}|j |}|r2tj ddgtjd d gdd }n2tjdddgitjd d gd}t j||y) NrN)rrr)rrr-rMrPr!rrrrr)rr=rrrUrr!r") rWkwargr4rwarnr%rrrs rtest_get_group_len_1_list_likesr[_ s ,,Y'rxx /L MC#h  &u~ &B \\$ F99aV288QF+ECP<<q!f RXXq!f35OPOOFH%rctjtjtjgdgd}|j ddgddj }tjdg}tj||y)NrrrMrNF)rrr) rr=rrrrrbrr!rGrVs rtest_groupby_ngroup_with_nanr]w sf BNNBFF84A3? @B ZZc 55Z A H H JFyy!~H68,rc (tjdddddtjtjgigd}|j d j }tjdgd igd}t j||d y) NrMrr-rrrtr!rr)rr-rrr-rF) check_dtype)rr=rrrrXr!r"rVs r(test_groupby_ffill_with_duplicated_indexr` so sQ1a89AS TB ZZaZ & & (F||S"45=OPH&(>rc ttjtrJt j t dt dddgt dt dt dt dgd}|jdd }|r|d }|jj}t jt ddgd }tj||y) Nrr-rrr)r r4r Frr4rP) r0decimalInvalidOperationr:rr=rrrY result_indexrr!r)rWrHrrrs rtest_decimal_na_sortrf s '22I>> > AJ D$7aj'!*gaj'!*E  B E% (B [ [[ % %FxxT*7H&(+rcgdgdgdgdg}tj|gd}|jgdd jd d d gi}tjj d dt jdfdt jdfdt jdfggd}tjj ddg}tjddgfddgfddgfddgfg||}tj||y)N)rrrr )rNrr )rNr-r )rNrr )rMrNrpartnerr;rFrbrhrqunique)rrrrr-rrB)rhrq)rhrir r) rr=rrorrDrrr!r")rH df_dropnarrr<rs r'test_groupby_dropna_with_nunique_uniquerk s +->@Q RB R)CDI   u  = A A Y)*F MM % % q"&&!nq"&&!nq"&&!nE & Emm'')?AV(WXG|| cUa#Z!cUa#Z8wH&(+rctjgdgdgdgdgdd}|jgdd jtjd d tjd d tjdd }tjgdgdgdgdgdd}t j ||y)N)r-rrrr-r)rr$rrr)g@g @ rrr)rrrHrr)rrrrrr)r#r$col3col4col5)r#r#r$Fr[r#r)rrrr$r)new_colnew_col1new_col2)rrrrr-)rrrrmr)rrrrr-)r#r$rsrtru)rr=rroNamedAggr!r"rVs r0test_groupby_agg_namedagg_with_duplicate_columnsrw s &'.)*   BZZ3eZ D H H 659FE:FG<IF ||#$&''  H&(+rc Jtjgddtddtdggdd}|jddgd j }|j }t j|tjj|jy) Nrrrr-)r-rrr.rYr r Frb) rr=rKrrrr!rr from_framer)rHrrs rtest_groupby_multi_index_codesrz s} !U5\1eEl!C,W BS#Ju599;J   E%!9!9%..:J!KLrctjtjdtjdtjdtjggdd}|j dd}t |}|dk(sJy) Nz 2019-02-12z 2019-02-13rrrMFrbr)rr=rrrrC)rHr$rs rtest_groupby_datetime_with_natr| su |,|,|,    BjjUj+G \F Q;;rc6tjtjj djdtj dd}|j d}|j}t|jdtsJy)Nr-rcz1/1/2000r9r!c2|j|jfSri)rrkrfs rrgz2test_groupby_function_tuple_1677.. s!&&!''):rr) rr=rr>r?rrrr0rrw)rH monthly_grouprs r test_groupby_function_tuple_1677r sr  a '',mmJ4 BJJ:;M    !F fll1ou -- -r)rrcrr8numpyrr pandas.errorsrrpandasrpandas._testing_testingr!pandas.core.arraysrpandas.core.commoncorecommonrxrr%rJrXrzrrrrmark parametrizerrrrr rr&r(r2r<rNrWr^rkrorxr|rrrrrrrrrrrrrrrrrrrrrrr rrrrrr'r/r8r>rArErLrPr^rarlrpr~rrrrrrrrrrrrrrrrrrrrrAr)r-r/r1r7r;r>rFrIrPrJrarcrerorrrtryr~r=rrrrrrrrrrrrrrrrrrrrrrrrrrrrrr rrrrr r,r1r<rCrMrRrVr[r]r`rfrkrwrzr|rrrrrs@    + D-*596E4 $,@3%#s!45969,6#r0%&P868+-+`6r 13EFG-H-@ -*,D.2D(&,.,4 +( EN, ,:$708  0@ '0( + (..D-,$,. ';D,,( ,-"(&, , /., -B* , C&L 8#$*0Z0-0, -S3*$5674 #/@*5:$*/4d-: , Xy68"4vy6IJ\?\?4ST 2 (S3%L13%#s!45    s   aA,/ ac:!G$!I&% *  :#;< Kz&=;62Fz&z(891#1v/,0:,$-&",8 -  - ,<:  BLL!67 BLL!67     BLL!67 BLL!67 BLL!67 BLL!67 BLL!67 BLL!67     BLL!67 BLL!67 BLL!67 BLL!67 BLL!67 BLL!67   7(+X,Y+X,,--&.$2, " " 0 w$ z5) vu% |U+ z4( x' GS! J& FE" L%( J% He$ "<#"<!89,:, ,-edD\*VeT],CD,, > ) $$--$7 ,8 ,,$,.,(,(* M,8- ,tS3%&89 : ,(sElMQC; U!EF,G,z;&?@ ,A , m'DE,F,0UBEEN(CD,E, & 0  + 8 !  , , 45,6,#KL,M,(*>,$7080"&-RQH- /. / , $-, $ (38- -u 6 udD! 1#tT"  &7 &-?u 6,7,&,(,>M$ .r