U
    ¾mœdå  ã                   @   s:   d dl Z d dlmZmZmZ d dlmZ G dd„ dƒZdS )é    N)Ú	DataFrameÚIndexÚ
MultiIndexc                   @   s   e Zd Zdd„ ZdS )ÚTestDropLevelc              	   C   sú   t jddgddgd�}t jdddgd	d
gd�}tddgddgddgg||d�}|tk	rh|jd d …df }|jd	dd�}|jd	dd�}t ||¡ |tkrÎ| ¡ }t	ddgdd�|_
|jddd�}t ||¡ n(tjtdd�� |jddd� W 5 Q R X d S )N)ÚcÚe)ÚdÚfZlevel_1Zlevel_2)Únames)é   é   )é   é   )é	   é
   ÚaÚbé   é   é   é   é   é   )ÚindexÚcolumnsr   T)Zdropr   )Zaxisr   r   )Únamer   zNo axis named columns)Úmatchr   )r   Úfrom_tuplesr   ZilocZreset_indexZ	droplevelÚtmZassert_equalÚcopyr   r   ÚpytestZraisesÚ
ValueError)ÚselfZframe_or_seriesÚcolsÚmiZdfÚexpectedÚresult© r'   úb/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/tests/frame/methods/test_droplevel.pyÚtest_droplevel   s$     ÿ zTestDropLevel.test_droplevelN)Ú__name__Ú
__module__Ú__qualname__r)   r'   r'   r'   r(   r      s   r   )	r    Zpandasr   r   r   Zpandas._testingZ_testingr   r   r'   r'   r'   r(   Ú<module>   s   