U
|
ý°d ã @ s: d dl Zd dlmZmZmZ d dlmZ G dd dZ dS )é N)Ú DataFrameÚ
|
MultiIndexÚSeriesc @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestDataFramePopc C sÞ d|j _| d¡ d|ks td|d< | d¡ d|ks<t|j jdksLttdddgdd d
|
ggdd d gd dgd}| d ¡}|d7 }tddgdd
|
ggdd gd dgd}t ||¡ tdd gd dgd dd }t ||¡ d S )NZbazÚAÚbarZfooé é é é é é Ú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 úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_pop.pyÚtest_pop s
|
|
(
|
" zTestDataFramePop.test_popc C s t ddgddgddgd}dddg|_| d¡}t|t ksBtt|dksRtt|jdksdtd|jksrtd|jkstt|jdkstd 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æ ddddddgddddddgdd d
|
dddgg}t t| }t |¡}ttj d d ¡|d }| ¡ }| ¡ }| d¡}| d¡}t
|
j ||dd t
|
||¡ |j dks¨t|d }|jdgdd}| d¡}t
|
||¡ t
|
||¡ d S )Nr ÚtopZroutine1Zroutine2Ú ZODZresult1Zresult2ZwxZwyr r )r )r r' r' F)Z check_namesr )Zaxis)ÚsortedÚzipr Ú from_tuplesr ÚnpÚrandomZrandnÚcopyr r r r r r Zdrop) r ZarraysZtuplesr r# Zdf1Zdf2Ú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 )
|
Únumpyr+ Zpandasr r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
