B
    ²ô`Ê  ã               @   s.   d dl mZmZ d dlmZ G dd„ dƒZdS )é    )Ú	DataFrameÚSeriesNc               @   s   e Z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   ÚtmZassert_frame_equalr   Zassert_series_equal)ÚselfZfloat_frameÚaÚbÚexpected© r   úR/home/dcms/DCMS/lib/python3.7/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œƒ}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s€t‚t|jƒdks’t‚d S )	Nr   r   r
   r   )r   r   r   r   r   r   )r   r   r   Útyper   Úlenr   )r   ZdfÚresr   r   r   Útest_pop_non_unique_cols   s    
z)TestDataFramePop.test_pop_non_unique_colsN)Ú__name__Ú
__module__Ú__qualname__r   r!   r   r   r   r   r      s   r   )Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   