a
    Pf                     @   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g dddgd}td	d
gddgddgg||d}|turf|jd d df }|jddd}|jddd}t|| |tu r| }t	ddgdd|_
|jddd}t|| n<tjtdd |jddd W d    n1 s0    Y  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/var/www/ai-form-bot/venv/lib/python3.9/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   