U
|
®ý°d! ã @ s d Z ddlmZ ddlZddlZddlZddlmZ ddl m
|
Z ddddd d
|
d d d ddddg Z ej dd Zej e ddd Zdd ZG dd dZdS )z
|
Note: for naming purposes, most tests are title with as e.g. "test_nlargest_foo"
|
but are implicitly also testing nsmallest_foo.
|
é )ÚproductN)ÚSeriesÚdatetimeÚ
|
datetimetzÚ timedeltaÚint8Úint16Úint32Úint64Úfloat32Úfloat64Úuint8Úuint16Úuint32Úuint64c C sp t t dddddg¡t dddddg¡ d¡t ddddd g¡d
|
¡} d D ] }td d dd dg|d| |< qL| S )z¯
|
A DataFrame with many dtypes
|
|
* datetime
|
* datetimetz
|
* timedelta
|
* [u]int{8,16,32,64}
|
* float{32,64}
|
|
The columns are the name of the dtype.
|
Z2003Z2002Z2001Z2005z
|
US/EasternZ3dZ2dZ1dZ5d)r r r )
|
r r r r
|
r r r r r r é é é é ©Údtype)ÚpdZ DataFrameÚ to_datetimeZ tz_localizeZ to_timedeltar )Zdfr © r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_nlargest.pyÚ s_main_dtypes s ÿþûÿ
|
r )Úparamsc C s
|
|| j S )z Each series in s_main_dtypes.)Úparam)Úrequestr r r r Ús_main_dtypes_splitG s r c C sP t | |d}t||d}|dkr,dddgndddg}|j| }t ||¡ d S )Nr r Ú nsmallestr r r )r ÚgetattrÚlocÚtmÚassert_series_equal)Úvalsr ÚmethodÚserÚresultZ expected_idxrÚexpectedr r r Ú assert_check_nselect_boundaryM s
|
|
r* c @ s e Zd Zej dedddddgddedddddgddedddddgd deed
|
eed
|
d dg¡d d Zdd Z dd Z
|
ej de dd¡dd Z dd Z dd Zej dddg¡d d Zd d Zej d d!d"gd!gfd!d"d!d!gd!gfg¡d#d$ Zd%d& Zd'd( Zd)S )*ÚTestSeriesNLargestNSmallestÚrç @r r Ú5Úobjectr r Z
|
complex128ZabcdeÚcategoryc
|
C sf |j }d| }dt|ddf}|j|jf}t||D ]*\}}tjt|d || W 5 Q R X q6d S )Nz3Cannot use method 'n(largest|smallest)' with dtype r r éÿÿÿÿ©Úmatch)r ÚlenÚnlargestr r ÚpytestÚraisesÚ TypeError)Úselfr, ÚdtÚmsgÚargsÚmethodsr& Úargr r r Útest_nlargest_errorW s
|
z/TestSeriesNLargestNSmallest.test_nlargest_errorc C s" |}t | d¡|jddg ¡ t |jddd|jddg ¡ |jdd
}t | d¡|¡ t | d¡|¡ t | d¡|¡ t | d¡|¡ t | t|¡| ¡ ¡ t | t|d ¡| ¡ ¡ t | t|¡|jdddddg ¡ t | t|d ¡|jdddddg ¡ d S ) Nr r Úlast©Zkeepr r r1 é )r# r$ r Úilocr5 r4 Ú sort_values)r9 r r' Úemptyr r r Útest_nsmallest_nlargestl s &z3TestSeriesNLargestNSmallest.test_nsmallest_nlargestc C sd t dtjdddg}| ¡ }|jdddddg }t ||¡ | ¡ }|jdddddg }t ||¡ d}tj t
|
|d |jd
|
d W 5 Q R X tj t
|
|d |jd
|
d W 5 Q R X t dgd dddddgd }t dgd dddgd }t dgd dddgd }| d¡}t ||¡ |jdd d }t ||¡ | d¡}t ||¡ |jdd d }t ||¡ d S )Nr- r r r rB r r z#keep must be either "first", "last"r2 ÚinvalidrA ©Úindexr@ ) r ÚnpÚnanr5 rC r# r$ r r6 r7 Ú
|
ValueError)r9 r' r( r) r; Zexpected_firstZ expected_lastr r r Útest_nlargest_misc s.
|
|
z.TestSeriesNLargestNSmallest.test_nlargest_miscÚnc C sl t ddddgddddgd}| |¡}|jdd |¡}t ||¡ | |¡}| ¡ |¡}t ||¡ d S ) Nr rB r r r rH F)Z ascending)r r5 rD Úheadr# r$ r )r9 rN r' r( r) r r r Útest_nlargest_n s
|
|
z+TestSeriesNLargestNSmallest.test_nlargest_nc C s< t |¡}|j|j }}||d |d |g}t||| d S ©Nr ©rJ ZiinfoÚminÚmaxr* )r9 Únselect_methodZany_int_numpy_dtypeÚ
|
dtype_infoÚmin_valÚmax_valr% r r r Ú test_nlargest_boundary_integer« s
|
z:TestSeriesNLargestNSmallest.test_nlargest_boundary_integerc C sL t |¡}|j|j }}t j||gd|d\}}||||g}t||| d S )Nr r )rJ ZfinforS rT Z nextafterr* ) r9 rU Zfloat_numpy_dtyperV rW rX Zmin_2ndZmax_2ndr% r r r Ú test_nlargest_boundary_float² s
|
|
z8TestSeriesNLargestNSmallest.test_nlargest_boundary_floatr zdatetime64[ns]ztimedelta64[ns]c C sB t d¡}|j|j }}|d |d |d ||g}t||| d S )Nr
|
r r rR )r9 rU r rV rW rX r% r r r Ú#test_nlargest_boundary_datetimelikeº s
|
z?TestSeriesNLargestNSmallest.test_nlargest_boundary_datetimelikec C s t ddddddddg}|jddd}t dddddddg}t ||¡ |jd dd}t dddddgdd
|
dd dgd }t ||¡ d S ) Né
|
é é é é rB ÚallrA r r r rH )r r5 r# r$ r )r9 r' r( r) r r r Ú%test_nlargest_duplicate_keep_all_tiesÄ s zATestSeriesNLargestNSmallest.test_nlargest_duplicate_keep_all_tiesz data,expectedTFc C s* t |}| d¡}t |}t ||¡ d S rQ )r r5 r# r$ )r9 Údatar) r' r( r r r Útest_nlargest_booleanÏ s
|
z1TestSeriesNLargestNSmallest.test_nlargest_booleanc C sz |}t j d¡j| ¡ dd}t| ¡ |d}tj|d< | d¡}tt
|
|d¡|j
|
d¡d d¡ |¡}t ||¡ d S )Nr\ F)Úcopyr r r rH )rJ ÚrandomZrandnZastypeÚlowerr re r ZNAr5 ÚdeleterI r# r$ )r9 Zany_numeric_ea_dtyper Zarrr' r( r) r r r Útest_nlargest_nullableÙ s
|
|
ÿþÿz2TestSeriesNLargestNSmallest.test_nlargest_nullablec C s t dddddd g}|jddd}t dddddg}t ||¡ t ddd d d g}|jddd}t ddd d d g}t ||¡ d S ) Nr r r ra rA g ð?g @r- )r r r# r$ )r9 Úsr( r) r r r Ú#test_nsmallest_nan_when_keep_is_allé s z?TestSeriesNLargestNSmallest.test_nsmallest_nan_when_keep_is_allN)Ú__name__Ú
|
__module__Ú __qualname__r6 ÚmarkZ parametrizer Úlistr? rF rM ÚrangerP rY rZ r[ rb rd ri rk r r r r r+ V s4
|
ùþ
|
|
|
ÿ
|
r+ )Ú__doc__Ú itertoolsr ÚnumpyrJ r6 Zpandasr r Zpandas._testingZ_testingr# Z main_dtypesZfixturer r r* r+ r r r r Ú<module> s4 ó
|
'
|
|
|
