j ddlZddlZddlmZddlZddlZddlm Z ddl m Z ddl m Z ddlZddlmZmZmZmZmZmZddlmZddlmZddlmZej<j?d gd d Z ej<j?d gd gdgdgdgdgdgdgdgdgdgdgdejBejBejBgg dZ"dZ#dZ$ej<j?dgdgdejJejJejJgdejJdgdejJdgdejJdggdZ&ej<j?d d!d"ejNdge(#ejNejJgd$#fd%d!ejNejJgd$#ejNdge(#fd%d%ejNejJgd$#ejNejJgd$#fd"d$ejNejJgd$#ejNejJgd$#fgd&Z)ej<j?d gd'd(Z*ej<j?d)d*ejJejBgdfd*ejJd+ejBgdfd,ejJejVgdfd,ejJdejVgdfgd-Z,d.Z-d/Z.ej<j?d0d1d2gd3Z/ej<j?d0d1d2gd4Z0ej<j?d5d1dd6gd+d7gd+d6gd8fd2d+d7gdd6gdd7gd8fgej<j?d9ddgd:Z1ej<j?ded;ed<fd=gej<j?d>gd?d@Z2ej<j?d0dAdBgdCZ3dDZ4dEZ5ej<j?dFdd+ejBd7dGdHdIdJdKdLg dMdMfdd+ejJd7dGdHdIdJdKdLg d"d"fdd+ejJd7dGdHdIdJdKdLg d%d"fdd+ejBd7dGdHdIdJdKdLg dNdNfejldOejVdPdQdRdSdTdUdVdWg dXdXfgdYZ7ej<j?dZdd+ejBd7dGdHdIdJdKdLg dMfdd+ejJd7dGdHdIdJdKdLg d"fdd+ejJd7dGdHdIdJdKdLg d%fdd+ejBd7dGdHdIdJdKdLg dNfgd[Z8d\Z9ej<j?d]gd^dd+d7ejBdGdHdIdJd_dLg dMdMfd^dd`d7dGdHejJdIdJdKdLg d"d"fd^dd+d7dadHdIdJd_ejJdLg d%d%fd^ejBgdbzdMdMfd^ejJgdbzd"d"fd^ejJgdbzd%d%fdcdd`d7dGejBdHdIdJdKdLg dMdMfdcdd+d7dadHdIdJd_dLejJg d"d"fdcdd`d7dGdHdddJejJdKdLg d%d"fdcejBgdbzdMdMfdcejJgdbzd"d"fdcejJgdbzd%d"fdedd+d7dadHdIdJd_ejBdLg dMdMfdedd`d7dGdHdddJejJdKdLg d"d"fdedd+d7dadHdIdJd_dLejJg d%d"fdeejBgdbzdMdMfdeejJgdbzd"d"fdeejJgdbzd%d"fdfdd`d7dGdHdddJejBdKdLg dMdMfdfdd+d7dadHdIdJd_ejJdLg d"d"fdfdd`d7dGdHdddJdKdLejJg d%d"fdfejBgdbzdMdMfdfejJgdbzd"d"fdfejJgdbzd%d"fdgdd`d7dGdHdddJejBdKdLg dMdMfdgdd+d7dadHdIdJd_ejJdLg d"d"fdgdd`d7dGdHdddJdKdLejJg d%d%fdgdd+ejBd7dGdHdIdJdKdLg dNdNfdgejldOejVdPdQdRdSdTdUdVdWg dXdXfdgejBgdbzdMdMfdgejJgdbzd"d"fdgejJgdbzd%d%fdhdd`d7dGdHdddJejBdKdLg dMdMfdhdd+d7dadHdIdJd_ejJdLg d"d"fdhdd`d7dGdHdddJdKdLejJg d%d%fdhdd+ejBd7dGdHdIdJdKdLg dNdNfdhejldOejVdPdQdRdSdTdUdVdWg dXdXfdhejBgdbzdMdMfdhejJgdbzd"d"fdhejJgdbzd%d%fdidd`d7dGdHdddJejBdKdLg dMdMfdidd+d7dadHdIdJd_ejJdLg d"d"fdidd`d7dGdHdddJdKdLejJg d%d"fdidd+ejBd7dGdHdIdJdKdLg dNdNfdiejldOejVdPdQdRdSdTdUdVdWg dXdXfdiejBgdbzdMdMfdiejJgdbzd"d"fdiejJgdbzd%d"fdjZ:dkZ;dlZdoZ?dpZ@dqZAej<j?drdgdhgdsZBej<j?drdgdhgdtZCduZDej<j?drdgdhgdreEfdvZFej<j?d gdwdxZGdyZHej<j?drgdzd{ZIej<j?drdgdhgd|ZJej<j?d}dd+gej<j?d~ddgdZKdZLej<jej<j?dddgej<j?ddgddggdZNdZOdZPdZQdZRej<j?dgdedejVedgded+gejdgddfgdejdd+d+ejVejdd+d+gded+gejdgddfgdejdd+d+ejVejdd+d+ejVejdd+d+gded6d6gejddgddfgdejdd+d+ejVejdd+d+ejVejdd+d+gded6d+gejddgddfgdZUdZVdZWdZXdZYej<j?d}ddbgdZZdZ[dZ\ej<j?d gdej<j?ddAdd+d+dd6d7dgifdBdd+d6dd6dGdgifdgdd+d+dd6d7dgifdhdd+d6dd6dGdgifdd+d6dd6d6dgddfgdZ]dZ^ej<j?ddejfdiejfdeejfdcejfdejfd^ejfdgejfdhejfdAdfdBdfdejfejdfe^ejdg dZjej<j?dgdd+ejJd6d+ejJd6d+ejJd6g dgdgddgej<j?dgddZkej<j?dgddZldZmdZndZoy)N)ascii_lowercase)iNaT) pandas_dtype)na_value_for_dtype) DataFrame MultiIndexSeries Timestamp date_rangeisna)get_groupby_method_args)_test_decoratorsdtype)int64int32float64float32cttjddztjd|}tjd}tjj dj ||j |}|jdd}|D]\}}t|dk(rJ|jtj}|ddk(sJ|jtj}tj||tj||j|j}|jtj}|d k(r|jd }tj|||j!d }|j#d } | d d k(sJtj| ||j|} | jtj}tj||d|jtjtj$g}d} t'j(t*j,j.| 5|jtjtj$ddddd} t'j(t0| 5|jddddy#1swYBxYw#1swYyxYw)N indexrc |dzS)Nrxs e/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/groupby/test_reductions.pyz)test_basic_aggregations..#s Q!VF) group_keysrc(||jzSNsumrs rrz)test_basic_aggregations..5sq1557{r c(||jzSr$r%rs rrz)test_basic_aggregations..6sa!%%'kr  )check_index_typeznested renamer is not supportedmatch)onetwozMust produce aggregated valuec |dzS)Nrrrs rrz)test_basic_aggregations..Is AEr )r nparangerandom default_rngshufflereindexgroupbylen aggregatemeanaggtmassert_series_equalr&astypeapply transformstdpytestraisespderrorsSpecificationError Exception) rdatargroupedkvaggedexpectedresult transformed value_groupedmsgs rtest_basic_aggregationsrQsS "))A,!#299QG11v{{   bgg &E 8q=={{277#H5(+5',,.1 [[]F{{266"H ??7+68,}}23H##$9:K q>R  ;1LL&M  $ $RWW -F655A   rww/ 0E ,C ryy333 ?;"''"&&9:; *C y ,+/*++ ;; ++s1K5L5K>L vals)foobarbaz)rSrV)rVrVrVr"rr)r"rrrrr)?@@)rYr\)r\r\r\)TTT)TFF)FFFcftdgdzdgdzz|dzd}tt||}|rtt |r|dk(rd}t|gdzdgt j ddgd  }t|jd || }tj||y) Narbr)keyvalanyFrar`namecolumnsrskipna) rgetattrbuiltinsallr rCIndexr6r;assert_frame_equal)rhall_boolean_reductionsrRdfexprLrMs rtest_groupby_bool_aggsrqLs& C519uqy0B CB 4'(2 3D 9C#d4j/&<&E  E7"((C:E*JH@WRZZ&(> ?v NF&(+r c,tgddtjdgdtjdgggd}tddgd dggd d gddg }d |j_|j d j }tj||y)N)r"rrSr"rTrrUABCrfTFrurvrert) rr0nanrrdr6rbr;rmrorLrMs rtest_anyrzos  BFFE*Q,>? B t}%Sz!QHHNN ZZ_ "F&(+r ctddggddg}|jdg}t||}|jt j dg}t j||y)NTr^rwr)rr6riset_axisr0arrayr;rm)rnrogrp_byrMrLs r test_bool_aggs_dup_column_labelsr|sa T4L>C: 6B ZZ_F 4WV3 4 6F{{288QC=)H&(+r rGFTct|d}t|||}t|tjdgd}|j gdj ||}t j||y)NbooleanrrgrrrX)r rir0r}r6r:r;r<)rnrhrGser expected_datarLrMs rtest_masked_kleene_logicrsm Y 'C9GC!78GMm288QC= JH [[ # ' '(>v ' NF68,r zdtype1,dtype2,exp_col1,exp_col2floatFloat64rrInt64c|jdrdtjgndtjg}|jdrdtjgndtjg}t tj ||tj ||d}|jddgjdd}t ||dtj dg }tj||y) NfrYr)col1col2r"rkFrgr) startswithr0rxrCNArr}r6r:r;rm) dtype1dtype2exp_col1exp_col2data1data2rorMrLs rtest_masked_mixed_typesrs>$..s3S"&&M#ruuE#..s3S"&&M#ruuE %v.f8UV BZZA  # #E% # 8F(H=RXXqc]SH&(+r )rrrc|tjdg|}d}|s|dk(rtj}||gtjdgd}|j ddgj ||}t j||y)Nr"rTrkrrrg)rCrr0r}r6r:r; assert_equal)rnrrhframe_or_seriesobj expected_resrLrMs rtest_masked_bool_aggs_skipnars| 255!*E 2CL ,5uu  ~RXXqc])TH [[!Q $ $%;F $ KFOOFH%r zbool_agg_func,data,expected_resrbr"rkc||t}|jdgt|zj|}||gt j dgd}t j||y)Nrr"boolr)objectr6r7r:r0r}r;r) bool_agg_funcrGrrrrMrLs rtest_object_type_missing_valsrs[ $f -C [[!s4y ) - -m RSH&(+r c tj|j}tj|j}t t tj tddddt tj|tj||tj|tjg|d}|jd}|s=|d }tjt| 5t||| dddyt||| }t d gd itjtdddd} t!j"|| y#1swYyxYw)Nr"r)repeatsintprrr^z with skipna=Falser+rgr_)r"rrr)r0rrrrr repeatrangerxr6rArB ValueErrorrirCrlr;rm) rhrfloat_numpy_dtyperrrorrPrMrLs r"test_idxmin_idxmax_extremes_skipnar s2*+//I*+//I  %1+q9HFFFFFFFF (   B$ CB '( ]]:S 1 , GB F + , WR V ,F l288E!QKc#PH&(+ ,s 5E--E6z func, valuesrr)c_intc_floatc_date numeric_onlyctgdgdgdgdd}tj|d|d<|djj d|d<|d|dj d z |d <|djj d |d <|d jd|d<|djd|d<t|jd||}t|tjddgd}|r|jdg}n|d|d<|d|d <|d|d <|d |d<|d|d<tj||y)Nrtrtrurur"rrr)gGz@g= ףp=@gRQ@g?)2019201820162017)rdrrrrz US/Pacific c_date_tzr c_timedeltaWc_periodrr c_Integerrr c_Floatingrdrrtrurcrrw)rrC to_datetimedt tz_localizeiloc to_periodr=rir6rldropr;rm)funcvaluesrrorMrLs r$test_idxmin_idxmax_returns_int_typesr2si (!/6   B>>"X,/BxLloo11,?B{O8 r(|'8'8';;B}\__..s3BzNk((1B{O)}++I6B| .WRZZ' .L IFrxxc 'HIH==(=4 ( 2"*8"4'1$W-H[%i0H\&(+r z2011-01-15 12:50:28.502376z2011-01-20 12:50:28.593448)l m;-Ll m;-Lmethod)countrrfirstlastcDtddg|d}|jd}t||}|dk(rd}n*|dk(r|d}n|dk(r|d}nt|d |}td |gitjdgd  }t j ||y) Nr"rr^rrrrrr_rcr)rr6rirCrlr;rm)rrGrorHrMexpected_valuerLs r2test_groupby_non_arithmetic_agg_int_like_precisionrZs !Qd+ ,BjjoG %WWf % 'F  7 a 6 a1C&13#/0!38OPH&(+r rrc ntt|}tgd|d|d||g|d|d||gd|}|jd|}t ||}||}gd gd gd gd d ||f} |j | j d} |r| j} tj|| y)N)rr"r"rrrr[@r^r_crr^)sortrg)rr"r)rr"r)rr"r)rrr))rT)rF)rT)rF) rrrr6rir set_index sort_indexr;rm) any_real_nullable_dtyperrhrna_valuerorrrMilocsrLs rtest_first_last_skipnarxs",/F"GHH #C3(CC3(C  & B Cd #B R F 6 "F##!"    6k  E wwu~'',H &&(&(+r c|tgdgdd}|jddjddk(sJy)N)rtrtrtrtrt)ijiioi`l@~)user connectionsrrrtl,fL)rr6r9)ros rtest_groupby_mean_no_overflowrsD -I  B ::f m , 1 1 3C 8.sv.r rg)rr=r6r>r9r;r<)rr result_dtyperhrorLrMs ` rtest_mean_skipnars: :>    feU^  5% . /   ZZ u % * *& * 9F68,r z values, dtypec,t|ddgdzdjd|i}|jddjfdj|}|jddj }t j ||y) Nrtrurrrarc(|jSrr%rs rrz!test_sum_skipna..sf1Er rg)rr=r6r>r&r;r<)rrrhrorLrMs ` rtest_sum_skipnars :>    feU^  5% &&'EFMMeT ZZ u % ) ) ) 8F68,r c tddtjddddddd g d d gd zd jdti}|r;t ddgt jd d gddjt}nHt tjdgt jd d gddjt}|jddj|}tj||y)Nr^r_derghijrtrurrraaegibdfhjrrcrrdrg) rr0rxr=rr rCrlr6r&r;r<)rhrorLrMs rtest_sum_skipna_objectrs bffc3S#sCH:>    feV_   W RXXsCju%EE &.  VVW RXXsCju%EE &. ZZ u % ) ) ) 8F68,r z!func, values, dtype, result_dtypeprodirvarir@semrrmedianc2t|ddgdzdjd|i}|jddjfdj|}t |jdd}t j ||y) Nrtrurrrarc*t|Srri)rrrhs rrz'test_multifunc_skipna..s)D)8r rg)rr=r6r>rir;r<)rrrrrhrorLrMs` ` rtest_multifunc_skipnarsx :>    feU^  5% 8 9   5WRZZ&u-t 4F CF68,r cPtjjdjd}tj|ddd<t |}tjjdj dddjt}tj|ddd<|j|j}|j|jtj}tj||t tjjdjd}|j|jtj}|j|j}tj||y)Nrr2size)rr)r0r2r3standard_normalrxrintegersr=rr6rr: nanmedianr;rm)arrrolabelsrMrprsxps rtest_cython_medianr+s) ))   " 2 24 8CvvC!H 3B YY " "1 % . .q"4 . @ G G NF66F4R4L ZZ  & & (F **V   .C&#& 299((+;;IF GB F    *B F  " " $B"b!r cvttjjdj ddd}t ddd}t j|d|}|j||j}|j||jd}tj||y) Nrr,7robservedc"|jSr$)rrs rrz(test_median_empty_bins..s r ) rr0r2r3r%rrCcutr6rr:r;rm)r1rogrpsbinsrMrLs rtest_median_empty_binsr6s 299((+44QC@ AB B?D 66"Q% D ZZxZ 0 7 7 9Fzz$z2667KLH&(+r cbtgdgdddgzd}|jdj}d|vsJ|jdjd }d|vsJ|jdj}d|vsJ|jdjd }d|vsJy) N) r8r9rrmama)nniissr;r=Fr)rr6rr)aarMs rtest_max_min_non_numericr?s *,a6(lS TB ZZ  ! ! #F 6>> ZZ  ! !u ! 5F 6>> ZZ  ! ! #F 6>> ZZ  ! !u ! 5F 6>>r ctgdgdgdd}|j|r$t|jjdk(sJ |j d}|dgj }tjgdd }tdgd i| }tj|||dgj}tjgdd }tdgd i| }tj||y) N)r"r"rrr)r"rSrrTF)r^r_rr r rsrrtrvrWrc)r_r r r)r^rr ) r_consolidate_inplacer7_mgrblocksr6rrCrlr;rmr)using_infer_stringrorrMeirLs r$test_max_min_object_multiple_columnsrFs  ,*  B'93rww~~ ! #@ @q CB Y]]_F )# &B#/r:H&(+ Y]]_F )# &B#/r:H&(+r cLtjtgddjj}t t jdt jggd|d}|jdd d jd }tjtd d gd djj}tj|||jdd j}d|j_ tj||y) N)rrrz%Y-%m-%d)format1)rr"r"rr_Fas_indexrrrc)rCrr rdaterr0rxr6rr;r<rrd)datesrorMrLs rtest_min_date_with_nansrNs NN9:: b  "&&#rvv.YUK LB ZZeZ ,S 1 5 5 7 VVaV Fq$irxx0H68>VVaV Fq"&&k"H68>r c^tdtdtdzg}|jgd}|jd}ttj tj tdzd}|j jt|_tj||dy)Nr"r)r"rrrrPrWTrR) r rr6rr0rxrr=rr;r<)rrrMrLs rtest_max_inat_not_all_narVs !T1dQh' (C \ "B VVaV F"&&RVVq9:H^^**3/HN68>r rcddg}tjddd}t||d}t|j dd|}tj ddgd }t ||d }tj||y) Nr"r2020Yperiodsfreqrr^r_rcr) rC period_rangerrir6rlr r;r<rgroupsr[rorMidxrLs r$test_groupby_aggregate_period_columnras~VFoofac:G g. /B 0WRZZ_S)4 0 2F ((Aq6 $CgSs3H68,r cddg}tjddd}t||d}t|j d|}tj ddgd}td |i| }t j||y) Nr"rrXrYrZrr^rcr_r)rCr]rrir6rlr;rmr^s r#test_groupby_aggregate_period_framerc"s{VFoofac:G g. /B +WRZZ_d + -F ((Aq6 $C#ws3H&(+r ctgdtdtjgdzdj t }|j dj}tddgdd gtjtjgdjdj t }tj||tgdtdtddj t }|j dj}tddgdd gd d gdjdj t }tj||y) Nrabcdr)r`rrr`rtrur^rrr) rlistr0rxr=rr6rrr;rmrrs r#test_aggregate_numeric_object_dtyperg0s& $d6lRVVHqLQ  fVnZZ  " " $F#Jc RVVRVV @F  C:FD ICNNAt9!QN4[  H&(+r c |}tdgddd|jdd}t|jd|}tddg|t j g|d}t j||y) Nr^rrrr_rr)rfrr)rrrir6rCrlr;rm)rstring_dtype_no_objectrrorMrLs rtest_min_empty_string_dtypersx #E #Ss35 A F Fr JB +WRZZ_d + -Fc %rxx%c/RH&(+r rQ test_seriesc|dk(ry|}|dvrd|i}n|dvrd|i}n |dvri}n||d}|tj}}|dvr d }|d k(r| nd }nd|d vrA|r3|d k(r.|jdk(r|jtjurd}nd}|d k(rdnd}n|r|dkDrtj}n|dk(rd}t dgtjgd|} |r| dn| } t || } | j | d} t| |} |dvr6d|d|d}tjt|5| | i|dddy|dvr:|r|d }n|d!}tjt|5| | i|dddy| | i|}tjdgd|"}|s|d k(r|s|d k(rdnd}t||||#}nt d|i||$}tj||y#1swYyxYw#1swYyxYw)%Ncorrwith)rbrkrrr9rr@rrh)kurtrQ)rnuniquequantilerr")rhrQ)rkrbrrbT)rrr"r"pyarrowrrr"rr&rVrrrr_r^) r9rrrrrskewr@rzdtype 'z' does not support operation ''r+)rrz+ with skipna=True encountered all NA valuesz+ with skipna=False encountered an NA value.r)rrrdr)rCrstoragerrr r6rirArBrrrlr r;r)rreduction_funcrhrQrrrexpected_dtyperrorargsrrrPrMrrdrLs rtest_string_dtype_all_narsX # "E  F# 8 #y) J J";%*BEENN''5'>VD 7 7 &( *"%%'$N$N,6A y1} 5  #beeW-U ;B "S'bC ">3 7D RW B R (F  w<^V+Ct.^RVWc>2%~VOOFH%' $  $s+ G%) G1%G.1G:c ~tgdgdttjtjdggdd}|j d}|dgj }|dj j }tj|||djjrJy)N)z-04-23z-05-06z-05-07)z2013-04-23 00:00:00z2013-05-06 00:00:00z2013-05-07 00:00:00OE)z log080001.logzlog.logxlsx)z Unnamed: 0DateappFilerr) rr r0rxr6rto_framer;rmr rb)rorrr s rtest_max_nan_bugrs 8 2662664018  B F B F8 A 6 !!#A!Qy~~##%% %%r with_nankeysjoejimc d}d}tdd}ttjj dj t t|tjj dj ||tjj djd||d}|r{|jdti}d|jd dd d f<d|jd dd df<d|jddddf<d|jddddf<d|jddddf<|j} |j|||} | dj|} |j|||} | djt j|} |s| j#d} |rt%j&| | dnt%j(| | dt%j(|| y)Nrsrz 2015-08-23)r[rr)rrjulierr"r#rr%rr(rr)rKrdropnaT)rF) check_names)r rr0r2r3choicerfrr%r=rloccopyr6rr>r reset_indexr;r<rm) rrrKrrnmdaysro original_dfgrleftrights rtest_series_groupby_nuniquer1s A A lB /D 99((+2243H!L99((+224;YY**1-66q!Q?  B YY' (#qu"ue|#qu"ue|!%qu"ug~!%qu"ug~!%qu"ug~'')K D8$ 7B g;  f  -D D8$ 7B wK  fnnV   dEu="k*r cNttdtdtdd}ttdgdgdd}|jdd j}t j ||td|_d|j _|jd }|jdj}t j |||jd dijdjd }t j ||tdgdzdgdzdtd}d|j _|jd dijdj}t j ||y)Nabbaccabxaccabbacxrsabc)r"rr")r"r"rrtFrJrwrrr"r)rurvr) rrfr6rr;rmrrdrreplacerys r test_nuniquerVsP hd8n4>R SBtE{KLH ZZeZ , 4 4 6F&(+%[HNHNN}}S})H ZZ_ $ $ &F&(+ZZd $ , ,S 1 9 9 9 GF&(+sQwaS1W5T%[IHHNN ZZd $ , ,S 1 9 9 ;F&(+r c.tgdgdgdgdgdggd}|jdd gd j}tj|j |j g}td gd zd | }tj||y)N)rsr"Alice)rBob)i,rCharlie)iprDan)r.rEdith)amountrqrdrwrqrrdr"rr) rr6rr from_arraysrqrr r;r<)rGrMrrLs rtest_nunique_with_objectros        ) D\\4* +F 3 ; ; =F  " "DGGT[[#9 :EqcAgF%8H68,r ctdt}|jdj}tdd}t j ||y)Nrdrrlevelr)r rr6rr;r<)rGrMrLs rtest_nunique_with_empty_seriesrsC vV ,D \\\ " * * ,F61H68,r cttdtdtdggddjd}|jt j dd j }|jt j dd jtj }tj||y) Nz2016-06-28 09:35:35z2016-06-28 16:09:30z2016-06-28 16:46:28)rI23)rrGrr )r\rG) rr rr6rCGrouperrr>r r;r<testrMrLs rtest_nunique_with_timegrouperrs /0/0/0 $   i \\"**#. / 7 ? ? AF||BJJC01&9??OH68,r zkey, data, dropna, expected)rrrz 2019-01-01rr`rcri)rrryrr)rrrrrct||d}|jdgdj|}tj||y)N)r`rGr`rGr)rr6rr;r<)r`rGrrLrorMs rtest_nunique_with_NaTrsGV 3- .B ZZ  ( 0 0 0 ?F68,r ctgdtjdgd}|jgdj }tdgt j dg|j }tj||y) N)r"rrrtlevel_0rcrwrXrr)rrf) rrCrlr6rr0r}rfr;rmrs r)test_nunique_preserves_column_level_namesrs` Y#Y(G HD \\) $ , , .F!BHHaSM4<<HH&(+r cttdddg}|jgddjd}t gdd}t j ||y) Nz 2008-12-31z 2009-01-02rLrw)rrr"r)rrr"rc)rr r6r?r r;r<rs r$test_nunique_transform_with_datetimersQ :lL9F8 LB ZZ "6 * 4 4Y ?Fif-H68,r ctdgjd}|dd}|j||}|j}|rtg|ddd}ntdg|d}t j ||y)Nr"rirr0rr)r r=r6rr;r<)r1rrrrMrLs rtest_empty_categoricalrsw !+  Z (C bq'C S8 ,B ZZ\F"CG7;1#S868,r ctddtjdg}|jgd}|j t j }|jt j }tddtjgtjgd}tj||tj||y)NrYrZr[)rr"rrrr"rr) r r0rxr6r:rjr&r>r}r;r<)srHrMrvrLs rtest_intercept_builtin_sumrsS"&&#&'Aii %G[[ &FmmHLL)GsC(0CDH68,7H-r c Bd}d}tj}tj|||||||gd}t gd|d}|j dj |}|d k(rTt d tjgd d itjgd d}tj||yt d tjtjgdzd itjgd d}tj||y)NTFrr)r"r"rrrrr"rjrtrPrru)rrrrrWrcrr) r0rxrCr}rr6r&rlr;rmr)rQr_r^nadfgrorMrLs r!test_groupby_sum_mincount_booleanrs A A B ((Aq"b!Q*) .s.'!R.PV:Wr c&t|Sr$r)r r s rrz3test_regression_allowlist_methods..s.'!R.:Jr )rr6rir>rr;rm)r rhrframerHrMrLs`` r!test_regression_allowlist_methodsrs( qcNEmm!$m/G ,,%"%V4==q=)//0WX **,H fh/%"%'==q=)//0JK **,H fh/r c(ddgddgddgddgddgddgddgdd gdd gdd gdd gdd gddgddgg}t|ddgd}|jdgjj}tdgdgdd}t j ||y)Nr"r8)r#rr(rr+r/rXrtrur)rfrlN+R@rjr)rr6rrr;rm)rGrorMrLs r"test_groupby_prod_with_int64_dtyper!s B B B B A B B A A B A B B B D 4#s7 ;B ZZ  # # % 1 1 3Fs*<)=>gNH&(+r ctjddd}t|}|dddxxdzcc<|jdj }|t dz|d <|t dd z|d <tj |jd <|jttddz}|j}tjd}tjd}t|dz||||gtjd}t|||d}t!j"||y)Nz1 Dayi'ns)r[unitrrrtrruUTC)tzrvriz2887 days 11:21:02.326710176z2886 days 00:42:34.664668096rrs)rCtimedelta_ranger rrr rrr6rfrr@rr0r1rr;rm) tdirrorrMtd1td4exp_serrLs rtest_groupby_std_datetimeliker,s   We$ ?C +C!HMH c    !BIaL BsGIaE**BsG&&BGGBK DqNT) *B VVXF ,,5 6C ,,5 6CcAgsCc2"))A,GGwW7CDH&(+r c4ttdtdd}d}tjt|5|j dgj dgdddtjt|5|j dgj ddddtjt|5|j dgj d dddy#1swYxYw#1swY\xYw#1swYyxYw) Nrrjz(numeric_only accepts only Boolean valuesr+rtruTruerr")rrrArBrr6r9)rorPs r%test_mean_numeric_only_validates_boolr/s qa1 2B 4C z -& C5u%& z -4 C5F34 z -/ C5A.// &&44//s##C6#D #D6C?D D)prjdatetimerstringrnumpyr0rApandas._libs.tslibsrpandas.core.dtypes.commonrpandas.core.dtypes.missingrpandasrCrrr r r r pandas._testing_testingr;pandas.tests.groupbyr pandas.utilrtdmark parametrizerQrxrqrzrrrr}rrrrrrrrrrrrrrrrrrrr+r6r?rFrNrTrVrarcrgstrrmrrrrrrslowrrrrrrlrLrrrrrrrrrrr9rr@rr&rrrr"param skip_if_norrrr!r,r/rr rrBsV" $298."JK-+L-+`   $,%$," ,,  ruu u ruud ruue   -  -%   BHHdV4 ( BHHbeeWI .     BHHbeeWI . BHHdV4 (     BHHbeeWI . BHHbeeWI .     BHHbeeWI . BHHbeeWI .  ': ,;: ,"AB &C &% ' 266"D) & rvv&. &&C&8 45,6,(8 45!,6!,H aVA1a&IJ aVA1a&IJ$7,8,@  2 3 2 3  /  #KL,M ,&& 12,3,:P-! Q1aAq! ,iC Qq!Q1a +Y B Qq!Q1a +Wi@ Q1aAq! ,.?ARS BNN FF            ! 6-76-& Q1aAq! ,i8 Qq!Q1a +Y7 Qq!Q1a +W5 Q1aAq! ,.?@ --"-('v !Q2661aAr15y)Lv !RAq"%%Aq!4iKv !Q2q!QBEE15wHv "&&B 95 v "%%2y)4 v "%%2w0 v B1bffaAq!4iKv Aq"aAr1bee4iKv B1aQq!4gyIv 2 y)4v " i3v " gy1v Aq"aAr26615y)Lv B1aQq!4iKv Aq"aAr1bee4gyIv 2 y)4!v" " i3#v$ " gy1%v& B1aQ15y)L'v( Aq"aAr255!4iK)v* B1aQ1bee4gyI+v, 2 y)4-v. " i3/v0 " gy11v2 B1aQ15y)L3v4 Aq"aAr255!4iK5v6 B1aQ1bee4gwG7v:  2661aAq!Q /    9vF  BNN FF            # Evj 2 y)4kvl " i3mvn " gw/ovp B1aQ15y)Lqvr Aq"aAr255!4iKsvt B1aQ1bee4gwGuvx  2661aAq!Q /    wvD  BNN FF            # Cvh 2 y)4ivj " i3kvl " gw/mvn Ar1aB2661a8)YOovp Aq!RAq"beeQ7INqvr Ar1aB1a7)Lsvv  2661aAq!Q /    uvB  BNN FF            # Avf BFF8b=)Y7gvh BEE7R<I6ivj BEE7R<)4kvyt-uyt-&"$,",<-$?( ?%0 -1 -%0 ,1 ,,<%0 , ,1 ,"UV,/W,/^)"!@A,B,,%0,1,q!f-u 6X&7.X&v&(dE]35'E5>!:;+<4+D,2-&--"!  | $bffi .E F  A3hbhhu59 G    RWWT1a "&&'"''$1*= >  A3hbhhu59 G   &a#a#a#    Aq63*5!A O  &a#a#a#    Aq63*5!A O 3%(R-S(R-,- - .q"g.0/0, ,*0 P 4*!!,<=>? $qq)+;<=> aa(*:;<= aa(*:;<= *!!,<=7ST  % %6(   299       %& %& "'' UI]R]]7-CD ",#", -RUUAq"%%Aruua8 !'9:  %>?,@ ,(  0! 0.,0,6 /r