U
|
¬ý°dPR ã @ sü d dl m Z d dlmZ d dlZd dlZd dlZd dlmZ m
|
Z d dl m mZ d dlmZmZmZmZmZ d dlmZ d dlZd dlmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ d dl,m-Z. d dl/m0 m1Z d dl2m3Z3m4Z4 d dl5m0 m6Z7 G d d
|
d
|
Z8G d d d Z9d d Z:G dd dZ;G dd dZ<G dd dZ=G dd dZ>G dd dZ?dd Z@G dd d ZAd d ZBd d ZCd!d" ZDd#d$ ZEd%d& ZFd'd( ZGG d)d* d*ZHG d+d, d,ZIejJ Kd-ejLejLg¡d.d/ ZMdS )0é )Údatetime)Ú permutationsN)ÚalgosÚ hashtable)Ú is_bool_dtypeÚis_complex_dtypeÚis_float_dtypeÚis_integer_dtypeÚis_object_dtype)ÚCategoricalDtype)Ú CategoricalÚCategoricalIndexÚ DataFrameÚ DatetimeIndexÚIndexÚ IntervalIndexÚ
|
MultiIndexÚNaTÚPeriodÚ PeriodIndexÚSeriesÚ TimedeltaÚ TimestampÚ
|
date_rangeÚtimedelta_rangeÚ to_datetimeÚ to_timedelta)Ú DatetimeArrayÚTimedeltaArrayc @ s´ e Zd Zej dddg¡dd Zdd Zdd Zd
|
d Z d d Z
|
dd Z dd Z dd Z ej ddddddgddddd gddddgfddddd gddddd gdddd gfddddgddddgdddgfg¡d d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zej dddg¡d0d1 Zej dddg¡d2d3 Zd4d5 Zej d6ejdddgd7d8ejd9dd9gd:d8ejd;d<d;gd=d8g¡d>d? Z ej d@ejddddgd7d8dfejddddgd7d8dfejd9dd9dgd:d8d9fejdd9ddgd:d8dfejdAdBdAdCge d8dAfejdDdEdDdFge d8dDfejdEdDdEdFge d8dEfg¡dGdH Z ej dddg¡ejjdIejdCdAdJdCge d8ejdCdAge d8fe jddej dgdKd8e jddgdKd8fgdLdMgdNdOdP Z!ej ddAdJdCdAgejddddge "dQ¡d8ejdAej dCge d8fdAej dCdAgejddddge "dQ¡d8ejdAej dCge d8fg¡dRdS Z#ej dddJddgejddddge "dQ¡d8ejdej dgdTd8fdej ddgejddddge "dQ¡d8ejdej dgej$d8fg¡dUdV Z%ej de&e'dAdAdCgejdddgej(d8e)dAdCgdAdCgdWdXfe*e'dAdAdCgejdddgej(d8e)dAdCgdAdCgdWdXfe*e+dYdYgdZd[ejddgej(d8e+dYgdZd[fg¡d\d] Z,dJS )^Ú TestFactorizeÚsortTFc
|
sÎ |}|j |d\}}t}t|tr(tj}| ¡ }|jtjkrH| tj
|
¡}||}t|trx|jt krx|jt krx| t ¡}|r| ¡ }t| fdd|D } tj| tjd} t || ¡ tj||dd d S )N©r c s g | ]} |¡qS © )Úindex)Ú.0Úval©Zexpected_uniques_listr" úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_algos.pyÚ
|
<listcomp>N s z0TestFactorize.test_factorize.<locals>.<listcomp>©ÚdtypeT©Úexact)Ú factorizer Ú
|
isinstancer Ú from_tuplesÚuniquer* ÚnpÚfloat16ÚastypeÚfloat32ÚboolÚobjectZ sort_valuesÚlistÚasarrayÚintpÚtmÚassert_numpy_array_equalÚassert_index_equal)
|
ÚselfZindex_or_series_objr ÚobjZ result_codesZresult_uniquesÚ constructorZ expected_arrÚexpected_uniquesÚexpected_codesr" r& r' Útest_factorize5 s.
|
ÿþý
|
z TestFactorize.test_factorizec C sp t dddt jg¡}t|}|jdd\}}t jddddgt jd}tddt jg}t ||¡ t ||¡ d S ) Né é F©Úuse_na_sentinelr r) ç ð?ç @)
|
r1 ÚarrayÚnanr r- r9 r r: r; r< )r= ÚvaluesÚserÚcodesÚuniquesrA r@ r" r" r' Ú+test_series_factorize_use_na_sentinel_falseT s z9TestFactorize.test_series_factorize_use_na_sentinel_falsec
|
C sJ t ddddddddg¡\}}t |tjdddgtd¡ t jddddddddgdd\}}tjdddddd d d gtjd}t ||¡ tjdddgtd}t ||¡ tjd
|
tjdd d d
}t |¡\}}tjddd d d gtjd}t ||¡ tjd d d ddg|j d}t ||¡ t j|dd\}}tjd d d ddgtjd}t ||¡ tjddd d d g|j d}t ||¡ t d¡d d d
}t |¡\}}tjddd d d gtjd}t ||¡ tjdddddg|j d}t ||¡ t j|dd\}}tjd d d ddgtjd}t ||¡ tjdddddg|j d}t ||¡ d S )NÚaÚbÚcr) Tr! r rC rD é éÿÿÿÿé é ç @ç @g @rH rG ç )
|
r r- r: r; r1 rI r6 r9 Úaranger* )r= rM rN ÚexpÚarrr" r" r' Ú
|
test_basic` s@ ÿ
|
zTestFactorize.test_basicc C sÈ t ddtjddtjg}t |¡\}}tjddddddgtjd }t ||¡ t
|
dddtjg}t ||¡ tj|d
|
d \}}tjddddddgtjd }t ||¡ t
|
dtjddg}t ||¡ d S ) NÚAÚBg
ëQ¸ @r rT rC rD rU r) Tr! ) r r1 rJ Úinfr r- rI r9 r: r; r r< )r= ÚxrM rN r[ r" r" r' Ú
|
test_mixed s zTestFactorize.test_mixedc C sN t d}t d}t||||||g}t |¡\}}tjddddddgtjd}t ||¡ t ||g}t
|
||¡ tj|dd\}}tjddddddgtjd}t ||¡ t ||g}t
|
||¡ t dd d
|
}t d d d
|
}t||||||g}t |¡\}}tjddddddgtjd}t ||¡ t
|
|t ||g¡ tj|dd\}}tjddddddgtjd}t ||¡ t
|
|t ||g¡ t d }t d }t|||||||g}t |¡\}}tjdddddddgtjd}t ||¡ t
|
|t ||g¡ tj|dd\}}tjdddddddgtjd}t ||¡ t
|
|t ||g¡ d S )Nz20130101 09:00:00.00004Ú20130101r rC r) Tr! Z201302ÚM©ÚfreqZ201303z 1 day 1 minú1 day)r r r r- r1 rI r9 r: r; r r< r r r )r= Zv1Zv2ra rM rN r[ r" r" r' Ú test_datelike sF zTestFactorize.test_datelikec C s t jdddt jgdd}t t|¡}dD ]h}|j||d}t jddd|gt jd}tt|tt|kspt t
|
t |¡||k¡ t
|
||¡ q*d S )NrC rD ÚOr) )rT é )Ú na_sentinelr )r1 rI rJ ÚhtÚObjectFactorizerÚlenr- r9 ÚsetÚAssertionErrorr: r; ÚpdÚisna)r= ÚkeyÚrizerrk ÚidsÚexpectedr" r" r' Útest_factorize_nanÈ s z TestFactorize.test_factorize_nanc C s t jddddddgdd}t ddddddg¡}t t|¡}|j||d }t jdddddd
|
gt jd}t ||¡ t jdddgdd}t |j
|
¡ |¡ d S ) NrC rD rU r Úint64r) FT)ÚmaskrT ) r1 rI rl ZInt64Factorizerrn r- r9 r: r; rN Úto_array)r= Údatary rt Úresultrv r@ r" r" r' Útest_factorizer_with_maskÕ s z'TestFactorize.test_factorizer_with_maskc C s t ddddt jg¡}t t|¡}| | t¡¡}t jdddddgt j d}t
|
||¡ t jdddgtd}t
|
|j ¡ |¡ d S )NrC rD rU r rT r) )r1 rI rJ rl rm rn r- r3 r6 r9 r: r; rN rz )r= r{ rt r| rv r@ r" r" r' Ú test_factorizer_object_with_nanà s z-TestFactorize.test_factorizer_object_with_nanz&data, expected_codes, expected_uniques©rC rC )rC rD ©r r Znonsenser rC rD rU )rC rD rU c C sD t |¡\}}t |tj|tjd¡ tj|t d}t ||¡ d S ©Nr) )
|
rq r- r: r; r1 rI r9 ÚcomZasarray_tuplesafer6 )r= r{ rA r@ rM rN Zexpected_uniques_arrayr" r" r' Útest_factorize_tuple_listê s z'TestFactorize.test_factorize_tuple_listc C sV t jdd tdD td}d}tjt|d tj|d d d
dd W 5 Q R X d S )
|
Nc S s g | ] }t |qS r" )Úcomplex)r$ Úir" r" r' r( s z6TestFactorize.test_complex_sorting.<locals>.<listcomp>é r) z,'[<>]' not supported between instances of .*©ÚmatchrT Tr! ) r1 rI Úranger6 ÚpytestÚraisesÚ TypeErrorr r- )r= Zx17Úmsgr" r" r' Útest_complex_sorting s z"TestFactorize.test_complex_sortingc C sn |}t jddddg|d}t jddddgt jd}t jddg|d}t |¡\}}t ||¡ t ||¡ d S )NrC rD r) r )r1 rI r9 r r- r: r; )r= Zany_real_numpy_dtyper* r{ rA r@ rM rN r" r" r' Ú test_numeric_dtype_factorize s z*TestFactorize.test_numeric_dtype_factorizec C s t jddddddgt jd}|j|d t jddddddgt jd}t jdddgt jd}t |¡\}}t ||¡ t ||¡ d S ) NrG g ×Ag:0âyE>r) ©Úwriter rC rD ) r1 rI Úfloat64Úsetflagsr9 r r- r: r; ©r= Úwritabler{ rA r@ rM rN r" r" r' Útest_float64_factorize s z$TestFactorize.test_float64_factorizec C sv t jdddgt jd}|j|d t jdddgt jd}t jddgt jd}t |¡\}}t ||¡ t ||¡ d S )Nl ÿÿÿÿ rC r) r r ) r1 rI Úuint64r r9 r r- r: r; r r" r" r' Útest_uint64_factorize s z#TestFactorize.test_uint64_factorizec C sv t jdddgt jd}|j|d t jdddgt jd}t jddgt jd}t |¡\}}t ||¡ t ||¡ d S )Nl ÿÿÿÿ ìûÿÿÿ r) r r rC ) r1 rI rx r r9 r r- r: r; r r" r" r' Útest_int64_factorize* s z"TestFactorize.test_int64_factorizec C s| t jdddddgtd}|j|d t jdddddgt jd}t jdddgtd}t |¡\}}t ||¡ t ||¡ d S ) NrP rR rQ r) r r rC rD ) r1 rI r6 r r9 r r- r: r; r r" r" r' Útest_string_factorize4 s z#TestFactorize.test_string_factorizec C s t jddd t jddtdgtd}|j|d t jdddddd ddgt jd}t jdddgtd}t |¡\}}t
|
||¡ t
|
||¡ d S )
|
NrP rR rQ r) r r rC rT rD ) r1 rI rJ r r6 r r9 r r- r: r; r r" r" r' Útest_object_factorize> s z#TestFactorize.test_object_factorizec C sn t jt d¡gdd}|j|d t jdgt jd}t jdgdd}t |¡\}}t ||¡ t ||¡ d S )Nz2020-01-01T00:00:00.000úM8[ns]r) r r z 2020-01-01T00:00:00.000000000údatetime64[ns]) r1 rI Ú
|
datetime64r r9 rq r- r: r; r r" r" r' Útest_datetime64_factorizeH s ÿ z'TestFactorize.test_datetime64_factorizec C s t j td¡}tjdtjd|f}tj||d}t
|
|d |d ¡ t j |d |d dd |j|d}t
|
|d |d ¡ t j |d |d dd d S )Né
|
r) r! r rC Tr+ © rq Ú
|
RangeIndexZ
|
from_ranger r1 rZ r9 r r- r: r; r< )r= r Úrirv r| r" r" r' Útest_factorize_rangeindexU s z'TestFactorize.test_factorize_rangeindexc C sÜ t j td¡}tjdtjd|f}|d d d
}|d |f}|rf|d d d d
|d d d d
f}tj||d}t
|
|d |d ¡ t j |d |d dd |j|d}t
|
|d |d ¡ t j |d |d dd d S ) Nr¡ r) rT r rC r! Tr+ r¢ )r= r r¤ rv Zri2r| r" r" r' Ú$test_factorize_rangeindex_decreasingc s $ z2TestFactorize.test_factorize_rangeindex_decreasingc C sb t jdddgt jd}tjtdd tj|dd W 5 Q R X t d¡ t |¡ W 5 Q R X d S ) Nì rC r) zgot an unexpected keywordr T)ÚorderF)
|
r1 rI r r r r r r- r: Zassert_produces_warning)r= r{ r" r" r' Útest_deprecate_orderv s
|
z"TestFactorize.test_deprecate_orderr{ Úu8r) r Úi8Z__nan__Úfoor6 c C sL t |¡\}}|ddg }tjdddgtjd}t ||¡ t ||¡ d S )Nr rC r) )r r- r1 rI r9 r: r; )r= r{ rM rN r@ rA r" r" r' Ú,test_parametrized_factorize_na_value_default s
|
|
z:TestFactorize.test_parametrized_factorize_na_value_defaultzdata, na_valuerP Ú rQ r" ©rP rC )rP rD c C sR t j||d\}}|ddg }tjddddgtjd}t ||¡ t ||¡ d S )N)Úna_valuerC rU rT r r) )r Zfactorize_arrayr1 rI r9 r: r; )r= r{ r° rM rN r@ rA r" r" r' Ú$test_parametrized_factorize_na_value s
|
z2TestFactorize.test_parametrized_factorize_na_valuez data, uniquesNZInt64Z numpy_arrayZextension_array)ru c C s t j||dd\}}|r<tjddddgtjd}t |¡}n tjddddgtjd}|}t ||¡ t|tj r~t ||¡ n t
|
||¡ d S )NT)r rF rC r rT r) ) r r- r1 rI r9 Z safe_sortr: r; r. ÚndarrayÚ assert_extension_array_equal)r= r r{ rN rM rA r@ r" r" r' Ú test_factorize_use_na_sentinel¢ s z,TestFactorize.test_factorize_use_na_sentinelr9 c C s6 t j|dd\}}tj||dd tj||dd d S ©NFrE T©Z
|
strict_nan©r r- r: r; ©r= r{ rA r@ rM rN r" r" r' Ú+test_object_factorize_use_na_sentinel_false¿ s z9TestFactorize.test_object_factorize_use_na_sentinel_falseri c C s6 t j|dd\}}tj||dd tj||dd d S rµ r· r¸ r" r" r' Ú(test_int_factorize_use_na_sentinel_falseÖ s z6TestFactorize.test_int_factorize_use_na_sentinel_falseÚcategory)Ú
|
categoriesr* Z2017ú
|
US/Eastern©Útzc C s* t |¡\}}t ||¡ t ||¡ d S ©N)r r- r: r; r< r¸ r" r" r' Útest_factorize_mixed_valuesí s z)TestFactorize.test_factorize_mixed_values)-Ú__name__Ú
|
__module__Ú __qualname__r ÚmarkÚ parametrizerB rO r] rb rh rw r} r~ r r r r r r r r r r¥ r¦ r© r1 rI r r6 r± rq rJ r´ r* r¹ r rº r r r9 r r r rÁ r" r" r" r' r 4 sà
|
(0
|
|
ý
|
ý õþ
|
|
|
|
|
|
|
|
ýþ
|
ùþ
|
þþû
|
ô
|
ý ýúþ
|
|
ý ýúþ
|
ýý ýõþr c @ sü e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d d ejd d ged!fd"ejd#ged!fg¡d$d% Zd&d' Zd(d) Zd*d+ Zej d,ejeg¡d-d. Z d/d0 Z d1d2 Z d3S )4Ú
|
TestUniquec C s0 t jjdddd}t |¡}t|t js,td S )Nr éd é2 ©Úsize)r1 ÚrandomÚrandintr r0 r. r² rp ©r= r\ r| r" r" r' Ú test_ints s
|
zTestUnique.test_intsc C s6 t jjdddd d¡}t |¡}t|t js2td S )Nr rÈ rÉ rÊ ri ) r1 rÌ rÍ r3 r r0 r. r² rp rÎ r" r" r' Ú test_objects s
|
zTestUnique.test_objectsc C s. dddddg}t dD ]}tt |¡ qd S )Nr^ r_ ÚCÚDÚEéè )r rn r r0 )r= Úlstr
r" r" r' Útest_object_refcount_bug s z#TestUnique.test_object_refcount_bugc C sb t t d¡ d¡t t d¡d¡g¡}|j}| ¡ | d¡}t |¡}| ¡ t
|
||¡ d S )NrS rD ) r Z from_arraysr1 rZ ÚrepeatÚtilerK r rq r0 r: Úassert_almost_equal)r= Zmindexrv r| r" r" r' Útest_on_index_object s ÿ
|
|
z TestUnique.test_on_index_objectc C sP |t jt j kr$dddg}ddg}n¼t|r@dddg}ddg}n t|r\dddg}ddg}nt|rtddtddtddg}tddtddg}nJt|r²dddg}ddg}n.t|rÎdd d g}dd g}ndddg}ddg}t ||d
|
|
¡ }t j ||d
|
}|t jkr| t¡}|jjd kr@t|ttfs6tt |¡}t ||¡ d S ) NrC rD rG rH r TFr^ r_ r) )Úmrd )r: Z BYTES_DTYPESZ STRING_DTYPESr r r r r r
|
r r0 r1 rI r3 r6 r* Úkindr. r r rp r; )r= Zany_numpy_dtyper{ rN r| rv r" r" r' Útest_dtype_preservation( s8
|
|
|
|
|
|
|
|
|
|
|
|
|
z"TestUnique.test_dtype_preservationc C s¤ t jddgdd}tdddg}t |¡}t ||¡ |j|jksFtt |}t |¡}t ||¡ |j|jkstt|j
|
}t |¡}t ||¡ |j|jks td S )Nz 2015-01-03T00:00:00.000000000z 2015-01-01T00:00:00.000000000r r) ) r1 rI r r r0 r: r; r* rp r rK )r= rv Zdt_indexr| Úsr\ r" r" r' Ú$test_datetime64_dtype_array_returnedM s. þûýÿ
|
|
|
z/TestUnique.test_datetime64_dtype_array_returnedc C s@ t jdddgdd}t |¡}t jddgdd}t ||¡ d S )NÚ2000Ú2001z datetime64[s]r) ©r1 rI rq r0 r: r; ©r= rP r| rv r" r" r' Útest_datetime_non_nsl s
|
z TestUnique.test_datetime_non_nsc C s@ t jdddgdd}t |¡}t jddgdd}t ||¡ d S )Nrà rá ztimedelta64[s]r) iÐ iÑ râ rã r" r" r' Útest_timedelta_non_nsr s
|
z TestUnique.test_timedelta_non_nsc C sª t jdddgdd}tdddddg}t |¡}t ||¡ |j|jksLtt |}t |¡}t ||¡ |j|jkszt|j
|
}t |¡}t ||¡ |j|jks¦td S )Niày in² é' úm8[ns]r) ) r1 rI r r r0 r: r; r* rp r rK )r= rv Ztd_indexr| rÞ r\ r" r" r' Ú%test_timedelta64_dtype_array_returnedx s
|
|
|
z0TestUnique.test_timedelta64_dtype_array_returnedc C sB t ddddgtjd}tjdddgtjd}t t |¡|¡ d S )NrC rD r§ r) )r r1 r rI r: r; r r0 )r= rÞ r[ r" r" r' Útest_uint64_overflow s z TestUnique.test_uint64_overflowc C s> dt jddg}t |¡}t jdt jdgtd}t ||¡ d S )NrP rR r) )r1 rJ rq r0 rI r6 r: r; )r= Zduplicated_itemsr| rv r" r" r' Útest_nan_in_object_array s
|
z#TestUnique.test_nan_in_object_arrayc C s t td}t tdtddd}t td}| ¡ }t ||¡ t |¡}t ||¡ t tddd}| ¡ }t ||¡ t |¡}t ||¡ tt tddd}| ¡ }t ||¡ t |¡}t ||¡ tt tdtdd }t|}| ¡ }t ||¡ t |¡}t ||¡ d S )
|
NZbacÚabcT©r¼ ÚorderedZbaabc©rí r¬ ©Úname©r¼ )
|
r r7 r0 r: Úassert_categorical_equalr r rq r r< )r= rv Z
|
expected_orR r| rÞ Úcir" r" r' Útest_categorical s.
|
|
|
|
zTestUnique.test_categoricalc C s t ttdddtdddg ¡ }t t tdddg¡¡}t ||¡ ttdddtdddg ¡ }t
|
dgdd d}t ||¡ t t ttdddtdddg¡}t t tdddg¡¡}t ||¡ t ttdddtdddg¡}t
|
dgdd d}t ||¡ d S ) NÚ20160101r½ r¾ z2016-01-01 00:00:00-0500ú2016-01-01 00:00:00údatetime64[ns, US/Eastern]©r* rf ú
|
2016-01-01) r r r r0 r Z_from_sequencer1 rI r: r³ r r< rq ©r= r| rv r" r" r' Útest_datetime64tz_aware¿ s`
|
|
þÿÿÿ
|
|
þÿ ÿ
|
|
þÿÿÿ
|
ÿ
|
|
þÿÿ ÿz"TestUnique.test_datetime64tz_awarec C s: t tddddg¡}t |tjdddgdd¡ t tdgdgd ¡}t |tjddgdd¡ t ttdtdg¡}tjdgd d}t ||¡ t ttdd
|
d tdd
|
d g¡}t d gd d d}t
|
||¡ t t d¡}tjdddgt d}t ||¡ t tt t d¡}t t d}t ||¡ d S )NrD rC rU rx r) rS rõ z 2016-01-01T00:00:00.000000000r r½ r¾ rö r÷ rø ZaabcrP rQ rR rë )rq r0 r r: r; r1 rI r r r r< r7 r6 r rò rú r" r" r' Útest_order_of_appearanceö s4
|
|
þÿÿ ÿ z#TestUnique.test_order_of_appearancez arg ,expected)Ú1rý Ú2rý rþ r) )r¬ r¬ c C s t |¡}t ||¡ d S rÀ ©rq r0 r: r; )r= Úargrv r| r" r" r' Útest_tuple_with_strings s
|
z"TestUnique.test_tuple_with_stringsc C sB t jdd gtd}t |¡}t jdd gtd}tj||dd d S )Nr¬ r) Tr¶ )r1 rI r6 rq r0 r: r; ©r= r\ r| rv r" r" r' Útest_obj_none_preservation% s
|
z%TestUnique.test_obj_none_preservationc C s4 t ddg¡}t |¡}t dg¡}t ||¡ d S )Ng rY râ rã r" r" r' Útest_signed_zero- s
|
zTestUnique.test_signed_zeroc C s~ t dt dd¡¡d }t dt dd¡¡d }||ks<t||ksHtt ||g¡}t |¡}t tjg¡}t
|
||¡ d S )NÚdú=Qì À r ì À ) ÚstructÚunpackÚpackrp r1 rI rq r0 rJ r: r; )r= ÚNAN1ÚNAN2rP r| rv r" r" r' Útest_different_nans4 s
|
z TestUnique.test_different_nansÚel_typec C s¦ d}d}t dt d|¡¡d }t dt d|¡¡d }||ksDt||ksPttj||g|d}t |¡}|jdksztt dt d|d ¡¡d }||ks¢td S )Nl À r r r r r) rC ) r r
|
r rp r1 rI rq r0 rË ) r= r Z bits_for_nan1Z bits_for_nan2r r rP r| Zresult_nan_bitsr" r" r' Útest_first_nan_kept@ s
|
z TestUnique.test_first_nan_keptc C sZ ||kr d S t j||gtd}t |¡}|jdks6t|d |ksFt|d |ksVtd S )Nr) rD r rC )r1 rI r6 rq r0 rË rp )r= Zunique_nulls_fixtureZunique_nulls_fixture2rP r| r" r" r' Ú test_do_not_mangle_na_valuesQ s
|
z'TestUnique.test_do_not_mangle_na_valuesc C sH t dtjdgd |d}t |¡}tjdtjdg|d}t ||¡ d S )NrC rD rU r) )r rq ZNAr0 rI r: r³ )r= Zany_numeric_ea_dtyperL r| rv r" r" r' Útest_unique_masked[ s
|
z TestUnique.test_unique_maskedN) r rà rÄ rÏ rÐ rÖ rÚ rÝ rß rä rå rè ré rê rô rû rü r rÅ rÆ r1 rI r6 r r r r r r r r r" r" r" r' rÇ s8 % )7#þþ
|
|
|
rÇ c C s< | t j ddd¡}t |¡}tt |¡}||ks8td S )Nr rj é )r1 rÌ rÍ r Z nunique_intsrn r0 rp )Zindex_or_series_or_arrayrK r| rv r" r" r' Útest_nunique_intsc s
|
r c @ sö e Zd Zdd Zdd Zdd Zej ddd d
|
d g¡ej d d ddg¡dd Z dd Z
|
dd Z dd Z dd Z dd Zd d Zd d Zd d! Zd"d# Zej d$g eed%e g ¡g¡d&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4S )5ÚTestIsinc C s~ d}t jt|d t dd¡ W 5 Q R X t jt|d t ddg¡ W 5 Q R X t jt|d t dgd¡ W 5 Q R X d S )NzQonly list-like objects are allowed to be passed to isin\(\), you passed a \[int\]r rC )r r r r Úisin)r= r r" r" r' Ú test_invalidl s ÿzTestIsin.test_invalidc C s² t ddgdg¡}t ddg¡}t ||¡ t t ddg¡dg¡}t ddg¡}t ||¡ t tddgdg¡}t ddg¡}t ||¡ t tddgtdg¡}t ddg¡}t ||¡ t tddgdh¡}t ddg¡}t ||¡ t ddgdg¡}t ddg¡}t ||¡ t tddgtdg¡}t ddg¡}t ||¡ t tddgdh¡}t ddg¡}t ||¡ t ddgdg¡}t ddg¡}t ||¡ d S )NrC rD TFrP rQ )r r r1 rI r: r; r rú r" r" r' r] x s6 zTestIsin.test_basicc C sD t dddj}t ||d g¡}t dddg¡}t ||¡ t ||dd
¡}t dddg¡}t ||¡ t |t|dd
¡}t dddg¡}t ||¡ t dddj}t ||d g¡}t dddg¡}t ||¡ t ||dd
¡}t dddg¡}t ||¡ t |t|dd
¡}t dddg¡}t ||¡ d S ) Nrc rU ©Úperiodsr TFrD rg )
|
r rK r r r1 rI r: r; ro r r r" r" r' Útest_i8 s( zTestIsin.test_i8Údtype1rç r ú M8[ns, UTC]ú period[D]r* r« Zf8rª c C s t dddj}|dkr"| d¡}n*|dkr6| d¡}nt| d¡ |¡j}| d¡ |¡}t ||¡}t j
|
|j t d }t ||¡ d S )
|
Nz
|
2013-01-01rU r r rÒ r ÚUTCr« r) )r Ú_valuesZ to_periodZ tz_localizer Úviewr3 r r r1 ÚzerosÚshaper5 r: r; )r= r* r Údtar\ Úcompsr| rv r" r" r' Ú+test_isin_datetimelike_values_numeric_comps¸ s z4TestIsin.test_isin_datetimelike_values_numeric_compsc C sV t ddddj}t ||dd
¡}tjt|td}d|d< d|d < t ||¡ d S )
|
NZ20000101i rÞ )r rf r rD r) TrC )
|
r rK r r r1 r! rn r5 r: r; ©r= rÞ r| rv r" r" r' Ú
|
test_largeÍ s zTestIsin.test_largec C s t ddddg¡}dddg}ttdg ||¡}ttdg t ddg¡|¡}t ddddg¡}t ||¡}t ||¡ d S © Nr rC rD rP rQ rR TF) r1 rI r r Ú
|
from_codesr r r: r; )r= ÚvalsÚcatsZSdZStrv r| r" r" r' Útest_categorical_from_codesÕ s
|
z$TestIsin.test_categorical_from_codesc C sx t ddddg¡}dddg}tdg ||¡}tdg t ddg¡|¡}t ddddg¡}t ||¡}t ||¡ d S r( )r1 rI r r) r r r: r; )r= r* r+ ÚcatÚotherrv r| r" r" r' Útest_categorical_isinß s
|
z TestIsin.test_categorical_isinc C s8 t jg}t jg}t dg¡}t ||¡}t ||¡ d S )NT)r1 rJ rI r r r: r; ©r= r$ rK rv r| r" r" r' Útest_same_nan_is_iné s
|
z TestIsin.test_same_nan_is_inc C sJ t dd¡}t j|d< t |t jdg¡}t jt|td}t ||¡ d S ©NrG iAB r rC r) )
|
r1 rØ rJ r r Úonesrn r5 r: r; r& r" r" r' Útest_same_nan_is_in_largeõ s
|
|
z"TestIsin.test_same_nan_is_in_largec C sT t dd¡}t|}t j|d< | t jdg¡}tt jt|td}t ||¡ d S r2 )
|
r1 rØ r rJ r r3 rn r5 r: Úassert_series_equal)r= rÞ Zseriesr| rv r" r" r' Ú test_same_nan_is_in_large_seriesý s
|
z)TestIsin.test_same_nan_is_in_large_seriesc C s` G dd d}| | }}t t |g|g¡t dg¡¡ t t |g|g¡t dg¡¡ d S )Nc @ s" e Zd ZedddZdd ZdS )z0TestIsin.test_same_object_is_in.<locals>.LikeNan)Úreturnc S s dS )NFr" )r= r. r" r" r' Ú__eq__ s z7TestIsin.test_same_object_is_in.<locals>.LikeNan.__eq__c S s dS )Nr r" )r= r" r" r' Ú__hash__ s z9TestIsin.test_same_object_is_in.<locals>.LikeNan.__hash__N)r rà rÄ r5 r8 r9 r" r" r" r' ÚLikeNan s r: TF)r: r; r r r1 rI )r= r: rP rQ r" r" r' Útest_same_object_is_in s z TestIsin.test_same_object_is_inc C s¸ t dg}t dg}|d |d k s(tt ||¡}t t dg¡|¡ t tj|t dtj|t d¡}t t dg¡|¡ t tj|tj
|
dtj|tj
|
d¡}t t dg¡|¡ d S )NrJ r Tr) ) Úfloatrp r r r: r; r1 rI r8 r6 r )r= r$ rK r| r" r" r' r s
|
|
ÿ ÿz TestIsin.test_different_nansc C s8 ddg}dg}t ddg¡}t ||¡}t ||¡ d S )NÚssé* Z42F)r1 rI r r r: r; r0 r" r" r' Ú test_no_cast1 s
|
zTestIsin.test_no_castÚemptyr) c C s6 t ddg}t ddg¡}t ||¡}t ||¡ d S )NrP rQ F)r r1 rI r r r: r; )r= r@ r* rv r| r" r" r' Ú
|
test_empty: s zTestIsin.test_emptyc C s^ t jdt jd tdgtd}t jtdgtd}t dddg¡}t ||¡}t ||¡ d S )NrJ ù ð?r) FT) r1 rI rJ r< r6 r r r: r; )r= r$ r* rv r| r" r" r' Útest_different_nan_objectsC s
|
z#TestIsin.test_different_nan_objectsc C sÐ t dt dd¡¡d }t dt dd¡¡d }||ks<t||ksHttj||gtjd}tj|gtjd}t ||¡}t ddg¡}t
|
||¡ tj|gtjd}t ||¡}t ddg¡}t
|
||¡ d S )Nr r r r r r) T) r r
|
r rp r1 rI r r r r: r; )r= r r r\ Zlookup1r| rv Zlookup2r" r" r' Ú test_different_nans_as_float64K s z'TestIsin.test_different_nans_as_float64c C s< t dddgi}| dg¡}t dddgi}t ||¡ dS )zComparing df with int`s (1,2) with a string at isin() ("1")
|
-> should not match values because int 1 is not equal str 1rK rC rD rý FN©r r r: Úassert_frame_equal©r= Zdfr| Zexpected_falser" r" r' Ú test_isin_int_df_string_searcha s z'TestIsin.test_isin_int_df_string_searchc C s> t dtjdgi}| dg¡}t dddgi}t ||¡ dS )zComparing df with nan value (np.nan,2) with a string at isin() ("NaN")
|
-> should not match values because np.nan is not equal str NaNrK rD ÚNaNFN)r r1 rJ r r: rF rG r" r" r' Ú test_isin_nan_df_string_searchi s z'TestIsin.test_isin_nan_df_string_searchc C s< t dddgi}| dg¡}t dddgi}t ||¡ dS )zComparing df with floats (1.4245,2.32441) with a string at isin() ("1.4245")
|
-> should not match values because float 1.4245 is not equal str 1.4245rK gnÀÊö?gö#Ed@z1.4245FNrE rG r" r" r' Ú test_isin_float_df_string_searchq s z)TestIsin.test_isin_float_df_string_searchc C s4 t dgtjd}| dg¡}t d}t ||¡ d S )Nl ª
|
G r) l
|
G F)r r1 r r r: r5 ©r= rL r| rv r" r" r' Útest_isin_unsigned_dtypey s z!TestIsin.test_isin_unsigned_dtypeN) r rà rÄ r r] r r rÅ rÆ r% r' r, r/ r1 r4 r6 r; r r? r r6 r1 rI rA rC rD rH rJ rK rM r" r" r" r' r k s. %
|
|
|
r c @ sz e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j dejedf¡dd Zdd ZdS )ÚTestValueCountsc C s t j d¡ ddlm} t j d¡}||d}t |¡}ddddd g}t |¡
|
t d
|
d ¡}t d d d d g|d d}t | ¡ | ¡ ¡ d S )NiÒ r )ÚcutrV g´Èv¾ó¿g
ëQ¸ á¿g`åÐ"Ûù¾?gw¾/Ýè?g!°rhíö?Trî rC Úcount©r# rð )r1 rÌ ÚseedZpandas.core.reshape.tilerO Zrandnr Ú value_countsr Z from_breaksr3 ÚCDTr r: r5 Z
|
sort_index)r= rO r\ Zfactorr| Zbreaksr# rv r" r" r' Útest_value_counts s
|
|
z!TestValueCounts.test_value_countsc C sz ddddg}t j|dd}tdgt dg¡dd}t ||¡ t j|dd d
|
}tddgt d d g¡dd}t ||¡ d S ) NrC rD rU rV ©Úbins)ç¬ Zd;ßï?rX rP rQ F)rW r )rX ç @)rY rX )r rS r r r/ r: r5 r& r" r" r' Útest_value_counts_bins s
|
ÿ ýz&TestValueCounts.test_value_counts_binsc C s t ddg¡}t|dks tt jddgdd}t|dks@tt tdddg¡}t|dksdtd}tjt|d t jddgdd W 5 Q R X d S )NrC rG rV rý rD z*bins argument only works with numeric datar )r rS rn rp r r r r )r= r| r r" r" r' Útest_value_counts_dtypes s z(TestValueCounts.test_value_counts_dtypesc C s t t d¡tgdd}tddg}||fD ]<}t |¡}tj|dd}t|dksVtt|d ks*tq*t t d
|
did d }t
|
t |¡|¡ d S ) Nræ útimedelta64[ns]r) r z
|
2014-01-01F©ÚdropnarC rD z2014-01-01 00:00:00rP rï ) r r1 Ú timedelta64r r r rS rn rp r r: r5 )r= ÚtdÚdtrÞ ZvcZ
|
vc_with_naZexp_dtr" r" r' Útest_value_counts_nat® s
|
z%TestValueCounts.test_value_counts_natc
|
C sÎ t tdddtdddtdddtdddtdddtdddg}| ¡ }ttdddtdddtdddgtd}t dddg|dd }t ||¡ tt d
|
tj gd d }t d
|
tj gtd}t ||¡ d S ) Ni¸ rC i ip r) rU rD rP rQ z
|
2362-01-01Úignore)Úerrors)
|
r r rS r r6 r: r5 r r1 rJ )r= rÞ ÚresZ exp_indexr[ r" r" r' Ú&test_value_counts_datetime_outofbounds¼ s&
|
|
|
|
|
|
úÿ
|
þ z6TestValueCounts.test_value_counts_datetime_outofboundsc C sx t ttd}| ¡ }t dddgtdddgdd }tj||d
|
d |j ¡ }| ¡ }|j ¡ |_ tj||d
|
d d S ) NZaaabbcrU rD rC rP rQ rR rP rQ T©Zcheck_index_type)
|
r r r7 rS r r: r5 r- Z
|
as_orderedr# r& r" r" r' rô Ö s ÿ
|
z TestValueCounts.test_categoricalc C sV t ttd}tj|jd< | ¡ }t dddgtdddgdddgd d
|
d }tj ||d d |jdd}t ddddgtdddtjgd
|
d }tj ||d d t ttdd dddgd}tj|jd< | ¡ }t dddgtdddgdddgd dd
|
d }tj ||d d |jdd}t ddddgtdddtjgdddgd dd
|
d }tj ||d d d S )NZ
|
aaaaabbbccrC rV rU rD rP rQ rR rñ rP rQ Trg Fr] )rí r¼ rì )
|
r r r7 r1 rJ ZilocrS r r: r5 r& r" r" r' Útest_categorical_nanså sT ý
|
ÿÿ ýù
|
ÿûz%TestValueCounts.test_categorical_nansc C sb t ttdtddd}| ¡ }t ddddgtd d
|
d d gtdddd d}tj||dd d S )NZbbbaacZabcdTrì rU rD rC r rQ rP rR r rP rQ rg )r r r7 rS r: r5 r& r" r" r' Útest_categorical_zeroes s
|
|
ÿûz'TestValueCounts.test_categorical_zeroesc C sÐ t tdddgjddtddgddgdd¡ t tdddgjddtddgddgdd¡ t tdgd dgd d gd jddtddgtddgtd
|
dd¡ t tdgd dgd d gd jddtd ddgddtjgdd¡ t td d d gjddtddgd d gdd¡ t td d d gjddtddgd d gdd¡ t td d d d gjddtddgd d gdd¡ td d d d d d gjdd}tdddgd d tjgdd}t ||¡ d S ) NTFr] rD rC rP rQ rU rS r) g$@rW )r: r5 r rS r r6 r1 rJ rú r" r" r' Ú test_dropna s> þþ( þ(þþþþ zTestValueCounts.test_dropnar* r c C s¨ t dgd dgd tjgd }| |¡}|jddd}t dd d
|
gt tjd d g|d dd}t ||¡ |jddd}t ddgt d d g|d dd}t ||¡ d S )NrC rD rU rS TF)Ú normalizer^ g à?g333333Ó?gÉ?rH rG r) Z
|
proportionrQ g333333ã?gÙ?)r r1 rJ r3 rS r: r5 )r= r* rÞ Zs_typedr| rv r" r" r' Ú test_value_counts_normalizedC s $
|
ý ÿz,TestValueCounts.test_value_counts_normalizedc C s| t jdgt jd}tdgdgdd}t |¡}t ||¡ t jddgtd}tddgddgdd}t |¡}t ||¡ d S )Nr§ r) rC rP rQ rT ) r1 rI r r r rS r: r5 r6 )r= r\ rv r| r" r" r' Útest_value_counts_uint64V s
|
|
z(TestValueCounts.test_value_counts_uint64N)r rà rÄ rU rZ r[ rb rf rô rh ri rj r rÅ rÆ r1 r r6 rl rm r" r" r" r' rN s + &
|
rN c @ s e Zd Zdd Zej de dddddddddd g
|
¡e d
|
d d
|
ej d d d d
|
ej dg
|
¡e ddddddddddg
|
¡ejddddddddddg
|
e
|
dejdd dd d d d dd d g
|
ej dg¡d!d" Z d#d$ Z ej dedddge d%d¡g¡d&d' Zej d(d)d*d+d,d)d*d+d,gd)d*d+d,gfd-d.d.d-gd-d.gfd/d0d1d/gd/d0d1gfg¡d2d3 Zej d4dd%dd5d6d6gejdd7d8d5d6ge
|
dfg¡d9d: Zd;S )<ÚTestDuplicatedc C s¶ t jddt jddt jgtd}t |¡}t ddddddg¡}t ||¡ tj|dd}t ddddddg¡}t ||¡ tj|d d}t ddddddg¡}t ||¡ tj|dd}t ddddddg¡}t ||¡ t jd
|
td}t t
|
ddt jt jgd dt jdt jgd D ]\}}|||< qt |¡}dgd }dgd }t || ¡}t ||¡ tj|d d}t || ¡}t ||¡ tj|dd}t || ¡}t ||¡ d S ) Nr rC rD r) FTÚfirst©ZkeepÚlasté rV ) r1 rI rJ r6 r Ú
|
duplicatedr: r; r@ Ú enumerateÚzip)r= Úkeysr| rv r
ÚtZfalsesZtruesr" r" r' Útest_duplicated_with_nase s:
|
(ÿ
|
|
|
z'TestDuplicated.test_duplicated_with_nasÚcaserC rD rS rU rV é gñ?g@gffffff
|
@g@gffffff@y ð? ð?y @ @y @ @y @ @y @ @y @ @rP rQ ÚerR r Úfr) r§ éó r¡ é' é c
|
C sh t ddddddddddg
|
¡}t ddddddddddg
|
¡}||B }tj|dd}t ||¡ tj|dd}t ||¡ tj|dd}t ||¡ t|t|ddfD ]L}|jdd}t ||¡ |jdd}t ||¡ |jdd}t ||¡ q¦t|t|ddfD ]Z} | jdd}t |t|¡ | jdd}t |t|¡ | jdd}t |t|¡ qd S )NFTro rp rq r» r) ) r1 rI r rs r: r; r r r5 )
|
r= ry Ú exp_firstÚexp_lastÚ exp_falseÚ res_firstÚres_lastÚ res_falseÚidxrÞ r" r" r' Útest_numeric_object_likes s6 ÿÿ z(TestDuplicated.test_numeric_object_likesc C s ddddddddddg
|
}ddddd dd
|
ddd g
|
}t d d |D ¡t dd |D ¡t dd |D ¡t dd |D ¡t dd |D ¡g}t ddddddddddg
|
¡}t ddddddddddg
|
¡}||B }|D ]:}tj|dd}t ||¡ tj|dd} t | |¡ tj|dd}
|
t |
|
|¡ t|t|ddt|tdfD ]N} | jdd}t ||¡ | jdd} t | |¡ | jdd}
|
t |
|
|¡ qHt|t|ddt|tdfD ]Z} | jdd}t |t|¡ | jdd} t | t|¡ | jdd}
|
t |
|
t|¡ q¶qÖd S )Nz
|
2011-01-01z
|
2011-01-02r ú
|
2011-01-03z
|
2011-01-04z
|
2011-01-06ú1 daysz2 daysz3 daysz4 daysz6 daysc S s g | ] }t |qS r" ©r ©r$ r r" r" r' r( è s z6TestDuplicated.test_datetime_likes.<locals>.<listcomp>c S s g | ]}t |d dqS )r½ r¾ r r r" r" r' r( é s c S s g | ]}t |d dqS )rÒ re )r r r" r" r' r( ê s c S s g | ]}t |¡qS r" )r1 r r r" r" r' r( ë s c S s g | ] }t |qS r" )r r r" r" r' r( ì s FTro rp rq r» r) )
|
r1 rI r rs r: r; r r6 r r5 ) r= ra r` Zcasesr r r ry r r r
r rÞ r" r" r' Útest_datetime_likesÍ s| ö öûÿÿ
|
|
|
ý
|
|
ý z"TestDuplicated.test_datetime_likesr c C s. |j dkstt | ¡ t dddg¡¡ d S )NTF)Z is_uniquerp r: r; rs r1 rI )r= ry r" r" r' Útest_unique_index s z TestDuplicated.test_unique_indexz arr, uniquesr )r rC )rC r r )rQ rR )rP rQ r¯ )rQ rD )rP rU c C s8 t jt|td}||d d
< t |¡}t ||¡ d S r )r1 r@ rn r6 rq r0 r: r; )r= r\ rN rv r| r" r" r' Útest_unique_tuples$ s
|
z!TestDuplicated.test_unique_tupleszarray,expectedrB y ð? @y y ð? c C s t |¡}t ||¡ d S rÀ rÿ )r= rI rv r| r" r" r' Útest_unique_complex_numbers: s
|
z*TestDuplicated.test_unique_complex_numbersN)r rà rÄ rx r rÅ rÆ r1 rI rJ r6 r r r r rq r£ r r r r" r" r" r' rn d sb ' öÿ ÿîþ
|
(R"
|
|
þ
|
þ÷þ
|
ýÿþ
|
rn c @ s e Zd Zej dejdfejdfej dfej
|
dfej dfg¡dd Z ej dejdfejdfej dfej
|
dfej dfg¡dd Z d
|
S ) Ú TestHashTablezhtable, tm_dtypeÚStringÚFloatZIntZUIntc C sÞ t td| d }t|d}|tjkr6tj|jd< n |tjkrVtjd t g|jdd
< |j
|
dddj dd }|j j |d
|
|jd d j }| |j ¡}t ||¡ | j|j dd \}} t ||¡ || }
|
t |
|
|j ¡ d S )NÚmaker rÔ éô éö rU T©ÚfracÚreplace©Zdropr ro rp )Zreturn_inverse)Úgetattrr: r rl ÚFloat64HashTabler1 rJ ÚlocÚPyObjectHashTabler ÚsampleÚ reset_indexrK r Údrop_duplicatesr0 r; ) r= ÚhtableÚtm_dtyper ÚmakerrÞ Ú s_duplicatedÚexpected_uniqueÚ result_uniqueÚresult_inverseZreconstrr" r" r' Útest_hashtable_uniqueK s$
|
|
ÿ
|
z#TestHashTable.test_hashtable_uniquec C sØ t td| d }t|d}|tjkr6tj|jd< n |tjkrVtjd t g|jdd
< |j
|
dddj dd }|j j |d
|
| ¡ j }| |j ¡\}} | ¡ ¡ j }
|
t ||
|
¡ || | } | ¡ j } t | | ¡ d S ) Nr r rÔ r r rU Tr r r )r r: r rl r r1 rJ r r r r r rK r rr r- r^ r r; ) r= r¡ r¢ r r£ rÞ r¤ Zna_maskr¦ r§ r¥ Zresult_reconstructZexpected_reconstructr" r" r' Útest_hashtable_factorizes s
|
|
|
|
z&TestHashTable.test_hashtable_factorizeN)r rà rÄ r rÅ rÆ rl r ZStringHashTabler ZInt64HashTableZUInt64HashTabler¨ r© r" r" r" r' r J s( ûþ
|
|
ûþ
|
r c @ s´ e Zd Zejej dej ej dddej dddej g
|
dej dddej dddej g
|
g¡dd Z
|
ej d ej d
|
¡d d Z ej d ej ejg¡d d Zdd Zejjdd ZdS )ÚTestRankr\ rW rC rD rU rX c C s` ddl m} t |¡}t |¡ }| ¡ }t |¡}tj||< ||}tj ||< t
|
||¡ d S )Nr )Úrankdata) Z scipy.statsr« r1 rI ÚisfiniteÚcopyÚlibalgosZrank_1dr` rJ r: rÙ )r= r\ r« ry r| r[ r" r" r' Útest_scipy_compat s
|
|
|
|
zTestRank.test_scipy_compatr* Ú
|
AllIntegerc C sT t jddgt jd}t jddg|d}|j|d t|}t |¡}t ||¡ d S )NrC rD r) rÈ r ) r1 rI r r r r Úrankr: r; )r= r r* r[ r{ rL r| r" r" r' r] ² s
|
zTestRank.test_basicc C s: t jddgt jd}tddg|d}t t |¡|¡ d S )NrC rD r) r§ )r1 rI r r r: r; r r± )r= r* r[ rÞ r" r" r' ré ¼ s z TestRank.test_uint64_overflowc C sP t dddgdddgddd ggg¡}d
|
}tjt|d t |¡ W 5 Q R X d S ) NrC rD rU rV rS rz r rr é z%Array with ndim > 2 are not supportedr )r1 rI r r r r r± )r= r\ r r" r" r' Útest_too_many_ndimsà s $z TestRank.test_too_many_ndimsc C s\ t d¡}tj|dd ¡ }|dks(tt d¡ dd¡}tj|dd ¡ }|dksXtd S )Ni T)ZpctrC i rD )r1 rZ r r± Úmaxrp Úreshape)r= rK r| r" r" r' Útest_pct_max_many_rowsÊ s
|
z TestRank.test_pct_max_many_rowsN)r rà rÄ r` Zskip_if_no_scipyr rÅ rÆ r1 rJ r¯ Ú typecodesr] r r ré r³ Z
|
single_cpur¶ r" r" r" r' rª s þþ
|
|
rª c C sÞ t jg dd} t jtdddgdd}tjd | |}t jdgt jd}t ||¡ tjd || }t jg t jd}t ||¡ tjd | |}t jdgt jd}t ||¡ tjd || }t jg t jd}t ||¡ d S )Nri r) iÚ é é r6 rT ) r1 rI r r® Úpadr9 r: r; Úbackfill)ÚoldÚnewr| rv r" r" r' Ú!test_pad_backfill_object_segfaultÖ s r¾ c @ s e Zd Zdd Zdd ZdS )ÚTestTseriesUtilc C s t dddg}t ttd}tjd |j|j}tjddddddddddddg tjd }t
|
||¡ t dd
|
g}t ttdd}tjd |j|j}tjdddddgtjd }t
|
||¡ d S ) NrC rS r¡ r¸ Úint64_tr rD rT r) rV ) r r7 r r® r» rK r1 rI r9 r: r; ©r= r¼ r½ ZfillerZ expect_fillerr" r" r' Ú test_backfillì s ( z TestTseriesUtil.test_backfillc C sÄ t dddg}t ttd}tjd |j|j}tjddddddddddddg tjd }t
|
||¡ t ddg}t tj dtj d }tjd |j|j}tjdddddgtjd }t
|
||¡ d S )
|
NrC rS r¡ r¸ rÀ rT r rD r) ) r r7 r r® rº rK r1 rI r9 r: r; rZ rx rÁ r" r" r' Útest_padý s ( zTestTseriesUtil.test_padN)r rà rÄ r rà r" r" r" r' r¿ ë s r¿ c ~ C s t jddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddg|ddt jddd d
|
d d d ddddddddddddddd d d d d d!ddddddd d
|
d d d ddddddddddddddd d d d d d!ddddddd d
|
d d d ddddddddddddddd d d d d d!ddddddd d
|
d d d ddddddddddddddd d d d d d!ddddg|ddg} t | ¡r td S )"NrU rD rC r rx r) r é é é é é é é é é rj é é r é é é é r¸ é r¡ r² rr r rz rS rV )r1 rI r® Z is_lexsortedrp )Zfailurer" r" r' Útest_is_lexsorted s ~ ~ þ rÔ c C s¬ t j ddd¡ t j¡} t j ddd¡ t j¡}t | d¡d }t j| dd}| t j¡}t ||¡ | d | }t |d¡d }t
|
|| f¡}| t j¡}t ||¡ d S )Nr rÔ rÈ Z mergesort)rÜ i@B ) r1 rÌ rÍ r3 r9 r® Zgroupsort_indexerÚargsortr: r; Zlexsort)rP rQ r| rv rs r" r" r' Útest_groupsort_indexer s rÖ c s t ¡ t ¡ tddddtd g} t fdd| D sDtt fdd| D s^t krn ksrt k s krtt ¡ t ¡ kstt ¡ t ¡ kr®ttfd d| D sÈttfd
|
d| D sâtkròksötk s
|
krtt ¡ t ¡ ks$tt ¡ t ¡ kr:tt| D ]}t|| ksBtqBt t ¡ gd ¡
|
¡ t t ¡ gd ¡
|
¡ d S ) Nz-infg}Ã%I²Ôr g}Ã%I²Tr` c 3 s | ]} |kV qd S rÀ r" ©r$ ra ©ÚInfr" r' Ú <genexpr>; s z%test_infinity_sort.<locals>.<genexpr>c 3 s | ]} |kp| kV qd S rÀ r" r× rØ r" r' rÚ < s c 3 s | ]} |kV qd S rÀ r" r× ©ÚNegInfr" r' rÚ B s c 3 s | ]} |k p| kV qd S rÀ r" r× rÛ r" r' rÚ C s é ) r® ÚInfinityÚ NegInfinityr< Úallrp r Úsortedr1 rI rÕ )Zref_numsÚpermr" ©rÙ rÜ r' Útest_infinity_sort0 s& rä c C s¼ t ¡ } t ¡ }| tjkr t| tjkr,t| tjk r:t| tjkrHt| tjkrVt| tjksdt|tjkrrt|tjkrt|tjk rt|tjkrt|tjkrªt|tjks¸td S rÀ )r® rÞ rß r1 rJ rp rã r" r" r' Útest_infinity_against_nanQ s rå c C s* t jdt jd} t | ¡}|| ks&td S )NrÈ r) )r1 rZ r9 r® Zensure_platform_intrp )r\ r| r" r" r' Útest_ensure_platform_intd s
|
ræ c C sè d} t t j¡j}t t j¡j}tjt| d t t
|
||g¡|¡ W 5 Q R X tjt| d& t t
|
||g¡t
|
||g¡¡ W 5 Q R X tjt| d t t
|
||g¡|¡ W 5 Q R X tjt| d& t t
|
||g¡t
|
||g¡¡ W 5 Q R X tjt| d& t t
|
||g¡t
|
||g¡¡ W 5 Q R X tjt| d4 tj t
|
||g¡t
|
||g¡t
|
ddg¡d W 5 Q R X tjt| d4 tj t
|
||g¡t
|
||g¡t
|
ddg¡d W 5 Q R X tjt| d@ tj t
|
||g¡t
|
||g¡t
|
ddg¡t
|
ddg¡d W 5 Q R X tjt| d( t t
|
||g¡t
|
t j |g¡¡ W 5 Q R X tj t
|
||g¡t
|
||g¡t
|
ddg¡d tj t
|
||g¡t
|
||g¡t
|
ddg¡d tj t
|
||g¡t
|
||g¡t
|
ddg¡t
|
ddg¡d d S )NzOverflow in int64 additionr FT)Úarr_mask)Úb_mask)rç rè ) r1 Ziinforx r´ Úminr r Ú OverflowErrorr Zchecked_add_with_arrrI rJ )r rÛ Únr" r" r' Útest_int64_add_overflowk sd * ** ÿ ÿ ü, ÿ ÿ ürì c @ s e Zd Zdd Zej dejd ejd ¡dd Z dd Z
|
ej dejd ejd ¡d
|
d Z d d Z ej de eg¡dd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd S ) ÚTestModec C s0 t g tjtg tdd}t t g ¡|j ¡ d S )Nr) )r* r# )
|
r r1 r r Úintr: r; r ÚmoderK ©r= r[ r" r" r' Ú test_no_mode s zTestMode.test_no_modera r° r c C s dg}dg}dg}ddg}t ||d}t ||d}t t |j¡|j¡ t | ¡ |¡ t ||d}t ||d}t t |j¡|j¡ t | ¡ |¡ d S )NrC r) ©r r: r; r rï rK r5 ©r= ra Z
|
exp_singleZ data_singleZ exp_multiZ
|
data_multirL r[ r" r" r' Útest_mode_single£ s zTestMode.test_mode_singlec C sT t dgtd}t t dg¡|j¡ t dddgtd}t t dddg¡|j¡ d S )NrC r) rP rQ rR )r rî r: r; r rï rK r6 rð r" r" r' Útest_mode_obj_int¶ s zTestMode.test_mode_obj_intc C sÀ dg}dgd dgd }ddg}dgd dgd dgd }t ||d}t ||d}t t |j¡|j¡ t | ¡ |¡ t ||d}t ||d}t t |j¡|j¡ t | ¡ |¡ d S )NrC rS rD rU r) rò ró r" r" r' Útest_number_mode½ s zTestMode.test_number_modec C s\ dg}dgd dgd }t |dd}t |dd}t t |j¡|j¡ t | ¡ |¡ d S )NrQ rP rD rU rR r) rò )r= r[ r{ rL r" r" r' Útest_strobj_modeÏ s zTestMode.test_strobj_modec C s\ dg}dgd dgd }t ||d}t ||d}t t |j¡|j¡ t | ¡ |¡ d S )NÚbarr¬ rD rU r) rò )r= ra r[ r{ rL r" r" r' Útest_strobj_multi_charØ s z TestMode.test_strobj_multi_charc C s t dddgdd}t dddgdd}t t |j¡|j¡ t | ¡ |¡ t ddgdd}t dddddgdd}t t |j¡|j¡ t | ¡ |¡ d S )Nz
|
1900-05-03r z
|
2013-01-02r r) ©r r: r³ r rï rK r r5 ©r= r[ rL r" r" r' Útest_datelike_modeâ s þzTestMode.test_datelike_modec C s t dddgdd}t dddgdd}t t |j¡|j¡ t | ¡ |¡ t ddgdd}t dddd ddgdd}t t |j¡|j¡ t | ¡ |¡ d S )
|
Nz-1 daysz0 daysr r\ r) ú2 minrg ú-1 dayú -1 day 2 minrú rû r" r" r' Útest_timedelta_modeð s þz TestMode.test_timedelta_modec C sB t dg}t dddg}t t |j¡|j¡ t | ¡ |¡ d S )Nr¬ rC rò rû r" r" r' Útest_mixed_dtypeþ s
|
zTestMode.test_mixed_dtypec C s t dgtjd}t dddgtjd}t t |j¡|j¡ t | ¡ |¡ t ddgtjd}t ddgtjd}t t |j¡|j¡ t | ¡ |¡ d S )Nr§ r) rC ) r r1 r r: r; r rï rK r5 rû r" r" r' ré s z TestMode.test_uint64_overflowc C sª t ddg}|}t| ¡ j}t ||¡ t dddg}t dgddgd}t| ¡ j}t ||¡ t dddddg}t ddgdddgd}t| ¡ j}t ||¡ d S )NrC rD rP rñ rU )r r rï r r: rò )r= rR r[ re r" r" r' rô s zTestMode.test_categoricalc C sÞ t dddg}tdddgtjd}t t |¡|j¡ t dddg}tdgt d}t t |¡|j¡ t dddddg}tddgtjd}t t |¡|j¡ t ddddd d gd
|
d}t
|
j t d d t |¡ W 5 Q R X d S ) NrC rD rU r) rP rg rþ rÿ rý r\ ZTimedeltaIndexr ) r r r1 rx r: r; r rï rK r6 r r ÚAttributeError)r= r r[ r" r" r' Ú
|
test_index s þzTestMode.test_indexc C s8 t dddgdd}| ¡ }t dgdd}t ||¡ d S )NrC rU r¬ rï )r rï r: r5 rL r" r" r' Útest_ser_mode_with_name4 s z TestMode.test_ser_mode_with_nameN)r rà rÄ rñ r rÅ rÆ r1 r· rô rõ rö r÷ Ústrr6 rù rü r r ré rô r r r" r" r" r' rí s
|
|
|
rí c @ sH e Zd Zej dddg¡dd Zdd Zej ddd g¡d
|
d Zd S ) ÚTestDiffr* r rç c C s¾ t d¡ t j¡ |¡ dd¡}|j dd¡|d d
df< tj |ddd }t j
|
|j d
|
d d }t dd¡|d d
df< t dd¡|dd d
f< t ||¡ tj |jddd }t ||j¡ d S ) Nr¸ rU rV r ÚnsrD rC r ©Zaxisr\ r) )r1 rZ r3 rx r rµ r* Útyper Údiffr3 r" r_ r: r; ÚT©r= r* r\ r| rv r" r" r' Útest_diff_datetimelike_nat= s z#TestDiff.test_diff_datetimelike_natc C sB t ddddj}d}tjt|d tj|ddd W 5 Q R X d S ) Nrù rU z
|
US/Pacific)r r¿ z#cannot diff DatetimeArray on axis=1r rC r )r Ú_datar r Ú
|
ValueErrorr r
|
)r= r# r r" r" r' Útest_diff_ea_axisM s zTestDiff.test_diff_ea_axisZint8Úint16c C sN t jdddddg|d}t |d¡}t jt jddddgdd}t ||¡ d S )Nr rC r) rT r4 )r1 rI r r
|
rJ r: r; r r" r" r' Útest_diff_low_precision_intT s z$TestDiff.test_diff_low_precision_intN) r rà rÄ r rÅ rÆ r r r r" r" r" r' r < s
|
|
r Úopc C sv | ddddg}| ddddg}| ddddddg}t |tjrZt ||¡}t ||¡ nt ||¡}t ||¡ d S )NrU rC rV rD )r. r1 r² r Zunion_with_duplicatesr: r; r³ )r ZlvalsZrvalsrv r| r" r" r' Útest_union_with_duplicates\ s r )Nr Ú itertoolsr r Únumpyr1 r Z pandas._libsr r® r rl Z pandas.util._test_decoratorsÚutilZ_test_decoratorsr` Zpandas.core.dtypes.commonr r r r r
|
Zpandas.core.dtypes.dtypesr rT Zpandasrq r r r r r r r r r r r r r r r r r Zpandas._testingZ_testingr: Zpandas.core.algorithmsÚcoreZ
|
algorithmsZpandas.core.arraysr r Zpandas.core.commonÚcommonr r rÇ r r rN rn r rª r¾ r¿ rÔ rÖ rä rå ræ rì rí r rÅ rÆ rI r r" r" r" r' Ú<module> s\ L W ] d gR:# !3
|
