U
|
ý°d, ã @ s² d dl mZ d dlZd dlZd dlmZ d dlmZm Z m
|
Z
|
m Z d dl m Z ej ddddgg¡dd Zd
|
d Zej d d dd dg¡dd Zdd Zej ddddgd d dgd dd gg¡dd Zdd Zdd Zdd Zej d d d g¡ej d d!d"g¡d#d$ Zej d%de d"d"d"d!d!d"g¡fd&e d"d!d!d"d"d"g¡fd"e d"d!d!d!d!d"g¡fg¡d'd( Zejj d)d* Z d+d, Z d-d. Z ej d/ej ej!g¡d0d1 Z"d2d3 Z#dS )4é )ÚproductN)Ú hashtable)ÚNAÚ DatetimeIndexÚ
|
MultiIndexÚSeriesÚnamesÚfirstÚsecondc C s t jddddgddddgg| d}| ¡ }t jdddgdddgg|jd}t ||¡ t jtdtdg| d}| ¡ }t jtdtdg|jd}t ||¡ t jtdtdg| d}| ¡ }t jdgdgg|jd}t ||¡ t jg g g| d}| ¡ }t ||¡ d S ) Né é ©r ZaaaaZababZaaÚabÚa)r Ú from_arraysÚuniquer ÚtmÚassert_index_equalÚlist)r ÚmiÚresÚexp© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_duplicates.pyÚ test_unique s " r c C s t ddddddg} t ddddddgdd}t | |g¡ ¡ }t ddddg}t ddddgdd}t ||g¡}t ||¡ d S )Nz
|
2015-01-01ZNaTz
|
2015-01-02z
|
Asia/Tokyo)Útz)r r r r r r )Zidx1Zidx2ÚresultZeidx1Zeidx2r r r r Útest_unique_datetimelike) s ÿþ
|
ÿr Úlevelr c C sª | j |d}| |¡ ¡ }t ||¡ tjddddgddddggddgd }|j |d}| |¡}t ||¡ tjg g gddgd }|j |d}| |¡}t ||¡ d S )
|
N©r r é r é é r r
|
r )r Zget_level_valuesr r r r )Úidxr r Úexpectedr r r r Útest_unique_level; s &
|
|
r% c C sª d} t jt| d, tdgd tdgdgd tdg}W 5 Q R X t dddddgddddd gg¡}d
|
} t jt| d( | dddddgddd d d gg¡ W 5 Q R X d S ) Nz3Level values must be unique: \[[A', ]+\] on level 0)ÚmatchÚAé
|
r ÚBr r r z4Level values must be unique: \[[AB', ]+\] on level 0éþÿÿÿr" )ÚpytestZraisesÚ
|
ValueErrorr Úranger Z
|
set_levels)Úmsgr r r r Ú test_duplicate_multiindex_codesO s 0"r/ r Úbr c C s t jddggd | d}|j| ks&tt ddggd ¡}| | ¡}|j| ksRt|j| d ddd |j| d | d gddgd}|j| kstd S ) Nr r r r T)r Zinplacer r )r Z from_productr ÚAssertionErrorÚrename)r r r r r Útest_duplicate_level_names] s
|
r3 c
|
C s t ddgdddggdddddddgdddddddggd} | | d d g¡| d dg¡| ddg¡fD ]"}|jsrt| ¡ j|jksdtqdd S )Nr r r ©ÚlevelsÚcodesZNumZUpper)r Z set_namesÚhas_duplicatesr1 Údrop_duplicatesr )r r# r r r Útest_duplicate_meta_datan s "ÿ ü
|
r9 c
|
C s2 | j dkst| jdks t|j dks*t|jdks8ttddgdddggdddddddgdddddddggd}|j dkst|jdksttddgddggd ddddgd ddddggd}|j dksÌt|jdksÚttddgddggd d ddddgd d ddddggd}|j dks t|jdks.td S )
|
NTFr r r r4 r r0 éÿÿÿÿ)Z is_uniquer1 r7 r )r# Úidx_dupr Zmi_nanZ
|
mi_nan_dupr r r Útest_has_duplicates~ s, "ÿ ÿ ÿr< c C s@ ddddddddd d
|
d d d dddddg} t | ¡}|jr<td S )N)ÚxÚoutÚzr" ÚyÚinr? é© )r= r> r? é r@ rA r? éw )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? éz )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? é´ )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? é )r= r> r? é r@ rA r? éo )r= r> r? é r@ rA r? ér )r= r> r? é r@ rA r? éy )r= r> r? é r@ rA r? é~ )r= r> r? é r@ rA r? é )r= r> r? é! r@ rA r? é{ )r= r> r? é r@ rA r? é )r Ú from_tuplesr7 r1 )Útr r r r Ú test_has_duplicates_from_tuples s* î
|
rg Únlevelsr! é Ú
|
with_nullsTFc s t t d¡d¡ t d¡}|rd d< fddt| D t| D ] }d | d| | d < qF t ddg¡ d¡g7 n g| t d¡ d¡g |g| ddgg }t| d}|jrÆt|rîd d
|
}t t
|
| t| d}n |j ¡ }t ||d g ¡}|jstd S ) Niô r r: c s g | ] } ¡ qS r )Úcopy)Ú.0Úi©r6 r r Ú
|
<listcomp>À s z0test_has_duplicates_overflow.<locals>.<listcomp>r r r4 c S s t | d| d ¡S )Néè r )ÚnpÚinsert)r r r r ÚfÑ s z'test_has_duplicates_overflow.<locals>.f)rq ZtileÚaranger- ÚarrayÚrepeatr r7 r1 r ÚmapÚvaluesÚtolistre )rh rj r rm r5 r rs rx r rn r Ú test_has_duplicates_overflowµ s&
|
|
|
rz zkeep, expectedÚlastc C s | j |d}t ||¡ d S )N©Úkeep)Ú
|
duplicatedr Úassert_numpy_array_equal)r; r} r$ r r r r Útest_duplicatedÝ s r c st d\ t ¡t ¡dt ¡ g} fdd|D }t||d}|j| d}tj|j| d}t ||¡ d S )N)éÈ i rp c s g | ]}t j ¡qS r )rq ÚrandomÚchoice)rl Zlev©ÚkÚnr r ro ï s z)test_duplicated_large.<locals>.<listcomp>r4 r| ) rq rt r ZmakeStringIndexr r~ r rx r )r} r5 r6 r r r$ r r r Útest_duplicated_largeê s r c C sú dD ]@} t d| gdtjgg¡}|jr*tt | ¡ tj ddd¡ qt
|
ddD ]¤}t
|
dd D ]}t t
|
d
|
|t
|
d
|
|}t t d d |
t d d |
gtj t |¡jd }t||d |d ksÊt|jrÔtt | ¡ tj t|dd¡ q^qPd S )N)ée éf r g @r Úbool©Údtyper é r" r: ZabcdeZWXYZr4 )r r rq Únanr7 r1 r r r~ Úzerosr- r r r Z permutationÚTÚlen)r r r Úmr6 r r r Útest_duplicated2÷ s"
|
þ
|
ÿr c C sz t ddddddgddddddgf¡} tjddddddgtd}| ¡ }t ||¡ |jtksbt t dddddgdddddgf¡}t
|
| ¡ |¡ t ddddddg¡}| jdd}t ||¡ |jtksÐt t dddddgdddddgf¡}t
|
| j dd|¡ t ddddddg¡}| jdd}t ||¡ |jtksDt t ddddgddddgf¡}t
|
| j dd|¡ d S ) Nr r r FTr r{ r| ) r r rq ru r r~ r r r r1 r r8 )r# r$ r~ r r r Ú test_duplicated_drop_duplicates s& & " " r r c C sn t ddddddddddg
|
td}t tjtjd dddddddtjtjtjd g
|
| d ¡ }t ||¡ d S ) NFTr y ð?r r y ð? ð?y ð? @)r r rq r r~ r Zassert_series_equal)r r$ r r r r Ú&test_duplicated_series_complex_numbers* s& þö ó
|
r c C s t ddttgdd} t ddddg¡}tj| |gddgd}| ¡ }t ddtgdd}t dddg¡}tj||gddgd}t ||¡ d S ) Nr r ZInt64r r r r0 r ) r r rq ru r r r r r )Zvals_aZvals_bZmidxr Z
|
exp_vals_aZ
|
exp_vals_br$ r r r Útest_midx_unique_ea_dtypeI s r )$Ú itertoolsr Únumpyrq r+ Z pandas._libsr Zpandasr r r r Zpandas._testingZ_testingr ÚmarkZ parametrizer r r% r/ r3 r9 r< rg rz ru r Zarm_slowr r r Z complex64Z
|
complex128r r r r r r Ú<module> sN
|
|
$
|
&ýþ
|
|
þþ
|
|
