gL iddlZddlZddlZddlmZmZmZddlm Z ejjdddgdZ ejjdddgdZdZejjd dgd fdgd fgd Zd ZejjdddgejjdddgejjdddgdZejjddej&fdgejjdgddZdZy)N) DataFrameIndex date_rangefuncffillbfillc2tgddtjtjggdgtgdd}|j dgd d g}t ||j }td d gd}tj||y) N)?gr )r g@g)typeabidxnamecolumnsr r r) rnpnanrgroupbygetattrrtmassert_index_equal)rdf df_groupedresultexpecteds g/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/groupby/test_missing.py.test_groupby_column_index_name_lost_fill_funcsr s  !RVVRVV,n=(u5 BVH%sCj1J &WZ & ( 0 0Fc3Ze,H&(+cRtgdgdd}tddtjdgi}tj||gdj dg}tdd gd tjgdd ggddg }t ||}tj||y) N)r )field1field2r$r r#)axisr%)byr r"g@r) rrrpdconcatrrrassert_frame_equal)rdf1df2rrrs r(test_groupby_fill_duplicate_column_namesr-s yI> ?C X2661~. /CC:A.668*6EJ SArvv;C)Hh3GH'WZ & (F&(+r c*tddgddgd}d}tjt|5t j t d5|jdjddddddy#1swYxYw#1swYyxYw)Nr r r%DataFrameGroupBy.fillna is deprecatedmatchzMust specify a fillr) rrassert_produces_warning FutureWarningpytestraises ValueErrorrfillna)rmsgs rtest_ffill_missing_argumentsr;'s !Qq!f- .B 1C # #M =% ]]:-B C % JJsO " " $ %%% % %%%s#B  A=,B =B B  Bzmethod, expected)Nr r )r r Nclttjgddgdd}|jd}d}t j t |5|j| }dddtd tj|di}t j|y#1swYBxYw) N)Nr Nstring)dtype)rrrr0rr1r2methodr ) rr(arrayrrr4r5r9r*)r@rrgrpr:rs rtest_fillna_with_string_dtyperC0s "38D9U VB **S/C 1C # #M =+6*+#rxxABCH&(+++s B**B3c :ttjjddgt ddgt ddgd }t jd t jt jd t jt jd t jt jd t jg |d<t jdt jt jdt jt jdt jt jdt jg |d<d}tjt|5|jddjd}dddd}tjt|5|jjdd jdj}dddtjy#1swY}xYw#1swY,xYw)Nvalue1value2z 2014-01-01z 2014-01-0612idr)indexrr r/r"!r#,z5The 'axis' keyword in DataFrame.groupby is deprecatedr2r)levelr&rr?z+DataFrame.groupby with axis=1 is deprecated)rr( MultiIndex from_productrrrrrr4r5rr9Tr*)rr:rrs rtest_fill_consistencyrS>s mm(( !:lL#I J sCjt,  B             BsG             BsG BC # #M =F::AA:.55W5EF 8C # #M =HAA.55W5EGGH&(+ FFHHs%$H.8HHHr@dropnaTF has_nan_groupc 6tddg}gd}|j|jd}|rtjnd}t j dgdz|gdzz|d <|jd | }t||d }gd gd gd gd gdgdgdgdd} | j|||f}|j|jd} | jjt| _ tj|| y)N)r g?)r/g?)rrWrWr rWT)droprr r" group_col)r'rT)limit)rWrrrWrWrW)rWrrrWr r )rrrWrWrWrW)rrrWr r rW))rTT)rTF)rFT)rFF)rTT)rTF)rFT)rFF)rreindex reset_indexrrr(Seriesrrgetrastypeobjectrr*) rTr@rUdf_without_nan_rowsridxrgroup_bgroupedr expected_rowsrs rtest_ffill_handles_nan_groupsrfos  $Xx$89 !D  $ $T * 6 6D 6 AB%bff3Gii WIM 9:B{OjjKj7G %WWf %D 1F 6 4 4!55 4 4!5 M   ffm< =D"**40<<$<GH''..v6H&(+r zmin_count, valuer/)rWr )firstlastmaxmincTtdgdzdtjtjgtjgdzd}t|j d||}t|gtjgdt dgd}t j||y) Nr r"r rcr ) min_count)rrmr)rJ)rrrrrrrr*)rrnvaluerrrs rtest_min_countrps !q266266':"&&AN OB +WRZZ_d +i @FwbffX6eQCc>RSH&(+r ctddtjggdgdd}|jddg}|j}tj dgtj dgd}||k(sJy) Nr )r/r"r#)rlr rr))r r/)r r")rrrrindicesrA)rgrrs rtest_indices_with_missingrwsd !QiiH IB C:A YYF((A3-288QC=AH X  r )numpyrr6pandasr(rrrpandas._testing_testingrmark parametrizerr-r;rCrSrfrrprwr rrsn  '7!34 ,5 ,'7!34 ,5 ,%'#34w@P6QR,,.,bGW#56D%=14-8,927,B+q"&&k9-EF!@A,BG,r