U
|
ý°d7 ã @ sü d dl Zd dlZd dlZd dlmZ ejdgddggddd Z e ¡ dd Z
|
ej d dej dd
|
fdgej dd
|
gfd dgd ej dd
|
gfdd gej dd
|
d gfg¡d d Zej dddd dgdgd gddgddgddd gd ddgg
|
¡dd ZdS )é NÚinnerÚouter)Úparamsc C sV | j }t ddddddgddddddgt d¡ddddddgd ¡}|rR| |¡}|S )
|
NÚaÚbé é é é ÚoneÚtwo©r r ÚAÚB)ÚparamÚpdÚ DataFrameÚnpÚarangeÚ set_index)ÚrequestÚlevelsÚdf© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_index_as_string.pyÚframe s üÿ
|
r c C sV t ddddddgddddddgt d¡ddddddgd ¡} | d
|
d d g¡d }|S )Nr r r r r r
|
r r r r r r r )r r r r r )r Úsr r r Úseries s üÿr zkey_strs,groupers©Úlevelr c C sd d|ksd| j kr8| |¡jdd}| |¡jdd}n | |¡ ¡ }| |¡ ¡ }t ||¡ d S )Nr r T)Z numeric_only)ÚcolumnsÚgroupbyÚmeanÚtmZassert_frame_equal)r Zkey_strsÚgroupersÚresultÚexpectedr r r Ú"test_grouper_index_level_as_string( s
|
r' r c C sR t |trdd |D }n tj|d}| |¡ ¡ }| |¡ ¡ }t ||¡ d S )Nc S s g | ]}t j|d qS )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# Zassert_series_equal)r r r$ r& r% r r r Ú)test_grouper_index_level_as_string_series; s
|
r. )Únumpyr ZpytestZpandasr Zpandas._testingZ_testingr# Zfixturer r ÚmarkZ parametrizer( r' r. r r r r Ú<module> s>
|
|
üþ
|
|
öþ
|
