a
    Pf(                     @   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|vsJ d S )NA )selffloat_framer   r   a/var/www/ai-form-bot/venv/lib/python3.9/site-packages/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sBJ d|jv sPJ d|jv s^J |d }t	|tstJ |d= t|jdksJ d|jvsJ t
jttd	d
 |d= W d    n1 s0    Y  d|jvsJ t
jttd	d
 |d= W d    n1 s0    Y  d S )Nr   B      )   r   columnsr   )r   z('A',)match)r   Zfrom_productr   nprandomdefault_rngstandard_normallenr   
isinstancepytestraisesKeyErrorreescape)r   Zmidxdfresultr   r   r	   test_delitem_multiindex   s    $z,TestDataFrameDelItem.test_delitem_multiindexc                 C   sp   |  }|d= t|jdks J tjtdd |d= W d    n1 sJ0    Y  |d= t|jdkslJ d S )ND   z^'D'$r   r   r   )copyr   r   r   r   r   )r   r   fr   r   r	   test_delitem_corner+   s    $z(TestDataFrameDelItem.test_delitem_cornerc                 C   s`   g dg dg dg}t t| }t|}ttjdd|d}|d= t	|j
ts\J d S )N)abctop) r*   r*   ZOD)r*   r*   r*   Zwxr   )r"   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   numpyr   r   Zpandasr   r   r   r   r   r   r	   <module>   s   