U
    sVc                     @   s:   d dl Z d dlZd dlZd dlmZmZ G dd dZdS )    N)	DataFrame
MultiIndexc                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestDataFrameDelItemc                 C   s   |d= d|kst d S )NA)AssertionError)selffloat_frame r	   L/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/frame/indexing/test_delitem.pytest_delitem   s    z!TestDataFrameDelItem.test_delitemc              	   C   s   t ddgddgg}ttjdd|d}t|jdks>td|jksLtd|jksZt|d }t	|tspt|d= t|jdkstd|jkstt
jttdd	 |d= W 5 Q R X d|jkstt
jttdd	 |d= W 5 Q R X d S )
Nr   B         columns)r   z('A',)match)r   Zfrom_productr   nprandomrandnlenr   r   
isinstancepytestraisesKeyErrorreescape)r   Zmidxdfresultr	   r	   r
   test_delitem_multiindex   s    z,TestDataFrameDelItem.test_delitem_multiindexc              	   C   s\   |  }|d= t|jdks ttjtdd |d= W 5 Q R X |d= t|jdksXtd S )ND   z^'D'$r   r   r   )copyr   r   r   r   r   r   )r   r   fr	   r	   r
   test_delitem_corner+   s    z(TestDataFrameDelItem.test_delitem_cornerc                 C   sh   ddddgddddgddddgg}t t| }t|}ttjdd	|d
}|d= t|j	tsdt
d S )Nabctop ZODZwxr"   r   r   )r&   r*   r*   )sortedzipr   from_tuplesr   r   r   r   r   r   r   )r   ZarraysZtuplesindexr   r	   r	   r
   !test_delitem_col_still_multiindex4   s    "
z6TestDataFrameDelItem.test_delitem_col_still_multiindexN)__name__
__module____qualname__r   r    r%   r/   r	   r	   r	   r
   r      s   	r   )r   Znumpyr   r   Zpandasr   r   r   r	   r	   r	   r
   <module>   s   