o
    vh                     @   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}t| |d7 }W d    n1 sPw   Y  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_cow_warningassert_frame_equalr   assert_series_equal)selffloat_framewarn_copy_on_writeabexpected r"   a/var/www/html/hyperkenya/venv/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sJJ |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-   r-   F)check_namesr*   r
   )axis)sortedzipr   from_tuplesr   nprandomdefault_rngstandard_normalcopyr   r   r   r   r   drop)	r   arraystuplesr   r'   df1df2resultr!   r"   r"   r#   test_mixed_depth_pop1   s&   



z%TestDataFramePop.test_mixed_depth_popN)__name__
__module____qualname__r$   r)   rC   r"   r"   r"   r#   r      s    r   )
numpyr8   pandasr   r   r   pandas._testing_testingr   r   r"   r"   r"   r#   <module>   s    