U
    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|kstd|d< |d d|ks<t|j jdksLttdddgdd	d
ggdddg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 )NZbazAbarZfoo                  BCXY)columnsindex)r   name)	r   r   popAssertionErrorr   tmassert_frame_equalr   assert_series_equal)selfZfloat_frameabexpected r   G/tmp/pip-unpacked-wheel-g7fro6k3/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}dddg|_|d}t|t ksBtt|dksRtt|jdksdtd|jksrtd|jkstt|jdkstd S )	Nr   r   r   r   )r   r   r	   r   r   r	   )r   r   r   typer   lenr   )r   dfresr   r   r   test_pop_non_unique_cols$   s    
z)TestDataFramePop.test_pop_non_unique_colsc           	      C   s   ddddddgddddddgdd	d
dddgg}t t| }t|}ttjdd|d}| }| }|	d}|	d}t
j||dd t
|| |jdkst|d }|jdgdd}|	d}t
|| t
|| d S )Nr   topZroutine1Zroutine2 ZODZresult1Zresult2ZwxZwyr   r   )r   )r   r'   r'   F)Zcheck_namesr   )Zaxis)sortedzipr   from_tuplesr   nprandomZrandncopyr   r   r   r   r   r   Zdrop)	r   ZarraysZtuplesr   r#   Zdf1Z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   )
Znumpyr+   Zpandasr   r   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   