o
    թZh                     @   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 )NZbazAbarZfoo)         )         )r   BCXY)columnsindexr   r   r
   r   r   r   r	   r   )r   name)	r   r   popr   tmZassert_cow_warningassert_frame_equalr   assert_series_equal)selfZfloat_frameZwarn_copy_on_writeabexpected r   Z/var/www/html/lang_env/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&   Zroutine2) ODr(   result1Zresult2r)   )r'   ZwxZwyr'   r'   r'   r	   )r   r   )r   r   )r   r'   r'   F)Zcheck_namesr%   r   )Zaxis)sortedzipr   from_tuplesr   nprandomZdefault_rngZstandard_normalcopyr   r   r   r   r   Zdrop)	r   ZarraysZtuplesr   r"   Zdf1Zdf2resultr   r   r   r   test_mixed_depth_pop1   s&   



z%TestDataFramePop.test_mixed_depth_popN)__name__
__module____qualname__r   r$   r1   r   r   r   r   r      s    r   )
numpyr-   Zpandasr   r   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s    