o
    5cD                     @   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
MultiIndexSeriesc                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestDataFramePopc                 C   s   d|j _|d d|vsJ d|d< |d d|vsJ |j jdks&J tg dg dgg ddd	gd
}|d}|d7 }tddgddggddgdd	gd
}t|| tddgdd	gddd }t|| d S )NbazAbarfoo)         )         )r   BCXY)columnsindexr   r
   r   r   r   r   r   r   )r   name)r   r   popr   tmassert_frame_equalr   assert_series_equal)selffloat_frameabexpected r    Y/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/tests/frame/methods/test_pop.pytest_pop   s   

"
"zTestDataFramePop.test_popc                 C   s   t ddgddgddgd}g d|_|d}t|t ks J t|dks(J t|jdks1J d	|jv s8J d|jvs?J t|jdksHJ d S )
Nr   r
   r   r   )r   r
   r   )r   r   r   r   r   r   )r   r   r   typelenr   )r   dfresr    r    r!   test_pop_non_unique_cols$   s   

z)TestDataFramePop.test_pop_non_unique_colsc           	      C   s   g dg dg dg}t t| }t|}ttjdd|d}| }| }|	d}|	d}t
j||d	d
 t
|| |jdksHJ |d }|jdgdd}|	d}t
|| t
|| d S )N)r   topr(   routine1r)   routine2) ODr,   result1result2r-   )r+   wxwyr+   r+   r+   r   r   )r   r   )r   r+   r+   F)check_namesr(   r
   )axis)sortedzipr   from_tuplesr   nprandomrandncopyr   r   r   r   r   drop)	r   arraystuplesr   r%   df1df2resultr   r    r    r!   test_mixed_depth_pop0   s&   



z%TestDataFramePop.test_mixed_depth_popN)__name__
__module____qualname__r"   r'   r@   r    r    r    r!   r      s    r   )
numpyr6   pandasr   r   r   pandas._testing_testingr   r   r    r    r    r!   <module>   s    