o
    �¨Êhâ  ã                   @   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   ú[/var/www/html/env/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   sd   d|vs	d| j v r|  |¡jdd�}|  |¡jdd�}n|  |¡ ¡ }|  |¡ ¡ }t ||¡ d S )Nr   r   T)Únumeric_only)ÚcolumnsÚgroupbyÚmeanÚtmÚassert_frame_equal)r   Úkey_strsÚgroupersÚresultÚexpectedr   r   r   Ú"test_grouper_index_level_as_string(   s   
r-   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   r.   r%   r&   r'   Úassert_series_equal)r    r   r*   r,   r+   r   r   r   Ú)test_grouper_index_level_as_string_series;   s   
r5   )Únumpyr   ÚpytestÚpandasr   Úpandas._testingÚ_testingr'   Úfixturer   r    ÚmarkÚparametrizer.   r-   r5   r   r   r   r   Ú<module>   s@    

üþ
	
öþ