o
    թZh                     @   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r(J t|| t|| t|| |  }| d}| dd}	| dd}
|j|jsXJ t|| t||	 t||
 |j}|j dddd}| ddj}t|| d}tjt	|d t
td	   W d    d S 1 sw   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   `/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/frame/methods/test_swaplevel.pytest_swaplevel   s2   
"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    