U
|
ý°d: ã @ sÞ d dl m Z d dlZd dlZd dlZd dlmZmZmZ d dl m
|
Z ej dddgddgg¡dd Zd d
|
Zd d Zd d Zdd Zej de eg dedddgdeg dedddgdg¡dd Zdd Zdd Zd d Zej d d d g¡ej d!dd"d"d#gidd"d#gid d d$gfdd"d"d#gidd"d#gid d d"gfd"d"d#gd"d"d%gd&d"d#gd"d%gd&d d d$gfd"d"d#gd"d"d%gd&d"d#gd"d%gd&d d d"gfg¡d'd( Zd)d* Zd+d, Zej d-d$gd$d.g d g¡d/d0 ZdS )1é )ÚdatetimeN)Ú DataFrameÚNaTÚconcatÚsubsetÚaÚBc C sT t dddgdddgdddgd}t d¡}tjt|d | | ¡ W 5 Q R X d S )Nr é )ÚAr ÚCz Index(['a'], dtype='object')©Úmatch)r ÚreÚescapeÚpytestÚraisesÚKeyErrorÚdrop_duplicates)r ÚdfÚmsg© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_drop_duplicates.pyÚ0test_drop_duplicates_with_misspelled_column_name s "
|
r c C sø t ddddddddgddddddddgddddddddgtdd} | d ¡}| d d
}t ||¡ | jd d
|
d }| jd d g }t ||¡ | jd dd }| jg }t ||¡ t|dksÄt| jddddg }| t d dg¡¡}t ||¡ | d dg¡}t ||¡ | jdd
|
d }| jddd d g }t ||¡ | jddd }| jdg }t ||¡ | jd d
d ddgf }| ¡ }| d dg¡}t ||¡ |jd
|
d }|jd dgd
|
d }t ||¡ |jdd }|jd dgdd }t ||¡ | d¡}| j
|
ddg }t ||¡ | jdd
|
d }| j
|
ddg }t ||¡ | d d¡| d< | d¡}| j
|
ddg }t ||¡ | jdd
|
d }| j
|
ddg }t ||¡ t d d dddddgdd dddddgd} | j| j dk }t | ¡ |¡ t ddgddgg} t | ¡ | ¡ t ddgdd gg} t | ¡ | ¡ t tj¡jd d }t | |gd|d gg} t | ¡ | ¡ t | |g||d gg} t | ¡ | ¡ t d d td D } t| t dgdgd ggd d!} d"D ] }| j|d ¡ dksÒtqÒd S )#NÚfooÚbarÚoneÚtwor é é ©ÚAAAr r ÚDr Úlast©Úkeepé é Fr é r )r r é r éþÿÿÿéÿÿÿÿZint8ÚEé é )ÚxÚyéüÿÿÿc s s | ]}|gd V qdS )r- Nr )Ú.0Úir r r Ú <genexpr>r s z'test_drop_duplicates.<locals>.<genexpr>é T©Ú ignore_index)Úfirstr" F)r Úranger ÚtmÚassert_frame_equalÚlocÚlenÚAssertionErrorÚnpÚarrayÚilocZastypeÚindexZiinfoÚint64Úmaxr Z
|
duplicatedÚsum)r ÚresultÚexpectedÚdf2r. r$ r r r Útest_drop_duplicates s üÿ
|
|
|
|
*"rH c C sd t dddgdddgdddggddd gd
|
} | ¡ }t || ¡ | d¡}| d d
}t ||¡ d S ) Nr r r( r' r, r% r& r Úb©Úcolumns)r r r9 r: )r Zresult0Zresult1Z expected1r r r Ú0test_drop_duplicates_with_duplicate_column_namesy s *
|
rL c C sZ t ddddddddgddddddddgddddddddgtd d
|
} | d ¡}| jd ddd g }t ||¡ | jd dd}| jddd dg }t ||¡ | jd dd}| jdd g }t ||¡ | d dg¡}| jd ddddd g }t ||¡ | jd dgdd}| jd dddd dg }t ||¡ | jd dgdd}| jd ddd g }t ||¡ d S )Nr r ÚbazÚquxr r r r r r r r r% r" r# r( r& Fr r' r, )r r8 r r@ r9 r: ©r rE rF r r r Ú!test_drop_duplicates_for_take_all
s2 üÿ
|
rP c C sð t ddddddddgddddddddgddddddddgtdd} | d ¡}| d d
}t ||¡ | jd d
|
d }| jd d g }t ||¡ | jd dd }| jg }t|dks¸tt ||¡ | jddddg }| d¡}t ||¡ d S )Nr r r r r r r )©ZAAZABr r r! rQ r" r# r% r& Fr r' )rQ r )r r8 r r9 r: r; r< r= rO r r r Útest_drop_duplicates_tuple© s( üÿ
|
|
|
rR r rJ r
|
r ©rA c C s8 | ¡ }t || ¡ | ¡ }|j dd t || ¡ d S )NT©Úinplace)r r9 r: Úcopy)r rE r r r Útest_drop_duplicates_emptyÆ s
|
rW c C s¬ t d d ddddddgddddddddgdtjtjtjddddgtdd} | d ¡}| jd
|
d d g }t ||¡ | jd d d}| jdddg }t ||¡ | jd dd}| jg }t ||¡ t|d
|
ksÐt | d dg¡}| jd
|
d d dg }t ||¡ | jd dgd d}| jddddg }t ||¡ | jd dgdd}| jdg }t ||¡ t ddddddddgddddddddgdtjtjtjddddgtdd} | d¡}| d d
}t ||¡ | jdd d}| jd dg }t ||¡ | jddd}| jg }t ||¡ t|d
|
ks"t | ddg¡}| jd
|
dd dg }t ||¡ | jddgd d}| jdd ddg }t ||¡ | jddgdd}| jdg }t ||¡ d S )Nr r r r ç ð?r r ©r
|
r r r! r
|
r r r' r" r# r% r& Fr r( r r, )
|
r r> Únanr8 r r; r9 r: r< r= rO r r r Útest_drop_duplicates_NAÚ sh üÿ
|
|
üÿ
|
|
r[ c
|
C s4 t d d ddddddgdtjtjtjddddgd} | d ¡}| jd
|
d dd d g }t ||¡ | jd dd}| jddd dd g }t ||¡ | jd dd}| jd d g }t ||¡ | d¡}| jd
|
dd dg }t ||¡ | jddd}| jdd dd g }t ||¡ | jddd}| jd dg }t ||¡ d S )Nr r rM rN rX g @r' )r
|
r r
|
r r r( r& r" r# r r, r% Fr )r r> rZ r r@ r9 r: rO r r r Ú$test_drop_duplicates_NA_for_take_all$ s. þÿ
|
|
r\ c C sà t ddddddddgddddddddgddddddddgtdd} | ¡ }|jd d
|
d }| d d
}|}t ||¡ |d kst| ¡ }|jd d d
|
d }| jddg }|}t ||¡ |d ksÆt| ¡ }|jd dd
|
d }| jg }|}t ||¡ t|dks
|
t|d kst| ¡ }|jd dgd
|
d }| jddddg }|}t ||¡ |d ksbt| ¡ }|jd dgd d
|
d }| jddddg }|}t ||¡ |d ks®t| ¡ }|jd dgdd
|
d }| jdg }|}t ||¡ |d ksôt| jd d
d ddgf ¡ }| ¡ }|jd
|
d }| d dg¡}|}t ||¡ |d ksPt| ¡ }|jd d
|
d }|jd dgd d}|}t ||¡ |d kst| ¡ }|jdd
|
d }|jd dgdd}|}t ||¡ |d ksÜtd S )Nr r r r r r r rY r
|
TrT r" )r$ rU r% r& Fr r r' r( r r# ) r r8 rV r r9 r: r= r; r< )Úorigr Z return_valuerF rE Zorig2rG r r r Ú test_drop_duplicates_inplaceJ s~ üÿ
|
r^ rU TFz4origin_dict, output_dict, ignore_index, output_indexr r' r r, )r
|
r c C s^ t |}t ||d}| r0| ¡ }|j|| d n|j|| d}t ||¡ t |t |¡ d S )NrS )r6 rU )r rV r r9 r: )rU Z origin_dictZ output_dictr6 Z output_indexr rF Z result_dfr r r Ú!test_drop_duplicates_ignore_index s r_ c C s0 t d| gddggtd}| ¡ }t ||¡ d S )Nr r r )Zdtype)r Úobjectr r9 r: )Z nulls_fixturer rE r r r Ú*test_drop_duplicates_null_in_object_columnµ s ra c C sª t dddddgddtjtjtjgddtjtjdgdddddgtdddtdddtdddttgd}|jD ]6}||g j| d }|| j| d }t || ¡ ¡ qnd S )
|
Nr r r r' Zthreer, iß )r rI ÚcÚdÚer# )
|
r r> rZ r r rK r r9 r: Zto_frame)r$ r ÚcolumnZ dropped_frameZdropped_seriesr r r Ú(test_drop_duplicates_series_vs_dataframe¼ s"
|
|
|
ûûÿ
|
rf ÚargÚTruec C sB t dddddgi}d}tjt|d |j| d W 5 Q R X d S )Nr r r r' zC^For argument "ignore_index" expected type bool, received type .*.$r r5 )r r r Ú
|
ValueErrorr )rg r r r r r Ú-test_drop_duplicates_non_boolean_ignore_indexÓ s rj )r r Únumpyr> r Zpandasr r r Zpandas._testingZ_testingr9 ÚmarkZ parametrizer rH rL rP rR rW r[ r\ r^ r_ ra rf rj r r r r Ú<module> sL
|
` $ ûþ
|
|
|
J&P ..üþ
|
