U
|
ý°d± ã @ sp d Z ddlmZ ddlZddlZddlZddlm Z
|
ej dd Z ej dd Z ej dd ZG d
|
d d ZdS ) z
|
Note: for naming purposes, most tests are title with as e.g. "test_nlargest_foo"
|
but are implicitly also testing nsmallest_foo.
|
é )Úascii_lowercaseNc C s> t jdddddgdddddgdddddgddddddgdS ) Né é é é r é ©ÚaÚbÚc©Úindex)ÚpdÚ DataFrame© r r ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_nlargest.pyÚ df_duplicates s ( þr c C s4 t tj d¡ttd d
tj d¡ d¡d¡S )Né
|
Úfloat64r )r r ÚnpÚrandomZ permutationÚlistr Úastyper r r r Ú
|
df_strings s
|
ýÿr c C s t jdddgdddgdddgtdt td¡ d¡d d
|
d gt jd dd t jd dddt jddddd ddddddddd g d S ) Nr r r g @g @g @ÚabcÚcategoryé é é Z20130101)Úperiodsz
|
US/Eastern)r Útzz1 sÚs)r Úfreq) ÚgroupÚintÚfloatÚstringÚcategory_stringÚ category_intÚdatetimeÚ
|
datetimetzÚ timedeltar# r$ r% r& r' r( r) r* r+ ©Úcolumns)r r r ZSeriesr Z
|
date_rangeZtimedelta_ranger r r r Údf_main_dtypes! s, ÷ ÷ôr. c @ s. e Zd Zej ddgdgddgddgddgddgdddgdddgdddgdddgdddgdddgg ¡ej dedd¡dd Zej d
|
d d gd d gg¡dd Zdd Z dd Z
|
dd Z ej ddddgdddgdgdgddgddgg¡ej dedd¡dd Z dd Z dd Zd d Zd d Zd!S )"ÚTestNLargestNSmallestÚorderr r r
|
Únr é c
|
C s |}d|krDd| d}t jt|d t|||| W 5 Q R X n8|dk}t||||}|j||d |¡} t || ¡ d S )Nr
|
z0Column 'b' has dtype object, cannot use method 'ú' with this dtype©ÚmatchÚ nsmallest©Ú ascending)ÚpytestÚraisesÚ TypeErrorÚgetattrÚ sort_valuesÚheadÚtmÚassert_frame_equal)
|
Úselfr Únselect_methodr1 r0 ÚdfÚ error_msgr8 ÚresultÚexpectedr r r Útest_nlargest_n@ s
|
ÿ z%TestNLargestNSmallest.test_nlargest_nr- r# r' r& c C s| |}|d }d| d|| j d| d}| dd¡ dd ¡ d
|
d ¡ d d ¡}tjt|d t||d| W 5 Q R X d S )Nr zColumn 'z ' has dtype z, cannot use method 'r3 ú(z\(ú)z\)ú[z\[ú]z\]r4 r )ZdtypeÚreplacer9 r: r; r< )rA r. rB r- rC ÚcolrD r r r Útest_nlargest_errorc s" ÿ ÿ þ ýÿz)TestNLargestNSmallest.test_nlargest_errorc C s@ |}| dtt|ddh ¡ | dtt|ddh ¡ d S )Nr r' r& )r6 r ÚsetÚnlargest)rA r. rC r r r Útest_nlargest_all_dtypesw s z.TestNLargestNSmallest.test_nlargest_all_dtypesc C sÀ t ddddddgddddddgd¡}|jdddgd}t jddddgddddgdddd dgd
|
}t ||¡ |jdddgd}t jddddgddddgdd ddd gd
|
}t ||¡ d S ) Nr r r ©r r
|
r r r
|
r, r r r ©r r rP r? r@ r6 ©rA rC rE rF r r r Ú+test_nlargest_duplicates_on_starter_columns| s (
|
ÿ
|
ÿzATestNLargestNSmallest.test_nlargest_duplicates_on_starter_columnsc C s t dgd dddddgd¡}| dd¡}t jdgd dddgddddgd }t ||¡ | dd¡}t dgd dddgd¡}t ||¡ d S )
|
Nr r r r r rR r r r rS rT r r r Ú test_nlargest_n_identical_values s & z6TestNLargestNSmallest.test_nlargest_n_identical_valuesé c C s\ |}| ||¡}| |¡ |¡}t ||¡ | ||¡}|j|dd |¡}t ||¡ d S )NFr7 )r6 r= r> r? r@ rP )rA r r1 r0 rC rE rF r r r Ú test_nlargest_n_duplicate_index s z5TestNLargestNSmallest.test_nlargest_n_duplicate_indexc C sÄ t ddddddddgdddddd dd
|
gd ¡}|jdd d d}t ddddddddddddd dd
|
dd ¡}t ||¡ |jdd d d}t ddddddddd dd
|
dd ¡}t ||¡ d S )Nr r r r r r r r é2 é rR r Úall)Zkeep)r r r r r rW r )r r r rW r rS rT r r r Ú%test_nlargest_duplicate_keep_all_tiesª s" (ÿþÿ þÿz;TestNLargestNSmallest.test_nlargest_duplicate_keep_all_tiesc C s t jt j dgddgg¡ddgddgddgd d
|
ggd }| d d ¡}|jddd g }t ||¡ | d d¡}|jd ddg }t ||¡ d S )NÚxr r
|
g
ëQ¸ Õ?g¤p=
|
×£À?g
ëQ¸
ë?g Ð?gffffffæ?g333333ë?g
ëQ¸ í?)r- Údatar )r] r r r )r] r
|
r ) r r Z
|
MultiIndexZ from_productr6 Zilocr? r@ rP rT r r r Ú&test_nlargest_multiindex_column_lookupÁ s þ z<TestNLargestNSmallest.test_nlargest_multiindex_column_lookupc C sJ t tjtjddddg¡}| dd¡}|jddd d¡}t ||¡ d S )Nr r r r r Fr7 ) r r r ÚnanrP r= r> r? r@ rT r r r Útest_nlargest_nanÓ s z'TestNLargestNSmallest.test_nlargest_nanc
|
C s¨ t jdddddd dgdddddddgdddddddgdtdd }|jdd
|
d gd }t jdddddgdddddgdddddgdtdd d
|
d i¡}t ||¡ d S )Nr r r r r r rW r r r r
|
r, r% )r r Úranger6 r r? r@ rT r r r Ú"test_nsmallest_nan_after_n_elementÚ s$ ýú ýúùz8TestNLargestNSmallest.test_nsmallest_nan_after_n_elementN)Ú__name__Ú
|
__module__Ú __qualname__r9 ÚmarkZ parametrizerb rG rN rQ rU rV rX r\ r_ ra rc r r r r r/ = sH óþ ÿ
|
&þ r/ )Ú__doc__r& r Únumpyr r9 Zpandasr Zpandas._testingZ_testingr? Zfixturer r r. r/ r r r r Ú<module> s
|
|
|
|
|
