U
|
ý°dÐ( ã @ sV d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
|
Z G dd dZ dS )é N)Ú DataFrameÚSeriesc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d d d g¡d d! Zd"S )#ÚTestDataFrameMissingDatac C s
|
t |j}tj |¡}tj|d d
< td|i|jd}t||jdd}| ¡ }| ¡ | ¡ }}|jdd} t
|
|d |¡ |jddd}
|
t
|
| d |¡ t
|
|d |¡ |
|
d ks¶t |jddgd } |jddgdd
|
}
|
t
|
| d |¡ t
|
|d |¡ |
|
d kst d S ) Né Úfoo©Úindex©r ÚnameÚall©ÚhowT)r Úinplace)r Úsubset)r r r ) Úlenr ÚnpÚrandomÚrandnÚnanr r ÚdropnaÚcopyÚtmÚassert_series_equalÚAssertionError) ÚselfÚ float_frameÚNÚmatÚframeÚoriginalÚexpectedZinplace_frame1Zinplace_frame2Ú smaller_frameÚ return_value© r# ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_dropna.pyÚtest_dropEmptyRows s$
|
z+TestDataFrameMissingData.test_dropEmptyRowsc C sF t |j}tj |¡}tj|d d
< td|i|jd}d|d< t||jdd}| ¡ | ¡ }}| ¡ }t
|
|d |¡ |j dd} t|dd
|jdd
dd}
|
t
|
|d |
|
¡ t
|
|d |
|
¡ | d ksÐt |j dgd} t
|
|d |¡ |d dk ¡ st |j dgdd } t
|
| j|j¡ t
|
|j|j¡ | d ksBt d S )
|
Nr r r Úbarr T©r ©r )r r )r r r r r r r r r r r r r r Zassert_index_equal) r r r r r r Z
|
inp_frame1Z
|
inp_frame2r! r" ÚexpZsamesize_framer# r# r$ Útest_dropIncompleteRows( s*
|
z0TestDataFrameMissingData.test_dropIncompleteRowsc C s® t tj dd¡}tj|jd d
df< |jdd}|jd d
dddgf }| ¡ }|jddd }t
|
||¡ t
|
||¡ |d kst |jdd}|jt t dd }| ¡ }|jddd }t
|
||¡ t
|
||¡ |d ksÞt |jdd
|
d }|jd d
dddgf }| ¡ }|jdd
|
dd }t
|
||¡ t
|
||¡ |d ksBt |jddd }|jt dd }| ¡ }|jdddd }t
|
||¡ t
|
||¡ |d kst |jddd }t
|
||¡ |jddd }t
|
||¡ |jddddgd }| ¡ }|jddddgdd}t
|
||¡ t
|
||¡ |d ks*t |jddd}t
|
||¡ tj|d< |jddd}|jd d
dddgf }t
|
||¡ d}tjt|d |jdd W 5 Q R X d S )Né é é é ©Úaxisr é T)r0 r r )r0 Úthresh)r0 r2 r )r0 r )r0 r r r )r0 r z)No axis named 3 for object type DataFrame©Úmatch)r r r r r Úilocr Úlocr r Úassert_frame_equalr ÚlistÚrangeÚpytestÚraisesÚ
|
ValueError)r ÚdfZdroppedr Úinpr" Úmsgr# r# r$ Ú test_dropnaC sb
|
z$TestDataFrameMissingData.test_dropnac C s t ddtjgdd}t ddg|jdd}td|j ¡ i}| ¡ }|d ¡ t |d |¡ |d }|jdd}t ||¡ t |d |¡ |d kst
|
|d dg¡ t |d |¡ |d }|j dgdd}t || dg¡¡ t |d |¡ |d kst
|
d S )Nr. r- ÚA)r
|
)Údtyper
|
Tr' ) r r r rB r Úvaluesr r r r r Zdrop)r r r r= Zdf2Zserr" r# r# r$ Ú test_drop_and_dropna_caching s$ z5TestDataFrameMissingData.test_drop_and_dropna_cachingc C sX d}t jt|d |jdd W 5 Q R X t jtdd |jddgd W 5 Q R X d S ) Nzinvalid how option: foor3 r r z ^\['X'\]$rA ÚXr( )r: r; r< r ÚKeyError)r r r? r# r# r$ Útest_dropna_corner s
|
z+TestDataFrameMissingData.test_dropna_cornerc C sÈ t dtjddgdtjddgtjtjtjtjgdtjdd gg}tjtd
|
d |jd d dgd W 5 Q R X tjtd
|
d |jd dd W 5 Q R X | ¡ }tjtd
|
d |jd ddd W 5 Q R X d S )Nr. r- r1 r, r r+ é é é zsupplying multiple axesr3 r r )r r0 )r r. T)r r0 r )r r r r: r; Ú TypeErrorr r )r r= r> r# r# r$ Útest_dropna_multiple_axes¥ s üÿ
|
z2TestDataFrameMissingData.test_dropna_multiple_axesc C s° t }tjdddtj ¡ d}tjdddtj ¡ d}|g|d< |jdd}t d|gi}t ||¡ t d|d tj |gi}|jdd}t ||gdgddgd }t ||¡ d S )
|
Niß r. )Útzinfor- ZTimer r/ r1 )Úcolumnsr )
|
r ÚdatetimeÚdateutilÚtzZtzutcr r r7 r r )r r= Zdt1Zdt2Úresultr r# r# r$ Ú test_dropna_tz_aware_datetime¹ s
|
z6TestDataFrameMissingData.test_dropna_tz_aware_datetimec C sN t j ddddg¡}t |¡}tdtdi|d}|}| ¡ }t ||¡ d S )Nr g=
|
×£p=@g
ëQ¸ @g
ëQ¸ @rA Úabcr ) ÚpdZ IntervalIndexZ from_breaksZCategoricalIndexr r8 r r r7 )r ÚiiÚcir= r rR r# r# r$ Ú&test_dropna_categorical_interval_indexÉ s
|
z?TestDataFrameMissingData.test_dropna_categorical_interval_indexc C sÈ t tj d¡tj d¡tj d¡dddddgd}tj|jdd d
|
dgf< tj|jd < tj|jd < tj|jd d
d f< |jdddgdd}ddddg|_ddddg|_|jddgdd}t ||¡ d S )Nr ÚaÚbÚcÚdÚe)rA ÚBÚCÚDr- r r. )r r )r. r. r1 rA r^ r_ r )r r )
|
r r r r r r5 r rN r r7 ©r r= r rR r# r# r$ Ú"test_dropna_with_duplicate_columnsÓ s
|
|
|
üÿ z;TestDataFrameMissingData.test_dropna_with_duplicate_columnsc C sb t dddgtddtjdgd}t ddgtdd d
|
gdd dgd }|jd d}t ||¡ d S )Nr. r- r1 rT r, r ©rA r^ r_ Úacç @g @r r r_ r( )r r8 r ÚNaNr r r7 ra r# r# r$ Ú test_set_single_column_subsetè s " ÿ z6TestDataFrameMissingData.test_set_single_column_subsetc C s> t ddddgi}tjtdd |jddd W 5 Q R X d S )
|
NrA r. r- r1 z['D']r3 r` r )r r0 )r r: r; rF r )r r= r# r# r$ Ú&test_single_column_not_present_in_axisñ s z?TestDataFrameMissingData.test_single_column_not_present_in_axisc C s` t ddtjgtddtjdgd}t dgdgd gd}|jt d
|
d g¡d }t ||¡ d S ) Nr. r- rT r, r rc g ð?rY re rA r_ r( )r r rf r8 r Úarrayr r7 ra r# r# r$ Útest_subset_is_nparrayù s $z/TestDataFrameMissingData.test_subset_is_nparrayc C sH t ddgddggt dd¡d}| ¡ }|j|d}tj||dd d S )
|
Nr. r- r1 r, r )rN r/ T)Zcheck_index_type)r rU Z
|
RangeIndexr r r r7 )r r0 r= r rR r# r# r$ Útest_no_nans_in_frame s z.TestDataFrameMissingData.test_no_nans_in_framec C s t ddtjg}d}tjt|d |jddd W 5 Q R X tjt|d |jddd W 5 Q R X tjt|d |jd d d W 5 Q R X d S )Nr. r- zAYou cannot set both the how and thresh arguments at the same timer3 r )r r2 Úany)r rU ZNAr: r; rK r )r r= r? r# r# r$ Ú"test_how_thresh_param_incompatible s z;TestDataFrameMissingData.test_how_thresh_param_incompatibleÚvalr. g ø?c C sd t ddd|gidddgd}|jdd}t ddd|gi}t ||¡ |jddd t ||¡ d S ) NrY r. r- r1 r T)Ú ignore_index)ro r )r r r r7 )r rn r= rR r r# r# r$ Útest_dropna_ignore_index s z1TestDataFrameMissingData.test_dropna_ignore_indexN)Ú__name__Ú
|
__module__Ú __qualname__r% r* r@ rD rG rL rS rX rb rg rh rj rk rm r: ÚmarkZ parametrizerp r# r# r# r$ r s A
|
r ) rO rP Únumpyr r: ZpandasrU r r Zpandas._testingZ_testingr r r# r# r# r$ Ú<module> s
|
