o
    Mf(                     @   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   j/var/www/html/analyze/labelStudio/lib/python3.10/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s!J d|jv s(J d|jv s/J |d }t	|ts:J |d= t|jdksFJ d|jvsMJ t
jttd	d
 |d= W d    n1 sew   Y  d|jvsqJ t
jttd	d
 |d= W d    d S 1 sw   Y  d S )Nr   B      )   r   columnsr   )r   z('A',)match)r   from_productr   nprandomdefault_rngstandard_normallenr   
isinstancepytestraisesKeyErrorreescape)r   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 s%w   Y  |d= t|jdks6J 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,   OD)r,   r,   r,   wxr   )r$   r   r   )r(   r,   r,   )sortedzipr   from_tuplesr   r   r   r   r   r   r   )r   arraystuplesindexr    r   r   r	   !test_delitem_col_still_multiindex4   s   
z6TestDataFrameDelItem.test_delitem_col_still_multiindexN)__name__
__module____qualname__r
   r"   r'   r5   r   r   r   r	   r      s
    	r   )r   numpyr   r   pandasr   r   r   r   r   r   r	   <module>   s
    