o
    i                     @   s   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dd Z	e dd Z
ejd	dejdd
fdgejdd
gfddgdejdd
gfddgejdd
dgfgd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d ZdS )    Ninnerouter)paramsc                 C   s>   | j }tg dg dt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df r   d/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/pandas/tests/groupby/test_index_as_string.pyframe   s   
r   c                  C   s<   t g dg dtdg dd} | g dd }|S )Nr   r   r   r   r   )r   r   r   r   )r   r   r   r   r   )r   sr   r   r   series   s   r    zkey_strs,grouperslevelr   c                 C   sx   d|vs	d| j v rtnd }d}tj||d | | }| | }W d    n1 s/w   Y  t|| d S )Nr   r   z!The default value of numeric_only)match)columnsFutureWarningtmassert_produces_warninggroupbymeanassert_frame_equal)r   key_strsgrouperswarnmsgresultexpectedr   r   r   "test_grouper_index_level_as_string(   s   
r1   r   )r   r   r   )r   r   r   c                 C   sR   t |trdd |D }ntj|d}| | }| | }t|| d S )Nc                 S   s   g | ]}t j|d qS )r!   )r   Grouper).0lvr   r   r   
<listcomp>M   s    z=test_grouper_index_level_as_string_series.<locals>.<listcomp>r!   )
isinstancelistr   r2   r(   r)   r&   assert_series_equal)r    r   r,   r0   r/   r   r   r   )test_grouper_index_level_as_string_series:   s   
r9   )numpyr   pytestpandasr   pandas._testing_testingr&   fixturer   r    markparametrizer2   r1   r9   r   r   r   r   <module>   s@    


		