j*<FddlZddlZddlZddlmZmZddlmZ ejjdgdejjdgdgdfgdgd fgdgd fgd gd fd ejd ejdgdejd ejd gfejdejdejgejdejdejgfejdddejddddddfejdddjdejddddddjdfejgdzejgdzfg ejjdgddZdZdZdZdZejjddd gejjd!dgdd"ggejjdd#d$ggd%Zd&Zd'Zd(Zejjd)d ejd ejged*d d gd+dgfd ejddged*d dgd+d,gfd-d.d.ejged*d-d.gdd/gfdgd0gdgd1gfggej2gd23ej2gd23fgd4Zejjd5ej2gd6d7zd83ej2gd9d7zd:3gejjdd#gd;gd<Zejjdd#gd;gejjd=d>d?gd@ZdAZdBZejjdCgdDdEZejjddFdGgdHZ ejjdIdJdKgdLZ!dMZ"dNZ#dOZ$dPZ%y)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|j||}tj|j||}t dgt |zdgt |zz|d}t ||gdgt ddgd}|jjd k(rA|jjd jd k(r|j|j}|jdj||} tj| |y) N)rabkeyvalr#r"name)columnsindexMr)pdconcatSeriesquantilerlenrdtypekinddtypesvaluesastypegroupbytmassert_frame_equal) ra_valsb_valsrall_vals a_expected b_expecteddfexpectedresults k/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_quantile.py test_quantiler? s%<yy"))F+RYYv->?@H6"++A]+KJ6"++A]+KJ C53v;.#V1DDXV WB Z 5'SzPU9VH~~c!hoo&<&>2 ZZ  ' ' ' GF&(+ctdgdi}tjgdtj}|j |j dg}t jjddgdgg}tdddgi| }tj||tgd gd d }t jjddgdd gg}tjgdtj}|j |j dd g}tgdgdd | }tj||y)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=s  C) *B ((?"(( 3C ZZ_ % %tf -F MM & &A'7 8E#d|,E:H&(+ L9 :B MM & &At '= >E ((test_quantile_array2rbRs ))   " + +Aqwbhh + OC 3U ,B ZZ_ % %sCj 1FCR mm(( sCj )#t) H&(+r@c:tgdgdd}tjgdtj}|j |dj gd}tgd gd dt jjd d ggdg }tj|||j |dj ddg}tgdgddt jjd d gddgg }tj||y)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_sortrjcs 3 4B ((9BHH -C ZZ%Z ( 1 12C DF,3QRmm((1a&2C)DEH&(+ ZZ%Z ( 1 14, ?F")=>mm((1a&4,)?@H&(+r@ctgdgdgdgdd}|jddgjdd g}tjj gd gd  }tgd gdd|}t j||y)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 ioOT BZZc # , ,dD\ :F MM % %P & E$+AB%H&(+r@ frame_size)r r )drSr3r rg333333?c t|\}}ttjt|Dcgc] }||dzgz c}t|}tjt |dgt |z|gz}tt |dDcgc] }|D]}| c}}gt |zttt |t |dzgz} tj|| g|d} tt |dDcgc]$}|D]}t|g|t |z z&} }}t|Dcgc] }||vs| } }t| | | } |j|j|}tj|| ycc}wcc}}wcc}}wcc}w)NrrV)levelscodesr\r'r&)rrLrMrangearangeminr-r`r)rOfloatr3r,r4r5)rpr3rnrowncol_r; idx_levelsx idx_codesexpected_indexexpected_valuesexpected_columnsr<r=s r>6test_groupby_quantile_with_arraylike_q_and_int_columnsrs JD$ 288U4[ATQUG^ABERVK XB))CaL)*S\9QC?J"3tQ<0<!test_quantile_raisesrsb UCL5#, =u~ VB FC y ,% 5""$%%%s AA(c`tgdtdd}|jgd}tjt d5|j ddddtjt d5|j ddddy#1swY?xYw#1swYyxYw) N)rrrr r r rGrr zGot '50.0' insteadr2zGot '-1.0' insteadr)rrvr3rr ValueErrorr,)r;gs r>$test_quantile_out_of_bounds_q_raisesrs +%(;  2 z)= > 2sB=B$B!$B-ctjdtjdg}t|t dd}|j d}t dD]}|j y)Nrr r!r"rq)rLrMnanrrvr3r,)datar;grpr|s r>/test_quantile_missing_group_values_no_segfaultsrsZ 88S"&&#& 'D 4a1 2B **U C 3Z r@z$key, val, expected_key, expected_valrrErr rg*gE@float64rDct||d}t|t|ddg}|jd}|jd}t j |||j}t j ||y)Nr!r"r$r#rur)rrr3r,r4r5)r"r# expected_key expected_valr;r<rr=s r>2test_quantile_missing_group_values_correct_resultsrsv 3s+ ,BE,U;eWH **U C \\# F&(+ \\^F&(+r@r1)r rNr Int64)TFNbooleanrrrc|tdgdzdgdzz|d}|jddj|}t|tr,t j jddg|fddg}gd}ntddgd }d g}t j|d z|dd }tj||y)Nr~r 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!Tmm((3*ad (L(S#JS)yy!+3S RH68,r@ numeric_onlyTFctdgdgdgd}|r]|jdj||}|jddgj|}tj||yd}t j t| 5|jdj||dddy#1swYyxYw) Nr rr~)rr rlr)rr z0dtype '.*' does not support operation 'quantile'r)rr3r,r4r5rrr)rrr;r=r<rs r>-test_groupby_quantile_raises_on_invalid_dtypers !C5u5 6BC))!,)G::c?C5)2215 fh/@ ]]9C 0 C JJsO $ $Q\ $ B C C Cs #B;;Cctjg|j}t|tjrtj ntj }tddgd|gd|}|jddjd}tdg|d }|d vr|}nd}tjdg||d }tj|||jddjdd g}tjdgd ztjj|dd gfddgd|}tj||y)NrDr 皙?r~rr~rrr)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_floatrs' IIb 0 6 6Erxx0266beeD !QsDk2/ JB ZZ_S ! * *3 /Fse?=I00(yy#nICPH68, ZZ_S ! * *C; 7Fyy  mm(()c4[)A#t(U  H 68,r@c tddgddgd|}|jddjd}tjd gd t dgd| d }t j|||jdjd}tdd id t dgd|  }t j||y)Nr r rrrDr~rrrhr)r%r.rr.r') rr3r,r)r+rr4rr5)any_int_ea_dtyper;r=r<s r>test_groupby_quantile_NA_intrs !Qq!f-5E FB ZZ_S ! * *3 /Fyy QCc)9:  H 8V, ZZ_ % %c *F c )5!3FV+WH&(+r@zinterpolation, val1, val2))rr r )rr r )r r r c tgddddtjgd|}|jdj dd g| }td ||tjtjgi|tj j tjgd |gd gddg}tj||y)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 !Q255!12:N BZZ_ % %c - % PF tT255"%%()"mm'' YY|+? @BV W+( H&(+r@r rc tgddddtjgd|}|jdj dd g| }|d k(r|}nd }td ddtjtjgi|tj j tjgd|gdgddg}tj||y)Nrr r r rrDrrrrrrr rrEr)rrrrr[rr)rrr;r=rr<s r>0test_groupby_quantile_all_na_group_masked_interprCs !Q255!12:N BZZ_ % %d = % QFy(-" sC&'mm'' ,.BC& + (  H&(+r@r.rrcHtddgtjgdzd|}|jddj d}tj tjg|t dg|d }d|j_tj||y) Nr r rrDr~rrrr) rr)rr3r,r+rr'r%r4r)r.r;r=r<s r>"test_groupby_quantile_allNA_columnr`s !Qruugk2% @B ZZ_S ! * *3 /Fyy uE3%u$=CHHNN8V,r@ctjtjddj d}t |gdd}|j djd}t d tjd tjd gitd d gd}tj||y)Nrr)unitusr)valuegrouprgGz?rz0 days 00:00:00.990000z0 days 00:00:02.990000r r r$rF) r) to_timedeltarLrwas_unitrr3r, Timedeltarr4r5)tdir;r=r<s r>test_groupby_timedelta_quantilerls //"))A,S 1 9 9$ ?C S<8 9B ZZ ) )$ /F  56 56 QF)H&(+r@c tjdddd|jd}t|t t ddt t dd d }|j d jd d g}tjj|dddd d ggd}tddddddddddddg|}tj||y)Nz2020-04-19 00:00:001minrqUTC)startrrtzr1hr e) timestampcategoryrrr皙?c)rNr[g)@g333333\@)rrgH@gfffffb@g333333Q@ge@g W@gfffffh@rF) r) date_rangefloorrr`rvr3r,rOrPr4r5)rdtir;r=mir<s r>test_timestamp_groupby_quantiler}s --#&#%d  eDk U1c]+%S/*  BZZ $ - -sCj 9F  # #S2Yc $;CV # WB . . . .  H&(+r@ctjddd}tj|jj }|j d|d<|j d|d<||d z |d <tj|jd <tjtjd d }|j|}|jd}td Dcic]$}||j|dd jd&}}t|j j#}|j$j't(|_t+j,||ycc}w)Nz 2016-01-01ins)rrz US/Pacificr rr rr rrr)r)rr+to_framecopy tz_localize to_periodNaTilocrLtilerwr3r,rvrT infer_objectsr'r2intr4r5)rr;bygbr=iexpr<s r>#test_groupby_quantile_dt64tz_periodrs& -- d >C 3 " ' ' )B OOL )BqE MM# BqE #a&LBqE&&BGGBK 1s #B BB [[ F498 +test_groupby_quantile_nonmulti_levels_orderrs -- # # ) $ C ))E!HC (C [[vE[ 2 ; ;S#J GF == $ $8 % Dyy-T:H68,ggoo,,77#sc3Z8PQOOOFLL''9r@)&numpyrLrpandasr)rrpandas._testing_testingr4mark parametrizerrrr?rQrbrjrorrrrrvrMrrrrrrrrrrrrr@r>rsc I /* |$ ,' "$=> rvvsBFFC (3RVVS*IJ &&#rvvsBFF +bffc2663-OP BMM(a 8 BMM(a 82 > BMM(a 8 @ @ E BMM(a 82 > F Fs K &&Ax!|$)256,73:,(,*,",$ , ':;aS1a&M2Szl+,,3<,0%* rvvsBFF #U1XSzC:F rvvsC %(S#Jc C sC %(S#JCA rdQC$ R"I.90MN  , ,& !1$q( :sO45 -6 - sO45$7 C86 C-4,&!W,,&:x*@A,B,89i"89-:-,",:,,:r@