
    A>i                        S SK rS SKrS SKrS SKJrJr  S SKJr	  \R                  R                  SSS/5      S 5       r\R                  R                  SSS/5      S 5       r\R                  R                  SSS/5      \R                  R                  S	S
S/5      S 5       5       r\R                  R                  SS\R                  4S/5      \R                  R                  S/ SQ5      S 5       5       rS rg)    N)	DataFrameIndexfuncffillbfillc                    [        / SQS[        R                  [        R                  // SQ/[        / SQSS9S9nUR	                  S/5      S	S
/   n[        X 5      " 5       R                  n[        S	S
/SS9n[        R                  " X45        g )N)         ?g      r	   )r	   g       @g       )typeabidxnamecolumnsr   r   r   )	r   npnanr   groupbygetattrr   tmassert_index_equal)r   df
df_groupedresultexpecteds        `/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/tests/groupby/test_missing.py.test_groupby_column_index_name_lost_fill_funcsr      s~     
	!RVVRVV,n=(u5
B VH%sCj1JZ&(00Fc3Ze,H&+    c                 >   [        / SQ/ SQS.5      n[        SS[        R                  S/05      n[        R                  " X/SS9R                  S/S9n[        SS	/S
[        R                  /SS//SS/S9n[        X05      " 5       n[        R                  " XT5        g )N)r	         )field1field2r#   r	   r"   )axisr$   )byr
   r!   g      @r   )	r   r   r   pdconcatr   r   r   assert_frame_equal)r   df1df2r   r   r   s         r   (test_groupby_fill_duplicate_column_namesr,      s     yI>
?C
X2661~.
/CC:A.668*6EJ
SArvv;C)Hh3GH Z&(F&+r   methodhas_nan_groupTFc           	      &   [        SS/5      n/ SQnUR                  U5      R                  SS9nU(       a  [        R                  OSn[
        R                  " S/S-  U/S-  -   5      US	'   UR                  S	U S
9n[        Xq5      " S S9n/ SQ/ SQ/ SQ/ SQ/ SQ/ SQ/ SQ/ SQS.n	U	R                  XU45      nUR                  U5      R                  SS9n
U
R                  R                  [        5      U
l
        [        R                  " X5        g )N)r	   g?)   g?)r   r1   r1   r	   r1   T)dropr   r   r!   	group_col)r&   dropna)limit)r1   r   r   r1   r1   r1   )r1   r   r   r1   r	   r	   )r   r   r1   r1   r1   r1   )r   r   r1   r	   r	   r1   ))r   TT)r   TF)r   FT)r   FF)r   TT)r   TF)r   FT)r   FF)r   reindexreset_indexr   r   r'   Seriesr   r   getr   astypeobjectr   r)   )r4   r-   r.   df_without_nan_rowsridxr   group_bgroupedr   expected_rowsr   s              r   test_ffill_handles_nan_groupsrA   &   s   
 $Xx$89!D		$	$T	*	6	6D	6	AB%bff3Gii	WIM 9:B{OjjKj7GW%D1F  6 4 4!55 4 4!5	M fm<=D"**40<<$<GH''..v6H&+r   zmin_count, valuer0   )r1   r
   )firstlastmaxminc                 B   [        S/S-  S[        R                  [        R                  /[        R                  /S-  S.5      n[        UR	                  S5      U 5      " US9n[        U/[        R                  /S.[        S/SS9S9n[        R                  " XE5        g )	Nr	   r!   r   r   cr   )	min_count)r   rH   r   )index)r   r   r   r   r   r   r   r)   )r   rI   valuer   r   r   s         r   test_min_countrL   I   s}     
!q266266':"&&AN	OBRZZ_d+i@FwbffX6eQCc>RSH&+r   c                      [        SS[        R                  // SQ/ SQS.5      n U R                  SS/5      nUR                  n[        R
                  " S/5      [        R
                  " S/5      S.nX#:X  d   eg )	Nr	   )r0   r!   r"   )         rG   r   r   r   ))r
   r0   )r
   r!   )r   r   r   r   indicesarray)r   gr   r   s       r   test_indices_with_missingrT   S   sb    	!QiiH	IB


C:AYYF((A3-288QC=AHr   )numpyr   pytestpandasr'   r   r   pandas._testing_testingr   markparametrizer   r,   rA   r   rL   rT    r   r   <module>r]      s       '7!34	, 5	, '7!34	, 5	, GW#564-8, 9 7,B +q"&&k9-EF!@A, B G,r   