a
    Pªfý  ã                   @   s2   d dl Z d dlmZ d dlmZ G dd„ dƒZdS )é    N)Ú	DataFramec                   @   s   e Zd Zdd„ ZdS )ÚTestSwaplevelc                 C   sL  |}|d   ¡ }|d   d¡}|d   dd¡}|d   dd¡}|j |j¡rPJ ‚t ||¡ t ||¡ t ||¡ |  ¡ }|  d¡}|  dd¡}	|  dd¡}
|j |j¡s°J ‚t ||¡ t ||	¡ t ||
¡ |j}|j dddd}|  dd¡j}t ||¡ d}tjt	|d  t
td	ƒƒ  ¡  W d   ƒ n1 s>0    Y  d S )
NÚAr   é   ÚfirstÚsecond)Zaxisz,Can only swap levels on a hierarchical axis.)Úmatché   )Z	swaplevelÚindexÚequalsÚtmZassert_series_equalÚTZassert_frame_equalÚpytestZraisesÚ	TypeErrorr   Úrange)ÚselfZ multiindex_dataframe_random_dataÚframeZswappedZswapped2Zswapped3Zswapped4ÚbackZback2Zback3Zback4ÚftÚexpÚmsg© r   úb/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/methods/test_swaplevel.pyÚtest_swaplevel   s0    
zTestSwaplevel.test_swaplevelN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r      s   r   )r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   