Pi*<ddlZddlZddlZddlmZmZddlmZ ej dgdej dgdgdfgdgd fgdgd fgd gd fd ej d ej dgdej d ej d gfej dej dej gej dej dej gfej dddej ddddddfej ddddej dddddddfej gdzej gdzfg ej dgddZdZdZdZdZej ddd gej d!dgdd"ggej dd#d$ggd%Zd&Zd'Zd(Zej d)d ej d ej ged*d d gd+dgfd ej ddged*d dgd+d,gfd-d.d.ej ged*d-d.gdd/gfdgd0gdgd1gfggejgd23ejgd23fgd4Zej d5ejgd6d7zd83ejgd9d7zd:3gej dd#gd;gd<Zej dd#gd;gej d=d>d?gd@ZdAZdBZej dCgdDdEZej ddFdGgdHZ ej dIdJdKgdLZ!dMZ"dNZ#dOZ$dPZ%dS)QN) DataFrameIndex interpolation)linearlowerhighernearestmidpointz a_vals,b_vals))rrr r r )r r r r)rr r r )?@@@@)rrrrrrrrrrz1/1/18Dr)freqperiodssq)r???r ctjtj|tj|g}tj|||}tj|||}t dgt |zdgt |zz|d}t ||gdgt ddgd}|jjd kr5|j j d jd kr| |j}| d||} tj| |dS) N)rabkeyvalr#r"name)columnsindexMr)pdconcatSeriesquantilerlenrdtypekinddtypesvaluesastypegroupbytmassert_frame_equal) ra_valsb_valsrall_vals a_expected b_expecteddfexpectedresults ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/groupby/methods/test_quantile.py test_quantiler? sS<y")F++RYv->->?@@H6""++A]+KKJ6""++A]+KKJ C53v;;.#V1DDXVV W WB Z 5'SzPU9V9V9VH~c!!ho&22 ZZ   ' ' ' G GF&(+++++ctdgdi}tjgdtj}||dg}t jddgdgg}tdddgi| }tj ||tgd gd d }t jddgdd gg}tjgdtj}||dd g}tgdgdd | }tj ||dS)NArr r r r)rrr r r r.rrr @r')rr r r )rrrBBr)rrr r )rrg@g@)g@g@g@g@ rnparrayint64r3r,r) MultiIndex from_productr4r5)r;r"r=r'r<s r>test_quantile_arrayrQ=sb C) * *B (???"( 3 3 3C ZZ__ % %tf - -F M & &A'7 8 8E#d|,E:::H&(+++ LLL99 : :B M & &At '= > >E (<<test_quantile_array2rbRs )   " " + +Aqwbh + O OC 3U , , ,B ZZ__ % %sCj 1 1FCCCRRR  m(( __sCj )#t)  H&(+++++r@cftgdgdd}tjgdtj}||dgd}tgd gd dt jd d ggdg }tj ||||dddg}tgdgddt jd d gddgg }tj ||dS)Nrr r r rrrI)r rr rDF)sort)rrr)rr?rrr) @r@rrrr rrFrr)rgrrr)rirhrrrK)r;r"r=r<s r>test_quantile_array_no_sortrjcsU 33 4 4B (999BH - - -C ZZ%Z ( ( 1 12C2C2C D DF,,,3Q3Q3QRRm((1a&2C2C2C)DEEH&(+++ ZZ%Z ( ( 1 14, ? ?F""")=)=)=>>m((1a&4,)?@@H&(+++++r@c:tgdgdgdgdd}|ddgdd g}tjgd gd  }tgd gdd|}t j||dS)Nrdre)rrr)rrr )rBrJcdrlrmrr))rrr)rrr)rr r)rr r)rlrmNr[)rrrr)g @g@rrrIrF)rr3r,r)rO from_tuplesr4r5)r;r=r'r<s r>#test_quantile_array_multiple_levelsrous iiiiioooOOOTT  BZZc # # , ,dD\ : :F M % %PPP &  E$$$+A+A+ABB%H&(+++++r@ frame_size)r r )drSr3r rg333333?cb |\} ttj fdt|Dt }tjt |dgt zgz}fdtt |dDgt zttt t |dzgz}tj ||gd} fdtt |dD}fdt D} t||| } |  } tj | | dS) Nc"g|] }|dzgz S)r).0_ncols r> zJtest_groupby_quantile_with_arraylike_q_and_int_columns..s#AAATQUG^AAAr@rVrcg|] }D]}| Srtrt)ruxrvrs r>rxzJtest_groupby_quantile_with_arraylike_q_and_int_columns..s&<<<!<rxzJtest_groupby_quantile_with_arraylike_q_and_int_columns..sS/0QRLMq dS\\)*r@cg|]}|v| Srtrt)rurzr3s r>rxzJtest_groupby_quantile_with_arraylike_q_and_int_columns..s#CCCa!72B2B2B2B2Br@r'r&)rrLrMrangearangeminr-r`r)rOr3r,r4r5) rpr3rnrowr; idx_levels idx_codesexpected_indexexpected_valuesexpected_columnsr<r=rws `` @r>6test_groupby_quantile_with_arraylike_q_and_int_columnsrs JD$ 28AAAAU4[[AAABBERVKK X X XB)CaLL))*S\\9QC?J<<<test_quantile_raisesrs UCL5#, =u~ V V VB FC y , , ,%% 5""$$$%%%%%%%%%%%%%%%%%%s(A//A36A3ctgdtdd}|gd}tjt d5|ddddn #1swxYwYtjt d5|dddddS#1swxYwYdS) N)rrrr r r rGrr zGot '50.0' insteadr2zGot '-1.0' insteadr)rrr3rr ValueErrorr,)r;gs r>$test_quantile_out_of_bounds_q_raisesrsA +++%((;; < > > 2 z)= > > > 2s$A66A:=A:B??CCctjdtjdg}t|t dd}|d}t dD]}|dS)Nrr r!r"rq)rLrMnanrrr3r,)datar;grprvs r>/test_quantile_missing_group_values_no_segfaultsrsv 8S"&#& ' 'D 4a11 2 2B **U  C 3ZZ r@z$key, val, expected_key, expected_valrrErr rg*gE@float64rDc>t||d}t|t|ddg}|d}|d}t j|||}t j||dS)Nr!r"r$r#rr)rrr3r,r4r5)r"r# expected_key expected_valr;r<rr=s r>2test_quantile_missing_group_values_correct_resultsrs 3s++ , ,BE,U;;;eWH **U  C \\#  F&(+++ \\^^F&(+++++r@r1)r rNr Int64)TFNbooleanrrrctdgdzdgdzz|d}|dd|}t|tr,t jddg|fddg}gd}ntddgd }d g}t j |d z|dd }tj ||dS)Nrzr yrrr r[rr$rr Float64r'r%r.) rr3r, isinstancer`r)rOrPrr+r4assert_series_equal)r1rr;r=idxtrue_quantilesr<s r>$test_groupby_quantile_nullable_arrayrs #cUQY.V<< = =B ZZ__S ! * *1 - -F!Tm((3*ad (LL(S#JS)))y!+3S RRRH68,,,,,r@ numeric_onlyTFctdgdgdgd}|rp|d||}|ddg|}tj||dSd}t jt| 5|d||ddddS#1swxYwYdS) Nr rrz)rr rlr)rr z0dtype '.*' does not support operation 'quantile'r)rr3r,r4r5rrr)rrr;r=r<rs r>-test_groupby_quantile_raises_on_invalid_dtypers> !C5u55 6 6BCC))!,)GG::c??C5)22155 fh/////@ ]9C 0 0 0 C C JJsOO $ $Q\ $ B B B C C C C C C C C C C C C C C C C C Cs&+CC"%C"ctjg|j}t|tjr tjn tj}tddgd|gd|}|dd d}tdg|d }|d vr|}nd}tjdg||d }tj |||dd dd g}tjdgd ztj |dd gfddgd|}tj ||dS)NrDr 皙?rzrrzrrr)r.r%)Float32rr.r'r%rr r[r)r)r+r.rrLrNArr3r,rr4rrOrP)any_float_dtyper.itemr;r= exp_indexexpected_dtyper<s r>test_groupby_quantile_NA_floatrsa Ib 0 0 0 6Erx00 ;266beD !QsDk22/ J J JB ZZ__S ! * *3 / /Fse?===I000(y#nICPPPH68,,, ZZ__S ! * *C; 7 7Fy  m(()c4[)A#t(UU  H 68,,,,,r@c tddgddgd|}|ddd}tjd gd t dgd| d }t j|||dd}tdd id t dgd|  }t j||dS)Nr r rrrDrzrrrhr)r%r.rr.r') rr3r,r)r+rr4rr5)any_int_ea_dtyper;r=r<s r>test_groupby_quantile_NA_intrs !Qq!f--5E F F FB ZZ__S ! * *3 / /Fy QCc)9:::  H 8V,,, ZZ__ % %c * *F c )5!3FV+W+W+WH&(+++++r@zinterpolation, val1, val2))rr r )rr r )r r r c tgddddtjgd|}|ddd g| }td ||tjtjgi|tjtjgd |gd gddg}tj ||dS)Nr r r r r r r rrDrrrXrrr r r r r )rrXrrXr[r rr)rr3r,rO from_arraysr+r4r5)rval1val2any_numeric_ea_dtyper;r=r<s r>)test_groupby_quantile_all_na_group_maskedr-s ll!Q25!122:N   BZZ__ % %c - % P PF tT25"%()"m'' Y|||+? @ @ @BVBVBV W+(  H&(+++++r@r rc tgddddtjgd|}|ddd g| }|d kr|}nd }td ddtjtjgi|tjtjgd|gdgddg}tj ||dS)Nrr r r rrDrrrrrrr rrEr)rrrrr[rr)rrr;r=rr<s r>0test_groupby_quantile_all_na_group_masked_interprCs ll!Q25!122:N   BZZ__ % %d = % Q QFy((-" sC&'m'' ,,,.BCCC&&& + (     H&(+++++r@r.rrcRtddgtjgdzd|}|ddd}tjtjg|t dg|d }d|j_tj ||dS) Nr r rrDrzrrrr) rr)rr3r,r+rr'r%r4r)r.r;r=r<s r>"test_groupby_quantile_allNA_columnr`s !Qrugk22% @ @ @B ZZ__S ! * *3 / /Fy uE3%u$=$=$=CHHN8V,,,,,r@ctjtjddd}t |gdd}|dd}t d tjd tjd gitd d gd}tj ||dS)Nrr)unitusr)valuegrouprgGz?rz0 days 00:00:00.990000z0 days 00:00:02.990000r r r$rF) r) to_timedeltarLras_unitrr3r, Timedeltarr4r5)tdir;r=r<s r>test_groupby_timedelta_quantilerls /")A,,S 1 1 1 9 9$ ? ?C S<<<88 9 9B ZZ ) )$ / /F  566 566 QF)))H&(+++++r@c  tjdddd|d}t|t t ddt t dd d }|d d d g}tj |dddd d ggd}tddddddddddddg|}tj ||dS)Nz2020-04-19 00:00:001minrqUTC)startrrtzr1hr e) timestampcategoryrrr皙?c)rNr[g)@g333333\@)rrgH@gfffffb@g333333Q@ge@g W@gfffffh@rF) r) date_rangefloorrr`rr3r,rOrPr4r5)rdtir;r=mir<s r>test_timestamp_groupby_quantiler}s- -#&#%d    eDkk U1c]]++%S//**    BZZ $ $ - -sCj 9 9F  # #S2Yc $;CV # W WB . . . . . . . .  H&(+++++r@ctjddd}tj||dd<|dd<||d z d <tjjd <tj tj d d } |}| d}fdtd D}t|j}|jt(|_t+j||dS)Nz 2016-01-01ins)rrz US/Pacificr rr rr rrrcXi|]&}|j|ddd'S)Nrr)ilocr,)ruir;s r> z7test_groupby_quantile_dt64tz_period..s5 < < #test_groupby_quantile_dt64tz_periodrs* - d > > >C 3 " " ' ' ) )B OOL ) )BqE MM#  BqE #a&LBqE&BGBK 1s # #B BB [[  F = < < <588 < < +test_groupby_quantile_nonmulti_levels_orderrs - # # )(( $  C )E!HHC ( ( (C [[vE[ 2 2 ; ;S#J G GF = $ $888 %  Dy---T:::H68,,,go,77#sc3Z8PQQOOFL'99999r@)&numpyrLrpandasr)rrpandas._testing_testingr4mark parametrizerrrr?rQrbrjrorrrrrrMrrrrrrrrrrrrrtr@r>rs  III ///* |||$ ,,,' " " "$=$=$=> rvsBFC (3RVS*IJ &#rvsBF +bfc263-OP BM(a 8 8 8 BM(a 8 8 82 > BM(a 8 8 8 @ @ E E BM(a 8 8 82 > F Fs K K &Ax!|$)255566,,763:,(,,,*,,,",,,$ , , , ':;;aS1a&M22Szl++,,,+32<;,0%%%* rvsBF #UU1XXSzC:F rvsC %%((S#Jc C sC %%((S#JCA rdQC$ R"I...90M0M0MN   ,,  ,& !111$$$q( :::sOOO455 - -65 - sOOO455$77 C C8765 C---4,,,&!W!W!W,,,&:x*@AA,,BA,89i"899--:9-,,,",,,:,,,,:::::r@