U
|
ý°d ã @ s² d dl Zd dlZd dlmZ d dlZd dlmZ d dl
|
m Z dd Z ej dejdg¡dd Zd d
|
Zej dejdg¡d d Zej jd d Zej dddg¡dd ZdS )é N)Úis_datetime64tz_dtype)Ú allow_na_opsc C s | }t |tdt|d ¡}| ¡ }tt |j¡}t |t
|
j rht
|
j |¡}|j |_ tj||dd nVt |t
|
jr¨t
|
j||jd}t|jr| ¡ }tj||dd nt |¡}t ||¡ d S )Né T©Úexact©Údtype)ÚnpÚrepeatÚrangeÚlenÚuniqueÚlistÚdictÚfromkeysÚvaluesÚ
|
isinstanceÚpdÚ
|
MultiIndexÚ from_tuplesÚnamesÚtmÚassert_index_equalÚIndexr r Ú normalizeÚarrayÚassert_numpy_array_equal)Úindex_or_series_objÚobjÚresultÚ unique_valuesÚexpected© r" úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/base/test_unique.pyÚ test_unique s
|
|
r$ Únull_objc C s& |}t |st d¡ n6t|dk r0t d¡ n t|tjrNt d| d¡ |j}| |dd
< t|}t
|
|t dt|d ¡}|||j d}| ¡ }t |j¡}d d
|
|D }| g| } t|tjrtj| |j d}
|
t|j rô| ¡ }|
|
¡ }
|
tj||
|
d d n t j| |j d}
|
t ||
|
¡ d S ) Nú$type doesn't allow for NA operationsr z%Test doesn't make sense on empty dataúMultiIndex can't hold 'ú'r é r c S s g | ]}t |¡s|qS r" )r Zisnull)Ú.0Úvalr" r" r# Ú
|
<listcomp>6 s
|
z$test_unique_null.<locals>.<listcomp>Tr )r ÚpytestÚskipr r r r Ú_valuesÚtyper r
|
r r r r r r r r r r r r r ) r% r r r ÚklassÚrepeated_valuesr Zunique_values_rawZunique_values_not_nullr r! r" r" r# Útest_unique_null s0
|
|
r3 c C sB | }t |tdt|d ¡}t| ¡ }|jdd|ks>td S )Nr F©Údropna)r r
|
r r r ÚnuniqueÚAssertionError)r r r! r" r" r# Ú test_nuniqueD s r8 c C sþ |}t |st d¡ n t|tjr6t d| d¡ |j}| |dd
< t|}t |t
|
dt |d ¡}|||j d}t|tj rº| ¡ t |jkst|jdd t |jd ksútn@t | ¡ }| ¡ td|d ksàt|jdd td|ksútd S )
|
Nr& r' r( r r) r r Fr4 )r r- r. r r r r/ r0 r r
|
r r r ZCategoricalIndexr6 Ú
|
categoriesr7 r Úmax)r% r r r r1 r2 Znum_unique_valuesr" r" r# Útest_nunique_nullK s r; c C sh d}| |gd }| ¡ }t|tjrHtjdgtd}tj||dd n tjdgtd}t ||¡ d S )Nu í ½r) r Tr )
|
r r r r Úobjectr r r r r )Zindex_or_seriesZuvalr r r! r" r" r# Útest_unique_bad_unicoded s r= r5 TFc C s@ t ddt jtjd t jg¡}| | ¡}| r4|dks<nds<td S )NÚyesr é )r ZSeriesZNAr ÚnanZNaTr6 r7 )r5 ZserÚresr" r" r# Útest_nunique_dropnat s
|
rB )Únumpyr r- Zpandas.core.dtypes.commonr Zpandasr Zpandas._testingZ_testingr Zpandas.tests.base.commonr r$ ÚmarkZ parametrizer@ r3 r8 r; Z
|
single_cpur= rB r" r" r" r# Ú<module> s
|
#
|
|
|
