U
    sVc                     @   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'   M/tmp/pip-unpacked-wheel-xj8nt62q/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   