U
|
ý°dÎ| ã @ sl d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z d dl mZ G dd dZG dd dZdS )é N)ÚCategoricalDtypeÚCategoricalIndexÚ DataFrameÚ IntervalIndexÚ
|
MultiIndexÚ
|
RangeIndexÚSeriesÚ Timestampc @ s e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z ej dddg¡dd Zd d Zej d d d g¡ej d!dd"d#d$gidd#d$d"gid d dd"d#gfdd"d#d$gidd"d$d#gid d dd"d#gfdd"d#d$gidd#d$d"gid d d%d$d#gfdd"d#d$gidd"d$d#gid d d#d$d%gfg¡d&d' Zej d d d g¡ej d(d d g¡d)d* Zej d d d g¡ej d!d"d#gd$d+gd,d"d#gd$d+gd,d d dd"gfd"d#gd$d+gd,d#d"gd+d$gd,d d dd"gfd"d#gd$d+gd,d"d#gd$d+gd,d d ejd-d.ged/d0fd"d#gd$d+gd,d#d"gd+d$gd,d d ejd.d-ged/d0fg¡d1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Z d;d< Z d=d> Z d?d@ Z ej dAdBdCdDdEgdFfd"d$d#d%gd+fe dGe dHe dIe dJge dKfdLdMdNdOgdPfg¡dQdR Z!ej dSej"dTdUdVgdWdXej"e# $dd"¡e# $d"d#¡e# $d#d$¡gdYdXg¡dZd[ Z%ej d\d]d d]gd d^gg¡d_d` Z&dadb Z'ej d\dcd d gg¡ddde Z(d]S )fÚTestDataFrameSortIndexc C sî t dddddgitddgdd ggd
|
d
|
ddgd
|
dd
|
dggd d }|j ¡ sNt|jjrZtt dddddgitddgd dggd
|
d
|
ddgd
|
dd
|
dggd d }| ¡ }|jjs®tt ||¡ | ¡ ¡ }|j
|
¡ |_|jjsÞtt ||¡ d S ) NÚvalueé é é é ÚaÚbZbbZaar ©ÚlevelsÚcodes©Úindex) r r r Ú _is_lexsortedÚAssertionErrorÚis_monotonic_increasingÚ
|
sort_indexÚtmÚassert_frame_equalÚcopyZ_sort_levels_monotonic)ÚselfÚdfÚexpectedÚresult© r" úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_sort_index.pyÚ.test_sort_index_and_reconstruction_doc_example s. ÿþ ÿþ zETestDataFrameSortIndex.test_sort_index_and_reconstruction_doc_examplec C s^ t dg t g g g¡d}t d ¡ d|jd< d|jd< W 5 Q R X | ¡ jj}|dksZt d S )Nr ©Úcolumnsr r )r Ú2)r Ú3T)
|
r r Ú from_productr Zassert_produces_warningÚlocr r r r )r r r! r" r" r# Ú-test_sort_index_non_existent_label_multiindex/ s
|
zDTestDataFrameSortIndex.test_sort_index_non_existent_label_multiindexc C s t tj dd¡tjddgddgddggd d
|
d gd d dgd}| ¡ }dd }|jd d
|
gd |¡ ¡ }tjddgddgddggd d
|
d gd }t
|
|j |¡ d S )Né r r r ÚbigÚsmallÚredZbluÚletterÚsizeÚcolor©ÚnamesZnearZfar©r r& c S s ddg| _ | S )NÚnewzÚnewar )Úgroupr" r" r# Úmy_funcD s
|
zFTestDataFrameSortIndex.test_sort_index_reorder_on_ops.<locals>.my_func©Úlevelr7 r6 ) r ÚnpÚrandomÚrandnr r) r ÚgroupbyÚapplyr Úassert_index_equalr )r r r9 r! r r" r" r# Ú test_sort_index_reorder_on_ops8 s þúþz5TestDataFrameSortIndex.test_sort_index_reorder_on_opsc C s ddgt jt jgt jdgddgg}t |¡}tt d¡ dd¡|tdd }tt d¡|d
|
}tt
|
d d d ddd dd ddddg ¡ddddddddddddg dt jddt jdddt jdddg ddd d d dd t jd d!ddg d" d#d$g¡}| ¡ }|j dd ddgd d
f }t ||¡ |j d%d&}|j dd ddgd d
f }t ||¡ |j d'd&}|j dddd gd d
f }t ||¡ | ¡ ¡ }| ¡ ¡ }t ||¡ | ¡ }|j dd ddg }t ||¡ |j d%d&}|j dd ddg }t ||¡ |j d'd&}|j dddd g }t ||¡ d S )(Né é r r r é r ÚABCDr5 r Z20121002Z20121007Z20130130Z20130202Z20130305Z20121207é iþ i i io éZ i8 i- ig i! é é
|
é é' r é- é" )ÚdateÚuser_idZ
|
whole_costZcostrO rP Úlast©Ú na_positionÚfirst)r< Únanr Ú from_tuplesr ÚarangeÚreshapeÚlistr ÚpdZ DatetimeIndexÚ set_indexr Úilocr r ZdropnaÚassert_series_equal)r ZtuplesÚmir ÚsÚdf2r! r r" r" r# Ú test_sort_index_nan_multiindexP sx "
|
ôÿô àÿ#Ý& z5TestDataFrameSortIndex.test_sort_index_nan_multiindexc C s¸ t ddtjddddgdtjdddddgdddd dddtjgd
|
}|jd d d d}t ddtjddddgdtjdddddgdddd dddtjgd
|
}t ||¡ |jdd}t dddtjdddgddtjddddgdtjddd dddgd
|
}t ||¡ |jd dd}t dddtjdddgddddtjddgddddd ddtjgd
|
}t ||¡ |jd ddd}t ddddtjddgdddddtjdgdtjdddd ddgd
|
}t ||¡ d S )Nr r é r, r é rG ©ÚAÚBr r Z quicksortTrQ )ÚkindÚ ascendingrS rT rR F)rg rh )r r< rU r r r )r r Z sorted_dfr r" r" r# Útest_sort_index_nan¡ s@ (þ(þ (þ (þ ÿ(þz*TestDataFrameSortIndex.test_sort_index_nanc C s t dddgdddgdddgtdd}| td¡jtdd}t dddgdddgdddgtd d}| td¡}t ||¡ d S )
|
Nr r r r Úabc©r r ÚcÚdÚbar: Úbca©r rY r[ r r r ©r r r! r r" r" r# Útest_sort_index_multi_indexÌ s "ÿ"ÿz2TestDataFrameSortIndex.test_sort_index_multi_indexc C sX t tj dd¡ddddgddddgd }|jddddg }|d }| ¡ }|jd
|
d }|d ksbt|}t ||¡ ||d k st| ¡ }|jd d
|
d }|d ks¤t|d d d
}t ||¡ |jd d
ddddgf }| ¡ }|jdd
|
d}|d ksút|}t ||¡ | ¡ }|jdd d
|
d}|d ks0t|j
|
d d
d d d
f }t ||¡ d S )Nr r r r re rf ÚCÚDr5 T)ÚinplaceF)rh ru éÿÿÿÿ)Úaxisru )rw rh ru ) r r< r= r> r* r r r r r r\ )r ÚframeÚ unorderedZa_valuesr Ú return_valuer r" r" r# Útest_sort_index_inplaceÚ s:
|
|
ÿ z.TestDataFrameSortIndex.test_sort_index_inplacec C sÜ t d¡ d¡}t t d¡d¡}t j d¡}| |¡}| |¡}t||t j d¡d}t |j
|
¡ |j
|
|j f¡}| |¡}| ddg¡}|jddgd }| |¡}t ||¡ |d
|
jddgd }t ||d
|
¡ d S ) Né rG éd )re rf rs re rf r r ©rh rs )r< rW ÚrepeatZtiler= Z permutationZtaker r> Zlexsortrf Úmaxre r[ r r r r] ) r re rf Zindexerr Z
|
ex_indexerr Zidfr! r" r" r# Ú#test_sort_index_different_sortorderÿ s
|
|
|
|
z:TestDataFrameSortIndex.test_sort_index_different_sortorderc C sþ t jdddgdddggtdd}tddgddgg|}|jddd }|}t ||¡ |jdd
|
gdd }|}t ||¡ |jd d
|
dgd }|jdd g }t ||¡ |jd
|
d dgd }|jdd g }t ||¡ |jd dgd }|jdd g }t ||¡ d S )Nr r ÚABCr3 r r re F©r; Zsort_remainingrf rs r: r )r rV rY r r r r r\ )r r^ r r! r r" r" r# Útest_sort_index_level s" " z,TestDataFrameSortIndex.test_sort_index_levelc C s t tjdddttd ttd¡d d¡}| ¡ }|j dd d
|
d d d g }t
|
||¡ |jdd}|j d d d d
|
d dg }t
|
||¡ d S )Nrb Úint64©ÚdtypeZaabbcaZcabrd rf r r r rG r r Fr~ ) r r< rW r rY Úastyper r[ r r\ r r rq r" r" r# Ú!test_sort_index_categorical_index1 s þÿû z8TestDataFrameSortIndex.test_sort_index_categorical_indexc C sè t t d¡ dd¡ddddgdddd gd
|
}|jddddg }|jd d }|}t ||¡ |jd d}|d d d
}t ||¡ |jd d
dddd gf }|jdd }t ||¡ |jdd d}|jd d
d d d
f }t ||¡ d S )NrE r r r r re rf rs rt r5 r ©rw Fr~ rv )rw rh ) r r< rW rX r* r r r r\ )r rx ry r! r r" r" r# Útest_sort_indexA s$
|
|
ý z&TestDataFrameSortIndex.test_sort_indexr; re r c C s t jdddgdddgdddggtdd}tddgddgddgg|d }t jdddgdddgdddggtdd}tddgddgddgg|d }|j|d
|
}t ||¡ t jdddgdddgdddggtdd}tddgddgddgg|d }|j|d d }t ||¡ d S ) Nr r r r r3 r rG rb r r: Fr )r rV rY r r r r )r r; r^ r Z expected_mir r! r" r" r# Útest_sort_index_multiindex] s& ÿ ÿ ÿ z1TestDataFrameSortIndex.test_sort_index_multiindexc C s´ t tj d¡}t t tj d¡¡}tjt tj d¡dddddgd}tj|||gdd d
|
d gd }|jd
|
d gd d ¡
|
¡ }t j ddddgdd}|j jd j}t ||¡ d S )Nr} éýÿÿÿç à¿r ç à?r )Zbinsr ÚYZX1ZX2)rw ÚkeysT)Zobserved)g Àr )r ç )r r )r ç @Úright)Úclosed)r r< r= r> ÚsignrZ ZcutÚconcatr? ZmeanZunstackr rV r& r Ú
|
categoriesr rA )r ÚyÚx1Zx2Úmodelr! r r" r" r# Ú test_sort_index_intervalindexv s $
|
ÿz4TestDataFrameSortIndex.test_sort_index_intervalindexru TFzAoriginal_dict, sorted_dict, ascending, ignore_index, output_indexr r r rG c C sx dddg}t ||d}t ||d} |||d}
|
|rH| ¡ } | jf |
|
n |jf |
|
} t | | ¡ t |t ||d¡ d S )Nr rG r r ©rh Ú ignore_indexru )r r r r r ) r ru Ú original_dictÚ sorted_dictrh r Ú output_indexZoriginal_indexr Ú expected_dfÚkwargsÚ result_dfr" r" r# Ú test_sort_index_ignore_index
s
|
ý z3TestDataFrameSortIndex.test_sort_index_ignore_indexr c C s| t ddddgitdddd}|jd ||d
|
}|r6|}|rNt ddddgi}n t ddddgitdddd}t ||¡ d S ) Nr r r r r rv éþÿÿÿr Fr )r r r r r )r ru r r r! r r" r" r# Útest_respect_ignore_index¥ s ÿ z0TestDataFrameSortIndex.test_respect_ignore_indexr )ZM1ZM2©r r ©r r ÚABr3 c C s t jddgtdd}t||d}t||d} |||d}
|
|rT| ¡ } | jf |
|
n |jf |
|
} t | | ¡ t |t||d¡ d S )Nr¨ r© rª r3 r r )r rV rY r r r r r ) r ru r r rh r r¡ r^ r r¢ r£ r¤ r" r" r# Ú(test_sort_index_ignore_index_multi_index· s & ý z?TestDataFrameSortIndex.test_sort_index_ignore_index_multi_indexc C sÈ t tdtjddddddgdddgddddddddgd }| d
|
d g¡ ¡ }t d d ddddgdgttdddgdddgdd
|
ddddggddddddgddddddggd
|
d gdd}t ||¡ d S )Nrb r r rl T)r Úorderedr r )r Úl1Úl2r r® r rG r r Úcategory)r r¬ Únamer )r r r4 r% )
|
r ÚrangerZ Z Categoricalr[ r r r r r rq r" r" r# Ú&test_sort_index_categorical_multiindexð s: ýùÿ ûø
|
ôýz=TestDataFrameSortIndex.test_sort_index_categorical_multiindexc C sì t ddgddggtdd}t ddgddgddgddggt ddddg¡d}|j ¡ s\tt ddgddgddgddggt d d
|
gtdg¡d}| ¡ }|jj s¦tt
|
||¡ t ddgddgddgddggtd d
|
gd d ggd d ddgd dd dggdd}| ¡ }|j ¡ stt
|
||¡ t j ||gd
|
d gd}| ¡ }|jj sNtt
|
||¡ t ddgddggtjddgddgdd}|jjt |jjd ¡dd|_|jj r²t|jdd}|jj sÌt|jddd}|jj sètd S )Nr r Úabr )r r )r r )çé?r )r´ r r r´ r r r r )r rb é )r z20160811 12:00:00)r z20160809 12:00:00r ÚDater3 )r& r: r )rw r; )r rY r rV r r r r) r r r r rZ r r& Z
|
set_levelsÚ to_datetimer )r r r r! Z concattedr" r" r# Ú"test_sort_index_and_reconstruction sZ
|
ÿþþ ÿþ þþ ÿ z9TestDataFrameSortIndex.test_sort_index_and_reconstructionc C sz |}| ¡ }t t|¡|_|d jdd}|jj|jjks@t| ¡ }|jddd}|d ksbtt ||jdd¡ d S )Nre r r: T)r; ru )
|
r r< rW Úlenr r r4 r r r )r Ú multiindex_dataframe_random_datarx r Za_sortedÚrsrz r" r" r# Útest_sort_index_level2N s z-TestDataFrameSortIndex.test_sort_index_level2c C s´ t t d¡gd ¡}ttj d¡ d¡|d}|jdd}|j j
|
dksLt t t d¡gd ¡}ttj d¡ d¡|d}|jdd}|j j |j j k ¡ s t |j j
|
dks°t d S )Ni r r
r r r: Úint32)r Z from_arraysr< rW r r= r> r r r Z_lexsort_depthr ZdtypesÚvaluesÚall)r r r r! r" r" r# Ú'test_sort_index_level_large_cardinalityb s z>TestDataFrameSortIndex.test_sort_index_level_large_cardinalityc C s8 |}ddg|j _|jdd}|jdd}t ||¡ d S )NrT Úsecondr: r )r r4 r r r )r rº rx r! r r" r" r# Ú test_sort_index_level_by_namet s
|
z4TestDataFrameSortIndex.test_sort_index_level_by_namec C s |}|j dd}| ¡ }d|d< |j dd}t ||jdgdd¡ |j}|j ddd}d|d< |j ddd}t |jdgdd|jdgdd¡ d S )Nr r: ÚbarÚfoor )r; rw )rÄ Zthree)r r r r ZdropÚT)r rº rx Z sorted_beforer Z sorted_afterZdftr" r" r# Útest_sort_index_level_mixed| s þz2TestDataFrameSortIndex.test_sort_index_level_mixedc C s$ |}| ¡ }|jj|jjks td S ©N)r r r4 r )r rº rx r! r" r" r# Ú test_sort_index_preserve_levels s z6TestDataFrameSortIndex.test_sort_index_preserve_levelsz gen,extrag ð?r g @g @g @Z20130101Z20130103Z20130102Z20130105Z20130104Z1oneZ3oneZ2oneZ5oneZ4onec C s$ t j d¡ t j dd¡}t dd |D ¡}t|td|d}tj |tdtdt d |fg¡dgd
|
d }t
|
| ¡ d ¡ d gkst | ¡ jd
|
d }|jd d
dd d
|
dgf }t ||¡ |jd
|
d }|jd d
dd d
|
ddgf }t ||¡ | ¡ }d|d |f< |jd
|
d }t ||¡ d S ) Nr r r c S s g | ] }d |fqS )r/ r" ©Ú.0Úir" r" r# Ú
|
<listcomp>« s zOTestDataFrameSortIndex.test_sort_index_multilevel_repr_8017.<locals>.<listcomp>Údefr5 Zworldr/ r r r )r< r= Úseedr> r rV r rY rZ r ÚstrÚ
|
splitlinesÚsplitr r r r\ r r ) r ÚgenÚextraÚdatar& r r` r! r r" r" r# Ú$test_sort_index_multilevel_repr_8017 s2 ýþ÷ z;TestDataFrameSortIndex.test_sort_index_multilevel_repr_8017r r r rl rÏ )Úidz pd.Intervalc C s t dtt|it||ddd}|j |jjd d d
¡|_| ¡ }t dttt|it|d d d
|d d d
ddd}t
|
||¡ d S )NrÄ T)rÔ r r¬ r rv ) r r± r¹ r r Zreorder_categoriesr r Úreversedr r )r r r r! r r" r" r# Ú test_sort_index_with_categoriesÑ s& ÿþ ÿþz6TestDataFrameSortIndex.test_sort_index_with_categoriesrh NÚTruec sb t t d¡}t|j fddt D |_d}tjt|d |j d|dd W 5 Q R X d S ) Né@ c s g | ]}| d qS )r r" rÉ ©Úlengthr" r# rÌ ù s zUTestDataFrameSortIndex.test_sort_index_ascending_bad_value_raises.<locals>.<listcomp>z+For argument "ascending" expected type bool©Úmatchr rT )rw rh rS )
|
r r< rW r¹ r r± ÚpytestÚraisesÚ
|
ValueErrorr )r rh r rÞ r" rÛ r# Ú*test_sort_index_ascending_bad_value_raisesí s
|
|
zATestDataFrameSortIndex.test_sort_index_ascending_bad_value_raisesc C sX t dddgdddgdtjdddd }t d
|
d ¡ | ¡ }W 5 Q R X t ||¡ d S ) Nr r r r rG )Zcol1Zcol2Z2020)Zperiodsr zmode.use_inf_as_naT)r rZ Z
|
date_rangeZoption_contextr r r ©r r r! r" r" r# Ú test_sort_index_use_inf_as_naþ s þz4TestDataFrameSortIndex.test_sort_index_use_inf_as_na)TFc C s t ddddddgitjdddddgd d
|
gd d }|jd |d}t ddddddgitjdddddgd d
|
gd d }t ||¡ d S )NZlegsr r )ÚmammalZdog)ÚbirdZduck)rå Zhorse)ræ Zpenguin)rå ZkangarooÚclassZanimalr3 r )r r )r; rh )r r rV r r r )r rh r r! r r" r" r# Ú test_sort_index_ascending_tuple s< ÿûøü ÿûøüz6TestDataFrameSortIndex.test_sort_index_ascending_tuple))Ú__name__Ú
|
__module__Ú __qualname__r$ r+ rB ra ri rr r{ r r r r rß ÚmarkZ parametrizer r r¥ r§ r rV rY r« r² r¸ r¼ rÀ r rÆ rÈ r rÕ ÚparamrZ ZIntervalrØ râ rä rè r" r" r" r# r
|
sÎ Q+%
|
&&&&üþ ûûûûêþ! "<üù ôþ
|
* þþþ
|
|
ýþ
|
|
|
þr
|
c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestDataFrameSortIndexKeyc C sÞ t dddgdddgdddgtdd td¡}|jtddd d
|
}t dddgdddgdddgtd d td¡}t ||¡ |jtdd d d
|
}t dddgdddgdddgtd d td¡}t ||¡ d S )Nr r r r rj rk Úacc S s | S rÇ r" ©Úxr" r" r# Ú<lambda>: ó zETestDataFrameSortIndexKey.test_sort_multi_index_key.<locals>.<lambda>©r; Úkeyro c S s | S rÇ r" rð r" r" r# rò A ró Zacbrp rq r" r" r# Útest_sort_multi_index_key4 s& "ÿþ"ÿþ "ÿþz3TestDataFrameSortIndexKey.test_sort_multi_index_keyc C s® t tjdddtdd}| ¡ }|jdddd d
|
d g }t ||¡ |jd d d}|jdd d
|
ddd g }t ||¡ |jdd dd}|jd dddd d
|
g }t ||¡ d S )Nrb r
r ZaaBBcar r r r r rG r c S s
|
| j ¡ S rÇ ©rÏ Úlowerrð r" r" r# rò O ró z?TestDataFrameSortIndexKey.test_sort_index_key.<locals>.<lambda>©rõ c S s
|
| j ¡ S rÇ r÷ rð r" r" r# rò S ró F)rõ rh )r r< rW rY r r\ r r rq r" r" r# Útest_sort_index_keyH s z-TestDataFrameSortIndexKey.test_sort_index_keyc C s| t tjdddtjdddd}| ¡ }t ||¡ |jdd d}|jdd }t ||¡ |jd
|
d d}t ||¡ d S ) Nrb r
r r c S s | S rÇ r" rð r" r" r# rò ] ró zCTestDataFrameSortIndexKey.test_sort_index_key_int.<locals>.<lambda>rù Fr~ c S s d| S )Nr r" rð r" r" r# rò a ró )r r< rW r r r rq r" r" r# Útest_sort_index_key_intW s z1TestDataFrameSortIndexKey.test_sort_index_key_intc C sÚ t dddgdddgtddddgd td¡}|jdd d
|
d }t dddgdddgtd dddgd td¡}t ||¡ |jtdd d
|
d }t dddgdddgtd dddgd td¡}t ||¡ d S )Nrf r rs r r rj r rk c S s
|
| j ¡ S rÇ r÷ rð r" r" r# rò j ró zITestDataFrameSortIndexKey.test_sort_multi_index_key_str.<locals>.<lambda>rô Zbacc S s | j dkr| j ¡ S | S )N)r rl )r° rÏ rø rð r" r" r# rò s ró rp rq r" r" r# Ú test_sort_multi_index_key_strd s, "ÿþ"ÿþ þ"ÿþz7TestDataFrameSortIndexKey.test_sort_multi_index_key_strc C s@ t ddddgi}tjtdd |jdd d W 5 Q R X d S )
|
Nre r r r zchange the shaperÝ c S s | d d
S )Nr r" rð r" r" r# rò ~ ró zFTestDataFrameSortIndexKey.test_changes_length_raises.<locals>.<lambda>rù )r rß rà rá r )r r r" r" r# Útest_changes_length_raises{ s z4TestDataFrameSortIndexKey.test_changes_length_raisesc C sJ t dd tddD t ddgddgg¡d}|jdd}t ||¡ d S ) Nc S s, i | ]$}|t jd d d d gt dd ¡dqS )r Úfloat64r )rZ ÚarrayZ SparseDtyperÉ r" r" r# Ú
|
<dictcomp> s ÿ zVTestDataFrameSortIndexKey.test_sort_index_multiindex_sparse_column.<locals>.<dictcomp>r r r r r r: )r r± r r) r r r rã r" r" r# Ú(test_sort_index_multiindex_sparse_column s þû zBTestDataFrameSortIndexKey.test_sort_index_multiindex_sparse_columnN) ré rê rë rö rú rû rü rý r r" r" r" r# rî 3 s rî )Únumpyr< rß ZpandasrZ r r r r r r r r Zpandas._testingZ_testingr r
|
rî r" r" r" r# Ú<module> s (
|
'
|
