3
UédÊ  ã               @   s,   d dl mZmZ d dlj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 _|jdƒ d|kst‚d|d< |j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}|jdƒ}|d7 }tddgdd
ggddgddgd}tj||ƒ tdd	gddgddd }tj||ƒ 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   úW/var/www/html/virt/lib64/python3.6/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|_|j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   Ú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   
