U
|
¬ý°d G ã @ s® d dl mZ d dlmZ d dlmZ d dlZd dlZd dlm Z m
|
Z
|
d dl m Z m Z mZmZmZmZmZ d dlmZ d dlmZ d dlm mZ d dl m Z m Z m Z m Z m!Z! ej"d d
|
Z#G d d d Z$G d d dZ%ej& 'de (d dddd dddgd¡ )ej*¡e (d dddd dddgd¡ )ej*¡e (ddd ddd ddgd¡ )ej*¡gdge (ej+dej*dd¡e (ej+dej*dd¡gdgg¡dd Z,G d d d Z-d d Z.d d! Z/dS )"é )Ú defaultdict)Údatetime)ÚproductN)Úis_ci_environmentÚis_platform_windows)ÚNAÚ DataFrameÚ
|
MultiIndexÚSeriesÚarrayÚconcatÚmerge)Ú safe_sort)Ú_decons_group_indexÚget_group_indexÚis_int64_overflow_possibleÚlexsort_indexerÚnargsortc C s d\} }}t tj | ||df¡tdd}|jdd|d< tj t|¡}|j| ¡ }|j
|
d d
¡ d g |_
|
t t|¡|_ |d d9 < ||fS )
|
N)é üÿÿé i é ÚABCDEFG©Úcolumnsé ©ZaxisÚleftéÿÿÿÿÚright)r ÚnpÚrandomÚrandintÚlistÚsumÚ permutationÚlenÚilocÚcopyr ÚtolistÚarangeÚindex)ÚlowÚhighÚnr Úir © r/ úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/test_sorting.pyÚ
|
left_right" s
|
r1 c @ s¦ e Zd Zejjdd Zdd Zej dddg¡dd Z ej d
|
d d e
|
e d de
|
e d e
|
e dd gd de
|
e d e
|
e dd e
|
e d d gdd e
|
e ddde
|
e d e
|
e dd gdde
|
e d e
|
e dd e
|
e ddd gg¡dd Z ej dd d e
|
e d de
|
e d e
|
e dd e
|
gd de
|
e d e
|
e dd e
|
e d d e
|
gdd e
|
e ddde
|
e d e
|
e dd e
|
gdde
|
e d e
|
e dd e
|
e ddd e
|
gd d e
|
e d de
|
e d e
|
e dd dd gd de
|
e d e
|
e dd e
|
e d d dd gdd e
|
e ddde
|
e d e
|
e dd dd gdde
|
e d e
|
e dd e
|
e ddd d d gg¡d d Z d S ) Ú TestSortingc C sv t t d¡t d¡t d¡f¡}t d¡}t||||||||t j d¡d }| ddddd d
|
d d g¡}| d d d
|
d ddddg¡}| ¡ d }| ¡ d }|j ¡ \}} t
|
|j|¡ |j d¡\}} t
|
|j|¡ t t t|ddddd d
|
d d g j}
|
t |
|
¡}
|
| |
|
¡ ¡ d } | ¡ D ]<\} } || || d d d
ksFt|| | ks tq t|t|ksrtd S )Néè iô iÄ ) ÚAÚBÚCÚDÚEÚFÚGÚHÚvaluesr4 r5 r6 r7 r8 r9 r: r; r< r r )r Z concatenater) r r ÚrandnÚgroupbyr# r* Z sortlevelÚtmÚassert_index_equalr" ÚmapÚtupler< ÚcomZasarray_tuplesafeÚitemsÚAssertionErrorr% )Úselfr5 r4 ÚdfZlgZrgr r Z exp_indexÚ_ZtupsÚexpectedÚkÚvr/ r/ r0 Útest_int64_overflow2 s: "
|
|
÷ÿ $
|
z TestSorting.test_int64_overflowc C sF t d}t ||||d¡}| ddddg¡}t|t|ksBtd S )NiE× )ÚaÚbÚcÚdrM rN rO rP )Úranger Ú from_dictr> r% rE )rF r< ÚdataZgroupedr/ r/ r0 Ú'test_int64_overflow_groupby_large_rangeZ s z3TestSorting.test_int64_overflow_groupby_large_rangeÚaggZmeanZmedianc s~ t j ddd¡}t j t|t|d ¡}t ||| f¡}t j t|¡}|| }t|tdd}t j dt|¡d \|d < |d
|
< |
|
td¡}t |j j s¤ttttt }}ttt||d |d
|
D ]&\}} }
|
|| | ¡ || |
|
¡ qÐt|t|ksttj| ¡ tdd } fd d } t | | ¡ | | ¡ f¡j}t|d d
|
g| d ¡ } t t | | ¡ d S )Ni ðÿÿi )i é é Zabcder é é
|
ÚjimÚjoe)Únamesc s t jttt | ddS )NZf8©Zdtype)r ZfromiterrA Úgetattr)rM ©rU r/ r0 Ú<lambda>z ó zKTestSorting.test_int64_overflow_groupby_large_df_shuffled.<locals>.<lambda>)r r* ) r r r! Úchoicer% Zvstackr$ r r" r= r> r ZgrouperÚshaperE r ÚziprA rB Úappendr Ú from_tuplesÚkeysr< ÚTZ
|
sort_indexr? Úassert_frame_equalr^ )rF rU Úarrr. rG ZgrrZ r[ ÚkeyrM rN ÚmiÚfÚresr/ r_ r0 Ú-test_int64_overflow_groupby_large_df_shuffleda s& "$ z9TestSorting.test_int64_overflow_groupby_large_df_shuffledzorder, na_position, expTÚlastrV éi én ÚfirstFéh rW r c C sP t jgd ttd t jgd g}t|||d}t |t j|t jd¡ d S )NrV éd )ZordersÚ na_positionr] ) r Únanr" rQ r r? Úassert_numpy_array_equalr Úintp)rF Úorderrv Úexprg Úresultr/ r/ r0 Útest_lexsort_indexer s &z TestSorting.test_lexsort_indexerz ascending, na_position, exp, boxc C s t j| ddS ©NÚOr] ©r r ©Úxr/ r/ r0 r` ½ ra zTestSorting.<lambda>c C s t j| ddS r~ r r r/ r/ r0 r` Ã ra c C s t j| ddS r~ r r r/ r/ r0 r` É ra c C s t j| ddS r~ r r r/ r/ r0 r` Ï ra c C sR |t jgd ttd t jgd }t|d||d}tj|t |¡dd d S )NrV ru Ú mergesort)ÚkindÚ ascendingrv F)Z check_dtype)r rw r" rQ r r? rx r )rF r
rv r{ ÚboxrD r| r/ r/ r0 Ú test_nargsort s 7(
|
ÿzTestSorting.test_nargsortN)Ú__name__Ú
|
__module__Ú __qualname__ÚpytestÚmarkÚslowrL rT Ú parametrizero r" rQ r} r r/ r/ r/ r0 r2 1 s
|
'
|
&ý&ý(ý(ýðþ
|
&ü&ü(ü(ü&ü&ü(ü(üÕþ5r2 c @ s e Zd Zdd Zejjdd Zejjej ddddd g¡d
|
d Z ejjd d Z
|
ejjej ddddd g¡ej dddg¡dd Z dS )Ú TestMergec C sb t tj dd¡tddg d}t tj dd¡tddg d}t||dd}t|d ks^td S )
|
Nr3 r ZABCDEFZG1r ZG2Úouter©ÚhowiÐ )r r r r= r" r r% rE )rF Zdf1Zdf2r| r/ r/ r0 Ú test_int64_overflow_outer_mergeæ s z)TestMerge.test_int64_overflow_outer_mergec C s |\}}t ||dd}t|t|ks*ttj|d |d dd |jd d
d d
f jdd }tj|d |dd |jd kstd S )
|
Nr r r r F)Z check_nameséþÿÿÿr r )r r% rE r? Zassert_series_equalr& r# Úname)rF r1 r r Úoutr| r/ r/ r0 Ú!test_int64_overflow_check_sum_colí s z+TestMerge.test_int64_overflow_check_sum_colr r r r Úinnerc C sV |\}}t ||dd}|j|j ¡ dd t t|¡|_t |t |||dd¡ d S )Nr r T)Zinplace©r Úsort)
|
r Ú sort_valuesr r( r r) r% r* r? ri )rF r1 r r r r r/ r/ r0 Ú test_int64_overflow_how_mergeø s
|
z'TestMerge.test_int64_overflow_how_mergec C sX |\}}t ||ddd}t |||j ¡ ¡ t ||ddd}t |||j ¡ ¡ d S )Nr Fr )r r? ri r r( )rF r1 r r r r/ r/ r0 Ú$test_int64_overflow_sort_false_order s
|
z.TestMerge.test_int64_overflow_sort_false_orderr TFc C s& d\}}}t tj |||df¡ d¡tdd}| tj¡j }t
|
|sJt t ||gdd}t tj |||d df¡ d¡tdd}tj t||¡} t |||j| gdd}tj t|¡|d < tj t|¡|d
|
< tj t|¡} |j| ¡ }t t|¡|_tj t|¡} |j| ¡ }t t|¡|_tttt }
|
} | td¡ ¡ D ] \} } |
|
| | d ¡ qV| td¡ ¡ D ] \} } | | | d
|
¡ qg }|
|
¡ D ]B\}}| |tjg¡}t ||D ] \}}| |||f ¡ qÒq°| ¡ D ]4\}}||
|
krü|D ]}| |tj|f ¡ qqüd d }t |tdd d
|
g d}||}|d ¡ |d
|
¡ |d ¡ |d
|
¡ @ tj t|d dd}|| }||| ¡ }| ¡ | ¡ A sÒ|dksÒt t!||||d}|rtd}t" #|| ¡ || j$|dd¡ t" #|||¡ d S )N)r r i r Úint64r r T)Z ignore_indexrX r r c S s$ | | j ¡ ¡} t t| ¡| _| S ©N)r r r( r r) r% r* )rG r/ r/ r0 ÚalignT s zCTestMerge.test_int64_overflow_one_to_many_none_match.<locals>.alignÚboolr] )r r r r r r r )r )%r r r r! Úastyper" Úapplyr
|
Znuniquer< r rE r rb r% r& r= r$ r' r) r* r Z set_indexZiterrowsre rD Úgetrw r ZnotnaZonesÚallÚanyr r? ri r )rF r r r+ r, r- r rc r r. ZldictZrdictÚidxÚrowÚvalsrJ ZlvalÚrvalÚlvÚrvr r ZjmaskÚmaskÚframern Zkcolsr/ r/ r0 Ú*test_int64_overflow_one_to_many_none_match s
|
þ þþÿÿ
|
þÿÿ
|
|
ü
|
ÿz4TestMerge.test_int64_overflow_one_to_many_none_matchN) r r r r r r r r r r r r¯ r/ r/ r/ r0 r å s
|
|
|
|
r zcodes_list, shaper rX é ru rW rV )rW rV é é' r] )r² r² c C s> t | |ddd}t||}t| |D ]\}}t ||¡ q$d S )NT)r Zxnull)r r rd r? rx )Z
|
codes_listrc Z group_indexZ codes_list2rM rN r/ r/ r0 Ú test_deconss s
|
r³ c @ s e Zd Zej ddddddgdddddggedejede d gg g gg¡d
|
d Z
|
ej d d dg¡ej dddddddddgddddddddggg g gg¡dd Z ejj e oºe dddd Zej ddd eg¡dd Zd d Zd d Zej d dd!ed"gdddgded#gddddgddged$gg¡d%d& Zej ddddgdddggddejdgdddejggg¡d'd( Zej d d dg¡d)d* Zd!S )+Ú TestSafeSortzarg, expr° r rX r rW ZbaaacbZaaabbcr] c C s" t |}t |¡}t ||¡ d S r )r r r r? rx )rF Úargr{ r| rI r/ r/ r0 Útest_basic_sort s
|
z TestSafeSort.test_basic_sortÚverifyTFzcodes, exp_codesr c C sb dddddg}t dddddg¡}t||d|d\}}t j|t jd}t ||¡ t ||¡ d S ) Nr° r rX r rW T©Úuse_na_sentinelr· r] ©r r r ry r? rx ) rF r· ÚcodesZ exp_codesr< rI r| Ú result_codesÚexpected_codesr/ r/ r0 Ú
|
test_codes s ÿ
|
zTestSafeSort.test_codeszRIn CI environment can crash thread with: Windows fatal exception: access violation)Úreasonc C s dddddg}t dddddg¡}ddddddddg}t||d d
|
\}}t jdd d dddd dgt jd }t ||¡ t ||¡ d S ) Nr° r rX r rW ée éf éc T)r¹ r r] rº )rF r< rI r» r| r¼ r½ r/ r/ r0 Útest_codes_out_of_bound° s z$TestSafeSort.test_codes_out_of_boundr c C s t j| tdS )Nr] )r r Úobjectr r/ r/ r0 r` À ra zTestSafeSort.<lambda>c C sF |ddddddg}t |}tjddddddgtd}t ||¡ d S )NrN r r rM r] )r r r rÄ r? rx )rF r r< r| rI r/ r/ r0 Útest_mixed_integerÀ s z TestSafeSort.test_mixed_integerc C s t jddddgtd}dddddddg}t||\}}t jddddgtd}t jdddddddgt jd}t ||¡ t ||¡ d S ) NrN r r rM r] rX r° r )r r rÄ r ry r? rx )rF r< r» r| r¼ rI r½ r/ r/ r0 Ú test_mixed_integer_with_codesÇ s z*TestSafeSort.test_mixed_integer_with_codesc C sF t jddt ¡ ddgtd}d}tjt|d t| W 5 Q R X d S )Nr rX r r° r] z,'[<>]' not supported between instances of .*©Úmatch) r r r ÚnowrÄ r ÚraisesÚ TypeErrorr )rF rj Úmsgr/ r/ r0 Útest_unsortableÐ s z TestSafeSort.test_unsortablezarg, codes, err, msgNz"Only list-like objects are allowedz Only list-like objects or Nonezvalues should be uniquec C s* t j||d t||d W 5 Q R X d S )NrÇ )r< r» )r rÊ r )rF rµ r» ÚerrrÌ r/ r/ r0 Útest_exceptions× s z TestSafeSort.test_exceptionsc C s0 t |dd}t|}t |dd}t ||¡ d S )NÚInt64r] )r r r? Ú assert_extension_array_equal)rF rµ r{ rM r| rI r/ r/ r0 Útest_extension_arrayã s z!TestSafeSort.test_extension_arrayc C st t dddgdd}t|ddddgd|d \}}t dddgdd}tj ddddgtjd}t ||¡ t ||¡ d S )
|
Nr r° rX rÐ r] r r Tr¸ )r r r ry r? rÑ rx )rF r· rM r| r» Zexpected_valuesr½ r/ r/ r0 Útest_extension_array_codesì s z'TestSafeSort.test_extension_array_codes)r r r r r r r" r r rÄ r¶ r¾ Zskipifr r rà rÅ rÆ rÍ rË Ú
|
ValueErrorrÏ rw rÒ rÓ r/ r/ r/ r0 r´ sP ýþ
|
&þþ
|
þ
|
|
|
ýþ
|
.ÿ
|
r´ c C sD t jd| ddgtd}t|}t jddd| gtd}t ||¡ d S )NrN rM r] )r r rÄ r r? rx )Z nulls_fixturer< r| rI r/ r/ r0 Útest_mixed_str_nullö s rÕ c C sl t ddttgdd} ddddg}t | |g¡}t|}t t ddttgddddddgg¡}t ||¡ d S )NrX r rÐ r] r° )r
|
r r Z from_arraysr r? r@ )Zarr1Zarr2Zmidxr| rI r/ r/ r0 Útest_safe_sort_multiindexý s ÿrÖ )0Ú collectionsr r Ú itertoolsr Únumpyr r Z pandas.compatr r Zpandasr r r r
|
r r r Zpandas._testingZ_testingr? Zpandas.core.algorithmsr Zpandas.core.commonÚcoreÚcommonrC Zpandas.core.sortingr r r r r Zfixturer1 r2 r r r Ztiler¢ r r) r³ r´ rÕ rÖ r/ r/ r/ r0 Ú<module> sF $
|
5 """ýú
|
þû÷þ
|
g
|
