U
|
®ý°dt" ã @ s
|
d dl Zd dlZd dlmZmZ d dlmZ ej
|
ddedddddddgfdedddddddgfdedddddddgfg¡dd Z ej
|
ddeddddgfdeddddgfdeddddgfg¡d
|
d Z ej
|
d g e ed g¡dd ZG dd dZdS )é N)Ú CategoricalÚSerieszkeep, expectedÚfirstFTÚlastc C s t dddddddgt | ¡d}|jdkr4t d¡ t |j|d |¡ t |j|d || ¡ | ¡ }|j|d
|
d }|d kst
|
t ||| ¡ d S ) Né r é é é ©ÚdtypeÚboolz.tested separately in test_drop_duplicates_bool©ÚkeepT©r Úinplace) r Únpr ÚpytestÚskipÚtmÚassert_series_equalÚ
|
duplicatedÚdrop_duplicatesÚcopyÚAssertionError)Úany_numpy_dtyper ÚexpectedÚtcÚscÚ return_value© r úgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_drop_duplicates.pyÚtest_drop_duplicates s
|
|
r! c C sv t ddddg}t |j| d|¡ t |j| d|| ¡ | ¡ }|j| dd}t ||| ¡ |d ksrtd S )NTFr r )r r r r r r r )r r r r r r r r Útest_drop_duplicates_bool! s r" Úvaluesr c C s t |t | ¡d}t dgt| dd}|jdkrJ|d d
}|d d
}t |j|d|¡ |j|d}t ||¡ ||k std S )Nr
|
Fr é r ) r r r Úlenr r r r r )r r r# r r Zresult_droppedr r r Ú"test_drop_duplicates_no_duplicates4 s
|
r& c @ s e Zd Zejddddddgddd Zejd
|
d Zd d Zdd Zdd Z ejdd Z
|
dd Z dd Z dd Z dd Zd d Zd d Zd S )!ÚTestSeriesDropDuplicatesÚint_ZuintZfloat_Zunicode_ztimedelta64[h]z datetime64[D])Úparamsc C s |j S )N)Úparam)ÚselfÚrequestr r r r I s z TestSeriesDropDuplicates.dtypec C sT t jdddddgt |¡d}t jddddgt |¡d}t|||d}t|}|S ©Nr r$ r r r r
|
©Ú
|
categoriesÚordered©r Úarrayr r r )r+ r r0 Ú cat_arrayZinput1ÚcatÚtc1r r r Úcat_series_unused_categoryO s
|
z3TestSeriesDropDuplicates.cat_series_unused_categoryc C sx |}t ddddg}| ¡ }t ||¡ | ¡ }t ||| ¡ | ¡ }|jdd}|d ksbtt ||| ¡ d S ©NFT©r ©r r r r r r r ©r+ r6 r5 r Úresultr r r r r Ú)test_drop_duplicates_categorical_non_boolY s zBTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_boolc C s |}t ddddg}|jdd}t ||¡ |jdd}t ||| ¡ | ¡ }|jddd}|d ksltt ||| ¡ d S ©NFTr r r r9 r: r r r Ú2test_drop_duplicates_categorical_non_bool_keeplasti s zKTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_bool_keeplastc C s |}t ddddg}|jdd}t ||¡ |jdd}t ||| ¡ | ¡ }|jddd}|d ksltt ||| ¡ d S ©NFTr r r9 r: r r r Ú3test_drop_duplicates_categorical_non_bool_keepfalse{ s zLTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_bool_keepfalsec C sZ t jdddddgt |¡d}t jdddddddgt |¡d}t|||d}t|}|S r- r1 )r+ r r0 r3 Zinput2r4 Útc2r r r Ú
|
cat_series s
|
"z#TestSeriesDropDuplicates.cat_seriesc C s~ |}t dddddddg}| ¡ }t ||¡ | ¡ }t ||| ¡ | ¡ }|jdd}|d kshtt ||| ¡ d S r7 r9 ©r+ rB rA r r; r r r r r Ú*test_drop_duplicates_categorical_non_bool2 s zCTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_bool2c C s |}t dddddddg}|jdd}t ||¡ |jdd}t ||| ¡ | ¡ }|jddd}|d ksrtt ||| ¡ d S r= r9 rC r r r Ú3test_drop_duplicates_categorical_non_bool2_keeplast§ s zLTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_bool2_keeplastc C s |}t dddddddg}|jdd}t ||¡ |jdd}t ||| ¡ | ¡ }|jddd}|d ksrtt ||| ¡ d S r? r9 rC r r r Ú4test_drop_duplicates_categorical_non_bool2_keepfalse· s zMTestSeriesDropDuplicates.test_drop_duplicates_categorical_non_bool2_keepfalsec C sp t tddddgddg|d}t ddddg}t | ¡ |¡ t | ¡ || ¡ | ¡ }|jdd}|d ksttt ||| ¡ t ddddg}t |jdd|¡ t |jdd|| ¡ | ¡ }|jddd}|d ksætt ||| ¡ t ddddg}t |jdd|¡ t |jdd|| ¡ | ¡ }|jddd}|d ksZtt ||| ¡ d S )NTFr. r8 r r r )r r r r r r r r )r+ r0 r r r r r r r Ú%test_drop_duplicates_categorical_boolÇ s8
|
ÿÿ z>TestSeriesDropDuplicates.test_drop_duplicates_categorical_boolc C s` t tdddd|gddgdd}| ¡ }t tddtjgddgdddddgd}t ||¡ d S )NTFr. r r r )Úindex)r r r r Únanr r )r+ Z nulls_fixtureÚserr; r r r r Ú(test_drop_duplicates_categorical_bool_naæ s ýÿþzATestSeriesDropDuplicates.test_drop_duplicates_categorical_bool_nac C s: t ddddg}|jdd}t dddg}t ||¡ d S )Nr r$ r T)Z ignore_index)r r r r )r+ rJ r; r r r r Ú!test_drop_duplicates_ignore_indexö s z:TestSeriesDropDuplicates.test_drop_duplicates_ignore_indexN)Ú__name__Ú
|
__module__Ú __qualname__r Zfixturer r6 r< r> r@ rB rD rE rF rG rK rL r r r r r' H s" ÿ
|
|
|
r' )Únumpyr r Zpandasr r Zpandas._testingZ_testingr ÚmarkZ parametrizer! r" ÚlistÚranger& r' r r r r Ú<module> s, ýþ
|
ýþ
|
|
|
