U
|
ý°d ã @ sP d dl Z d dlZd dlZd dlmZmZmZ d dlm Z
|
ej dddgddgg¡dd Z ej jdd Zej d
|
d ed d d d d gfded d d d d gfd ed d d d d gfg¡dd Zej jddej d
|
d ed d d d d gfded d d d d gfd ed d d d d gfg¡dd Zej ddddgdg¡dd Zdd Zdd ZdS ) é N)Ú DataFrameÚSeriesÚ
|
date_rangeÚ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ÚpytestZraisesÚKeyErrorÚ
|
duplicated)r ÚdfÚmsg© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_duplicated.pyÚ+test_duplicated_with_misspelled_column_name s "
|
r c C sF dd t dD } t| j}| ¡ }t|ts2t|jtj ksBtd S )Nc S s& i | ] }d |dt j ddd¡qS )Zcol_Z02dr iè i0u )ÚnpÚrandomÚrandint)Ú.0Úir r r Ú
|
<dictcomp> s
|
zBtest_duplicated_do_not_fail_on_wide_dataframes.<locals>.<dictcomp>éd )
|
Úranger ÚTr Ú
|
isinstancer ÚAssertionErrorÚdtyper Zbool_)Údatar Úresultr r r Ú.test_duplicated_do_not_fail_on_wide_dataframes s
|
|
r% zkeep, expectedÚfirstFTÚlastc C s> t dddddgdddddgd}|j| d}t ||¡ d S ) Nr r é r ÚbÚc)r
|
r ©Úkeep)r r ÚtmÚassert_series_equal©r, Úexpectedr r$ r r r Útest_duplicated_keep( s " r1 z+GH#21720; nan/None falsely considered equal)Úreasonc C s< t tjddd tjgddtd}|j| d}t ||¡ d S )Né r )r Úx)r" r+ )r r ÚnanÚobjectr r- r. r/ r r r Útest_duplicated_nan_none7 s
|
r7 r
|
c C s t dddddgdddddgtjddd tjgd}| d krFt|j} nt| trV| g} || j|d }|j|| d
|
}t ||¡ d S ) Nr r r( r r) r* r3 r r+ )r, r )
|
r r r5 ÚlistÚcolumnsr Ústrr r- r. )r r, r r0 r$ r r r Útest_duplicated_subsetG s ýÿ
|
r; c C s8 t ddgd} | d¡}| | }| ¡ }t ||¡ d S )Nr r) )r9 )r r Úcopyr- Zassert_frame_equal)r Zdupesr$ r0 r r r Ú test_duplicated_on_empty_frame] s
|
|
r= c C s^ t ddd} td| d}| ddg¡}| ¡ s6ttd| i}|j ¡ }| ¡ sZtd S )Nz
|
2010-07-01z
|
2010-08-05)ÚendZAAA)ÚsymbolÚdater@ r? )r r r Úallr! r@ )ÚdatesZtstr$ r r r Ú test_frame_datetime64_duplicatedh s
|
rC )r Únumpyr r Zpandasr r r Zpandas._testingZ_testingr- ÚmarkZ parametrizer Zslowr% r1 Zxfailr7 r; r= rC r r r r Ú<module> s:
|
|
ýþ
|
ýþ
|
|
