
    Eg                        d dl Zd dlZd dlZd dlmZ  ej        dgddgg          d             Z	 ej                    d             Z
ej                            dd ej        d          fdg ej        d          gfd	dgd	 ej        d          gfdd	g ej        d          d	gfg          d
             Zej                            dddd	dgdgd	gddgddgg dg dg
          d             ZdS )    Ninnerouter)paramsc                     | j         }t          j        g dg dt          j        d          g dd          }|r|                    |          }|S )Nar   r   br	   r	            r   r   r      oner   twor   r   r   r   r   AB)parampd	DataFramenparange	set_index)requestlevelsdfs      ^/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/tests/groupby/test_index_as_string.pyframer      sj    ]F	333'''1;;;		
 	

 
B  "\\&!!I    c                      t          j        g dg dt          j        d          g dd          } |                     g d          d         }|S )Nr   r
   r   r   r   )r   r   r   r   )r   r   r   r   r   )r   ss     r   seriesr#      sg    	333'''1;;;		
 	

 
B 	,,,--c2AHr    zkey_strs,grouperslevelr   c                    d|vs	d| j         v rS|                     |                              d          }|                     |                              d          }nN|                     |                                          }|                     |                                          }t          j        ||           d S )Nr   r   T)numeric_only)columnsgroupbymeantmassert_frame_equal)r   key_strsgroupersresultexpecteds        r   "test_grouper_index_level_as_stringr1   (   s     (g66x((--4-@@==**//T/BBx((--//==**//11&(+++++r    r   )r   r   r   )r   r   r   c                 :   t          |t                    rd |D             }nt          j        |          }|                     |                                          }|                     |                                          }t          j        ||           d S )Nc                 8    g | ]}t          j        |           S )r$   )r   Grouper).0lvs     r   
<listcomp>z=test_grouper_index_level_as_string_series.<locals>.<listcomp>M   s%    :::RBJR(((:::r    r$   )
isinstancelistr   r4   r)   r*   r+   assert_series_equal)r#   r   r.   r0   r/   s        r   )test_grouper_index_level_as_string_seriesr;   ;   s    " &$ ,::6::::F+++~~h'',,..H ^^F##((**F68,,,,,r    )numpyr   pytestpandasr   pandas._testing_testingr+   fixturer   r#   markparametrizer4   r1   r;    r    r   <module>rE      s                  	GW#56777  87     	*"*7+++,
ZRZg.../0
w#zrz8889:
3*"*7333S9:	 , , , 						'	' - - - - -r    