U
|
ý°dy% ã
|
@ s* d dl Z d dlZd dlZd dlmZmZ d dlmZm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ dd Zdd Zd d
|
Zd d Zd d Zdd Zdd Zdd Zej dddg¡dd Z ej ddd d gdd d d gg¡ej d d d!d"d#gd d!d"d#d gg¡d$d% Z d&d' Z d(d) Z d*d+ Z!d,d- Z"dS ).é N)ÚPerformanceWarningÚUnsortedIndexError)ÚCategoricalIndexÚ DataFrameÚIndexÚ
|
MultiIndexÚ
|
RangeIndexÚ Timestamp)Ú
|
FrozenListc C sÔ t | }t |¡ t |¡}| d¡\}}t t|¡}| |¡sFt|jddd\}}| |d d d
¡spt| d¡\}}t|dd d}t |¡}| |¡s¦t|jddd\}}| |d d d
¡sÐtd S ) Nr F©Z ascendingéÿÿÿÿé c S s | d | d fS ©Nr r © ©Úxr r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_sorting.pyÚ<lambda>% ó z test_sortlevel.<locals>.<lambda>©Úkey) ÚlistÚrandomÚshuffler Ú from_tuplesÚ sortlevelÚsortedÚequalsÚAssertionError)ÚidxÚtuplesÚindexÚ
|
sorted_idxÚ_ÚexpectedÚby1r r r Útest_sortlevel s
|
|
|
r& c C sF t jdddgdddggtdd} | jddd\}}| | ¡sBtd S )Nr é ÚABC©ÚnamesÚAF)Zsort_remaining)r r r r r r )Úmir" r# r r r Ú!test_sortlevel_not_sort_remaining- s "r- c C sÒ ddddddg} t | ¡}| d¡\}}t t| ¡}| |¡sDt|jddd \}}| |d d d
|
¡snt| d ¡\}}t| d d d}t |¡}| |¡s¤t|jd dd \}}| |d d d
|
¡sÎtd S )N)ÚbarÚone)ÚfooÚtwo)Úquxr1 )r0 r/ )Zbazr1 )r2 r/ r Fr r r c S s | d | d fS r r r r r r r G r z.test_sortlevel_deterministic.<locals>.<lambda>r )r r r r r r )r r! r" r# r$ r% r r r Ú test_sortlevel_deterministic3 s& ú
|
|
r3 c C s¸ t | ¡}| ¡ }t ||¡ tt| ttfr´d}tj t
|
|d t j| dd W 5 Q R X d}tj t
|
|d t j| dd W 5 Q R X d}tj t
|
|d t j| d d
|
W 5 Q R X d S ) Nz%the 'axis' parameter is not supported©Úmatchr ©Zaxisz%the 'kind' parameter is not supportedZ mergesort)Úkindz&the 'order' parameter is not supported)ÚaÚb)Úorder) ÚnpÚargsortÚtmÚassert_numpy_array_equalÚ
|
isinstanceÚtyper r ÚpytestÚraisesÚ
|
ValueError)r Úresultr$ Úmsgr r r Útest_numpy_argsortO s
|
rF c C s t jddddddgddgd } td
|
d td D | ddgd }|jddd }|jd }t ||¡ d}tj t
|
|d |jdddt df W 5 Q R X |j dd t |jdddd d
f dksÊttj tdd |jdddd d
f W 5 Q R X d S )N©Úzr8 ©r r8 ©Úyr9 ©r r9 ©rK r8 ©rH r9 r/ r1 r) c S s g | ]}|d | gqS )é
|
r )Ú.0Úir r r Ú
|
<listcomp>o s z&test_unsortedindex.<locals>.<listcomp>é )r! Úcolumnsr r6 z_MultiIndex slicing requires the index to be lexsorted: slicing on levels \[1\], lexsort depth 0r4 rH r8 T)Zinplaceé z'q'Úq)r r r ÚrangeÚlocZilocr= Zassert_series_equalrA rB r ÚsliceÚ
|
sort_indexÚlenr ÚKeyError)r, ÚdfrD r$ rE r r r Útest_unsortedindexi s þ
|
ÿ" $r^ c C sâ t ddddgddddgtj d¡d} | dd g¡} t t¡ | jd
|
W 5 Q R X d }t j
|
t |d | jd d
|
W 5 Q R X | j ¡ rt| j jdks t| ¡ } | jd
|
| jd d
|
| j ¡ sÎt| j jdksÞtd S )Nr r r rH rK é )ÚjimÚjoeZjolier` ra )r rH z@Key length \(2\) was greater than MultiIndex lexsort depth \(1\)r4 )r rK rU )r r; r ÚrandÚ set_indexr= Zassert_produces_warningr rX rA rB r r! Z _is_lexsortedr Z_lexsort_depthrZ )ZdfmrE r r r Ú test_unsortedindex_doc_examples s "ÿ
|
rd c C sZ t dddddgdddddgg¡} | js,t| ¡ }|js>t| |ksJt| |¡sXtt| j t|j¡srtt jdddd d
|
d gd d gd} | jrt| ¡ }|jr¬t| |¡sºtt| j t|j¡sÔtt dddgdddggddddgddddggddgd} | jrt| ¡ }|jr*t| |¡s:tt| j t|j¡sVtd S )Nr+ ÚBr rU r' rG rI rJ rL rM rN r/ r1 r) r9 Údr8 r Zcol1Zcol2©ÚlevelsÚcodesr* ) r Ú from_arraysZis_monotonic_increasingr Z_sort_levels_monotonicr r Úvaluesr )r, Zreconsr r r Útest_reconstruct_sort s4 "
|
|
þ
|
|
ý rl c C sò t dddgdddgdddggdd d
|
gd } | jdd gd d }||d dk }tdddgdddggddgddggdd gd}|j}t ||¡ tddgddggddgddggdd gd}|j ¡ }t ||¡ | ¡ }t ||¡ | |¡sîtd S )NZdeleteMer é ZkeepMerU Z keepMeToor' ÚfirstÚsecondÚthird)rT F)Zdroprg r ) r rc r r! r= Úassert_index_equalÚremove_unused_levelsÚis_r )r] Zdf2r$ rD Zresult2r r r Ú test_reconstruct_remove_unusedÀ s. þý ý
|
rt zfirst_type,second_type)Úint64ru )z datetime64[D]Ústrc C sè t j d¡}d}t| dd|¡ | ¡| dd|¡ |¡| |¡d}| ddg¡ ¡ }||j d k }|j
|
¡ }t |j d t |j
|
j d k stt |j d
|
t |j
|
j d
|
k s´t| |j
|
¡sÄt| ¡ ddg¡j
|
}t ||¡ d S ) Nr_ i r i i )rn ro rp rn ro g¹?r )r; r Z RandomStater ÚrandintZastyperb ÚgroupbyÚsumrp r! rr r[ rh r r Z reset_indexrc r= rq )Z
|
first_typeZ second_typeÚrngÚsizer] rD r$ r r r Ú test_remove_unused_levels_largeà s ýÿ
|
""r| Úlevel0r8 rf r9 ÚunusedÚlevel1Úwr rK rH c C s\ t | |gdddddgdddddggd}| ¡ }t ||¡ dD ]}d|j| ks@tq@d S ) Nr rU r r r' ©rh ri )r r r~ )r rr r= rq rh r )r} r r, rD Úlevelr r r Útest_remove_unused_nanÿ s
|
( r c C s" | ¡ }| j ¡ }t ||¡ d S )N)r< rk r= r> ©r rD r$ r r r Ú test_argsort s
|
r
c C sj t dtjfdg ddg¡} | jdtjgdd} | ¡ } | j}tdtjgdgg}t|t|ksft d S )Nr )r' r_ Zid1Zid2r8 )r r_ )
|
r r; ÚnanÚrenameZ
|
set_levelsrr rh r
|
rv r r r r r Ú"test_remove_unused_levels_with_nan s r c C sh t dddgdggdddgdddggd } | ¡ }t dddgdggdddgdddggd }t ||¡ d S )
|
Nr+ re ÚCÚDr r rU r r )r Ú sort_valuesr= rq )ZmidxrD r$ r r r Útest_sort_values_nan s ( ÿr c C sD t dtdgddgg¡} d}tjt|d | ¡ W 5 Q R X d S )Nr z
|
2000-01-01r' r_ z<'<' not supported between instances of 'Timestamp' and 'int'r4 )r rj r rA rB Ú TypeErrorr )r, r5 r r r Ú test_sort_values_incomparable' s
|
þÿr )#r Únumpyr; rA Z pandas.errorsr r Zpandasr r r r r r Zpandas._testingZ_testingr= Zpandas.core.indexes.frozenr
|
r& r- r3 rF r^ rd rl rt ÚmarkZ parametrizer| r r
r r r r r r r Ú<module> s: # ÿ
|
ÿ
|
|
|
