U
    sVÞcý  ã                   @   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   s6  |}|d   ¡ }|d   d¡}|d   dd¡}|d   dd¡}|j |j¡rPt‚t ||¡ t ||¡ t ||¡ |  ¡ }|  d¡}|  dd¡}	|  dd¡}
|j |j¡s°t‚t ||¡ t ||	¡ t ||
¡ |j}|j dddd}|  dd¡j}t ||¡ d}tj	t
|d ttd	ƒƒ  ¡  W 5 Q R X d S )
NÚAr   é   ÚfirstÚsecond)Zaxisz,Can only swap levels on a hierarchical axis.)Úmatché   )Z	swaplevelÚindexÚequalsÚAssertionErrorÚ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   úM/tmp/pip-unpacked-wheel-xj8nt62q/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   