
    Owg                        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                  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                  j                  dddd	dgdgd	gddgddgg dg dg
      d        Zy)    Ninnerouter)paramsc                     | j                   }t        j                  g dg dt        j                  d      g dd      }|r|j                  |      }|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/horilla/myenv/lib/python3.12/site-packages/pandas/tests/groupby/test_index_as_string.pyframer      sJ    ]]F	3'1;		

B \\&!I    c                      t        j                  g dg dt        j                  d      g dd      } | j	                  g d      d   }|S )Nr   r
   r   r   r   )r   r   r   r   )r   r   r   r   r   )r   ss     r   seriesr#      sD    	3'1;		

B 	,-c2AHr    zkey_strs,grouperslevelr   c                 V   d|vsd| j                   v rC| j                  |      j                  d      }| j                  |      j                  d      }n>| j                  |      j                         }| j                  |      j                         }t        j                  ||       y )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x(--4-@==*//T/Bx(--/==*//1&(+r    r   )r   r   r   )r   r   r   c                 J   t        |t              r$|D cg c]  }t        j                  |       }}nt        j                  |      }| j	                  |      j                         }| j	                  |      j                         }t        j                  ||       y c c}w )Nr$   )
isinstancelistr   Grouperr)   r*   r+   assert_series_equal)r#   r   lvr.   r0   r/   s         r   )test_grouper_index_level_as_string_seriesr8   ;   s}    " &$39:RBJJR(::::F+~~h',,.H ^^F#((*F68, ;s   B )numpyr   pytestpandasr   pandas._testing_testingr+   fixturer   r#   markparametrizer5   r1   r8    r    r   <module>rB      s@       	GW#567 8    	*"**7+,
ZRZZg./0
w#zrzz89:
3*"**73S9:	,, 						'	'--r    