U
|
ý°dò ã
|
@ sò d dl Z d dlZd dlZd dlZd dlmZmZmZm Z m
|
Z
|
d dl m Z G dd dZG dd dZejdd Zejd gd d
|
ggd d d Zejd
|
d gddgd
|
d gdd gddgd
|
d ggd dd Zejddgd dd ZG dd dZdS )é N)Ú CategoricalÚ DataFrameÚNaTÚ TimestampÚ
|
date_rangec @ sP e Zd Zej dejeg¡dd Z dd Z
|
dd Z dd Z d
|
d Z d d Zdd Zdd Zej ddddddddddd d d g d d ggddddddddd d d dg d d ggddd d d dddddddg d d ggdd d d ddddddddg d d ggg¡ej d!d"d#g¡d$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zej d6d d g¡ej d7d8dd9dgid8dd9dgid ddd9gfd8dd9dgid8dd9dgid d9ddgfdd9dgd9ddgd:dd9dgddd9gd:d ddd9gfdd9dgd9ddgd:dd9dgddd9gd:d d9ddgfg¡d;d< Zd=d> Zd?d@ Z dAdB Z dCdD Z dEdF Z dGS )HÚTestDataFrameSortValuesÚdtypec C sZ t tdddgdddgd¡}t j||dd}t d ¡ |j|j ¡ d W 5 Q R X d S )NÚaÚbÚc)Ú
|
categoriesT)r Úsparse©Úby) ÚpdÚSeriesr Z get_dummiesÚtmZassert_produces_warningÚ sort_valuesÚcolumnsÚtolist)Úselfr ÚserÚdf© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_sort_values.pyÚ"test_sort_values_sparse_no_warning s z:TestDataFrameSortValues.test_sort_values_sparse_no_warningc C s t dddgdddgdddggdddgtdd }|jd
|
d }|d
|
¡ j}|j|j| }t ||¡ |jd
|
d d }|d d d
}|j|j| }t ||¡ |jd
|
d d }t ||¡ |jd
|
gd gd }t ||¡ |jddgd }|jdddg }t ||¡ |jddgd d }t ||d d d
¡ |jdd
|
gdd gd }t ||¡ d}t j
|
t |d |jd
|
dgddd W 5 Q R X |jddd}|}t ||¡ |jddd d}|j ddd
|
gd}t ||¡ |jddgdd}|j dd
|
dgd}t ||¡ |jddgddd gd}t ||¡ |jddgdd d}|j ddd
|
gd}t ||¡ d}t j
|
t |d |jd
|
dgddgd d W 5 Q R X d S )Né é é r é é é ÚABC©Úindexr ÚAr F©r Ú ascendingéÿÿÿÿÚBÚCTz)No axis named 2 for object type DataFrame©Úmatch©r ÚaxisÚinplace©r r. )r r. r' ©r r z/Length of ascending \(5\) != length of by \(2\)) r Úlistr ÚargsortÚvaluesÚlocr$ r Úassert_frame_equalÚpytestÚraisesÚ
|
ValueErrorÚreindex)r ÚframeÚ sorted_dfÚindexerÚexpectedÚmsgr r r Útest_sort_values sX ÿ z(TestDataFrameSortValues.test_sort_valuesc C s@ t dddddddgi}|jg d}t ||¡ ||k s<td S ) Nr r r r r r r! r )r r r r6 ÚAssertionError©r r> Úresultr r r Ú test_sort_values_by_empty_listZ s z6TestDataFrameSortValues.test_sort_values_by_empty_listc C s* t tj dd¡ddddgddddgd }| ¡ }|jdd
|
d }|d ksJt|jdd }t ||¡ | ¡ }|jddd
|
d }|d kst|jddd}t ||¡ | ¡ }|jddd
|
d}|d ksÄt|jddd}t ||¡ | ¡ }|jddgdd
|
d}|d kst|jddgdd}t ||¡ d S )Nr r r r r% r) r* ÚDr# T©r r/ r r- r0 F)r r' r/ r& © r ÚnpÚrandomÚrandnÚcopyr rA r r6 )r r; r< Ú return_valuer> r r r Útest_sort_values_inplacea s:
|
|
ÿ ÿz0TestDataFrameSortValues.test_sort_values_inplacec C s t d¡ d¡}t t d¡d¡}t |¡ t |¡ t||t j d¡d}|jddgd}t |d |d f¡}|
|
|¡}t ||¡ |jddgdd }t |d j dd
|
|d j dd
|
f¡}|
|
|¡}t ||¡ |jddgd}t |d |d f¡}|
|
|¡}t ||¡ d S ) Nr é éd )r% r) r* r% r) r Fr& ©r' )rH ÚarangeÚrepeatZtilerI Úshuffler rJ r ZlexsortÚtaker r6 Zrank)r r% r) r; rC r= r> r r r Ú test_sort_values_multicolumn s&
|
|
|
ÿ
|
|
z4TestDataFrameSortValues.test_sort_values_multicolumnc C s t t ddg¡t ddg¡d}|d tj¡|d< | ddg¡}t t ddg¡t ddg¡dt ddg¡d }t ||¡ d S )
|
Nl (<9zü l i J³H r r ©r r
|
r r
|
r ©r$ )
|
r r r ZastyperH Zuint64r ÚIndexr r6 ©r r rC r> r r r Ú#test_sort_values_multicolumn_uint64 s þÿ þ ûz;TestDataFrameSortValues.test_sort_values_multicolumn_uint64c C s t ddtjddddgdtjdddddgd}t tjddddddgdddtjdddgddd d
|
ddddgd }|jd gd d}t ||¡ t tjddddddgddddtjddgddddddd d
|
gd }|jd gd dd}t ||¡ |jdd gd}|jddd d}t ||¡ t ddddddtjgddtjddddgdd
|
d dddddgd }| d dg¡}t ||¡ t tjddddddgdddtjdddgddd
|
d ddddgd }|jd dgd d}t ||¡ t tjddddddgdddtjdddgddd d
|
ddddgd }|jd dgdd gd d}t ||¡ t ddddddtjgdddtjdddgdddddd
|
d dgd }|jd dgd dgdd}t ||¡ d S )Nr r r! é r é r ©r% r) r r rW r% Úfirst©Ú na_positionF)r` r' r) r1 )r r. r` ©r' r` Úlast)r rH Únanr r r6 r: ©r r r> r< r r r Útest_sort_values_nan¯ sT (ÿ(þ (þ (þ (þ (þ (þz,TestDataFrameSortValues.test_sort_values_nanc C sH t ddgddgddgddggddgd }|jdd
|
d d }t ||¡ d S ) Nr r^ Úsecondr r r
|
Zsort_colÚorderr1 Ú mergesortF)r Úkindr' ©r r r r6 )r r r< r r r Ú'test_sort_values_stable_descending_sortè s þz?TestDataFrameSortValues.test_sort_values_stable_descending_sortz expected_idx_non_na, ascendingr r r r r r[ r! r\ é é
|
é é TFr` r^ rb c C s¨ t ddtjddddddddtjtjddgdtjdddddddddtjtjddgd }|d
|
krld d dg| n |dd d g }| |¡}|jd dg||d}t ||¡ d S )Nr r r! r[ r r\ r r r] r^ é é r% r) ra )r rH rc rT r r r6 )r Zexpected_idx_non_nar' r` r Z expected_idxr> r< r r r Ú(test_sort_values_stable_multicolumn_sortñ s &&þÿ
|
ÿ ý
|
ÿz@TestDataFrameSortValues.test_sort_values_stable_multicolumn_sortc C sJ t dtt ddddgd¡ddi}| ¡ }|jdd d
|
}t ||¡ d S ) NÚxr r r r r T)Úorderedrh )ri )r r rH rR rK r r r6 rd r r r Ú"test_sort_values_stable_categorial s $z:TestDataFrameSortValues.test_sort_values_stable_categorialc
|
C sà t dddddddddg dgtd d
|
d d }d d dD }|d d d
|dd d
|d< d|d< d|d< |jdd}|jdgd}t ||¡ |jdd}|jdgd}t ||¡ |jdd}|jddgd}t ||¡ d S )Nr r
|
r ÚdÚeÚfÚgr% Z20130101r\ )Zperiods©r r$ c S s g | ] }t |qS r ©r ©Ú.0rs r r r Ú
|
<listcomp>, s ÿzFTestDataFrameSortValues.test_sort_values_datetimes.<locals>.<listcomp>) z
|
2004-02-11z
|
2004-01-21z
|
2004-01-26z
|
2005-09-20z
|
2010-10-04z
|
2009-05-12z
|
2008-11-12ú
|
2010-09-28r r r r) ç @r* g @ÚA1r )r r r r r6 )r r ZdtsZdf1Zdf2r r r Útest_sort_values_datetimes% s(
|
ýþ z2TestDataFrameSortValues.test_sort_values_datetimesc C s| |d }| ¡ }|rB|jdd t ||d ¡ ¡ t ||¡ n&tjtdd |jdd W 5 Q R X | ¡ }| ¡ d S )Nr% T)r/ zThis Series is a viewr+ )rK r r Zassert_series_equalr6 r7 r8 r9 )r Z float_frameÚusing_copy_on_writeÚsZfloat_frame_origÚcpr r r Ú4test_sort_values_frame_column_inplace_sort_exceptionK s zLTestDataFrameSortValues.test_sort_values_frame_column_inplace_sort_exceptionc C s\ dt tjf}d}t||dddgd}t|d d d
|d d d
dddgdd gd
|
}|jddgd d }t ||¡ |jddgd d }t ||¡ |jddgdd}t ||¡ ttdtg|dddgd}tttdg|d d d
dddgdd gd
|
}|jddgd d }t ||¡ |jddgd d }t ||¡ |jddgdd}t ||¡ d S )Nr )r g`¹×ÿÿïÿ)ÚintÚfloatr r r1 r( r r rz rb r_ r^ FrP ú
|
2016-01-01)Údatetimer r )r r Ú_valuer r r r6 r )r Z
|
int_valuesZ float_valuesr Z df_reversedZ df_sortedr r r Ú)test_sort_values_nat_values_in_int_column^ s@ ÿ ý þý zATestDataFrameSortValues.test_sort_values_nat_values_in_int_columnc C s dd ddt jdfD }dd dD }t||ddd d
|
d gd }d d dddt jfD }dd dD }t||dd d dd
|
gd }|jddgd}t ||¡ d S )Nc S s g | ] }t |qS r r{ r| r r r r~ s z9TestDataFrameSortValues.test_sort_nat.<locals>.<listcomp>r ú
|
2015-01-01c S s g | ] }t |qS r r{ r| r r r r~ s ÿ©ú
|
2017-01-01ú
|
2014-01-01r r rV r r r r rW c S s g | ] }t |qS r r{ r| r r r r~ s c S s g | ] }t |qS r r{ r| r r r r~ s ÿ©r r r r r r
|
r ©rH rc r r r r6 ©r Úd1Zd2r Zd3Zd4r> r< r r r Ú test_sort_nat s þþz%TestDataFrameSortValues.test_sort_natc C st dddg}dddg}t jt jg}ddg}d }d
|
}d }t|d d }t|d d } t|}
|
t|tdt jdt jdg|d di} | j|d |d} t|t|| |d di|| d} t | | ¡ | j|d |d} t|t|| |d di|| d} t | | ¡ | j|d|d} t|t|| |d di|
|
| d} t | | ¡ | j|d|d} t|t|| |d di| |
|
d} t | | ¡ d S )Nr% r) r* r r r r r r^ rb r T)Úreverse©r rt ©r r' r` rW F)rH rc Úsortedr r r r r6 )r r Zcategory_indicesZ list_of_nansZ
|
na_indicesZna_position_firstZna_position_lastZ column_nameZreversed_categoriesZreversed_category_indicesZreversed_na_indicesr rC r> r r r Ú,test_sort_values_na_position_with_categories§ s¦
|
|
ÿÿÿ ÿ ÿÿú ÿ ÿÿú ÿ ýÿø ÿ ýÿø zDTestDataFrameSortValues.test_sort_values_na_position_with_categoriesc C s dd ddt jdfD }dd dD }t||ddd d
|
d gd }d d dddt jfD }dd dD }t||dd d dd
|
gd }|jddgd}t ||¡ d S )Nc S s g | ] }t |qS r r{ r| r r r r~ s z@TestDataFrameSortValues.test_sort_values_nat.<locals>.<listcomp>r r c S s g | ] }t |qS r r{ r| r r r r~ s ÿr rV r r r r rW c S s g | ] }t |qS r r{ r| r r r r~ s c S s g | ] }t |qS r r{ r| r r r r~ s ÿr r r
|
r r r r r r Útest_sort_values_natý s þþz,TestDataFrameSortValues.test_sort_values_natc C sX t dtdtjdtjdgdddgddi}tjtdd |jdd d
|
d W 5 Q R X d S ) Nr r% r) r* Tr z!invalid na_position: bad_positionr+ FZ bad_positionr )r r rH rc r7 r8 r9 r ©r r r r r Ú3test_sort_values_na_position_with_categories_raises s ýÿÿ
|
zKTestDataFrameSortValues.test_sort_values_na_position_with_categories_raisesr/ z6original_dict, sorted_dict, ignore_index, output_indexr% r r] c
|
C st t |}t ||d}||d}|r@| ¡ } | jdddi| n|jdddi|} t | |¡ t |t |¡ d S )NrW )Ú ignore_indexr/ r% r' F)r% )r% )r rK r r r6 )
|
r r/ Z original_dictZ sorted_dictr Z output_indexr r> ÚkwargsZ result_dfr r r Ú test_sort_values_ignore_index s
|
z5TestDataFrameSortValues.test_sort_values_ignore_indexc C sF t dddddgt ddddd g¡d
|
}| d d g¡}t ||¡ d S ) Nr r r r z2010-01-01 09:00:00z2010-01-01 09:00:01z2010-01-01 09:00:02z2010-01-01 09:00:03r )r% Údater% r¡ )r r Z DatetimeIndexr r r6 rB r r r Ú(test_sort_values_nat_na_position_defaultC s ûÿþÿz@TestDataFrameSortValues.test_sort_values_nat_na_position_defaultc C sÎ t tj dd¡dddgd}|d d |d< |d }|sLt|jjdksLt|jdd |rd
|
|j d < |j d |d d kst|j d d
|
ksÊtn6d
|
|j
|
d < |j d |d d ks¸t|j d d
|
ksÊtd S ) Nr r r% r) r* r1 r rE r éc r )r r ) r rH rI rJ ÚlenZ_mgrÚblocksrA r Úilocr4 )r Zusing_array_managerr r r r r r Útest_sort_values_item_cacheV s
|
|
z3TestDataFrameSortValues.test_sort_values_item_cachec C s@ t td}t|g|d}|j|jd ddd}t ||¡ d S )Né r1 r r T)r. r )r2 Úranger r r$ r r6 )r r4 r> r r r r Útest_sort_values_reshapingi s z2TestDataFrameSortValues.test_sort_values_reshapingc C sD t ddddgi}| ¡ }|jg dd}t ||¡ |d ks@td S )Nr r r r TrF )r rK r r r6 rA )r r r> rC r r r Ú test_sort_values_no_by_inplaceq s
|
z6TestDataFrameSortValues.test_sort_values_no_by_inplacec C sR t ddgddgdddgd}|jd d
|
d }t ddgddgd}t ||¡ d S ) Nrm rN r r r] r r rW r% T)r r rj rY r r r Ú"test_sort_values_no_op_reset_indexy s z:TestDataFrameSortValues.test_sort_values_no_op_reset_indexN) Ú__name__Ú
|
__module__Ú __qualname__r7 ÚmarkÚ parametrizerH Zuint8Úboolr r@ rD rM rU rZ re rk rr ru r r r r r r r r r¢ r§ rª r« r¬ r r r r r st
|
> 9 þþþþóþ &6V$$üü÷þ r c @ sn e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z e j dddg¡dd ZdS )ÚTestDataFrameSortKeyc C s, t tj dd¡ddddgddddgd }| ¡ }|jdd
|
|d }|d ksLt|jd|d }t ||¡ | ¡ }|jddd
|
|d }|d kst|jdd|d}t ||¡ | ¡ }|jddd
|
|d}|d ksÎt|jdd|d}t ||¡ | ¡ }|jddgdd
|
|d |jddgd|d}t ||¡ d S )Nr r r r r% r) r* rE r# T)r r/ Úkey©r r´ )r r. r/ r´ )r r. r´ F)r r' r/ r´ )r r' r´ rG )r Ú sort_by_keyr; r< rL r> r r r Ú test_sort_values_inplace_key sN
|
|
ÿ ÿ ÿ ÿz1TestDataFrameSortKey.test_sort_values_inplace_keyc C s¸ t t ddtjddtjg¡}| d¡}|jddddddg }t ||¡ |jddd d }|jddddddg }t ||¡ |jdd
|
d d d }|jddddddg }t ||¡ d S ) Nr r r r r r c S s | d S )Nr r ©rs r r r Ú<lambda>« ó z;TestDataFrameSortKey.test_sort_values_key.<locals>.<lambda>©r´ c S s | S ©Nr r¸ r r r r¹ ¯ rº F©r´ r' ©r rH Úarrayrc r r¦ r r6 rY r r r Útest_sort_values_key¤ s
|
z)TestDataFrameSortKey.test_sort_values_keyc
|
C sä t t ddtjddtjg¡t ddtjddtjg¡d}|jddd d }|jd
|
dd dddg }t ||¡ |jdd gd d d}|jdd
|
d dddg }t ||¡ |jdd gdd dd}|jdd d
|
dddg }t ||¡ d S )Nr r r r rV r c S s | S r¼ r r¸ r r r r¹ » rº z>TestDataFrameSortKey.test_sort_values_by_key.<locals>.<lambda>r» r r r
|
c S s | S r¼ r r¸ r r r r¹ ¿ rº rµ c S s | S r¼ r r¸ r r r r¹ Ã rº F)r r´ r' r¾ rY r r r Útest_sort_values_by_key³ s þÿ z,TestDataFrameSortKey.test_sort_values_by_keyc
|
C s t t ddtjddtjg¡t ddtjddtjg¡d}dd }|jd|d }|jd
|
dd dddg }t ||¡ |jdg|d }|jd
|
dd dddg }t ||¡ |jd |d }|jdd
|
d dddg }t ||¡ |jdd g|d }|jd
|
dd dddg }t ||¡ d S ) Nr r r r rV c S s | j dkr| S | S d S )Nr )Úname©Úcolr r r r´ Ï s
|
zATestDataFrameSortKey.test_sort_values_by_key_by_name.<locals>.keyr rµ r r r
|
r¾ )r r r´ rC r> r r r Ú test_sort_values_by_key_by_nameÇ s$ þÿ z4TestDataFrameSortKey.test_sort_values_by_key_by_namec C s t t ddgddgg¡}| d¡}|d d d
}t ||¡ |jddgdd d }t ||¡ |jddgd
|
d d d }|jdd d d d }t ||¡ d S )NZhelloÚgoodbyeÚHellor r( r c S s
|
| j ¡ S r¼ ©ÚstrÚlowerrà r r r r¹ ì rº zBTestDataFrameSortKey.test_sort_values_key_string.<locals>.<lambda>r» c S s
|
| j ¡ S r¼ rÈ rà r r r r¹ ð rº Fr½ c S s
|
| j ¡ S r¼ rÈ rà r r r r¹ ò rº )r rH r¿ r r r6 rY r r r Útest_sort_values_key_stringå s
|
ÿz0TestDataFrameSortKey.test_sort_values_key_stringc C s, t t g ¡}|jd|d |j|d d S )Nr r» )r rH r¿ r Z
|
sort_index)r r¶ r r r r Útest_sort_values_key_emptyõ s z/TestDataFrameSortKey.test_sort_values_key_emptyc C sB t ddddgi}tjtdd |jddd d W 5 Q R X d S )
|
Nr% r r r zchange the shaper+ c S s | d d
S )Nr r r¸ r r r r¹ þ rº zATestDataFrameSortKey.test_changes_length_raises.<locals>.<lambda>r» ©r r7 r8 r9 r r r r r Útest_changes_length_raisesû s z/TestDataFrameSortKey.test_changes_length_raisesc C sr t ddgddgd}|jddd d}|d d d
}t ||¡ |jdd
|
d d}|d d d
}t ||¡ d S ) NrÇ rÆ r r ©r r c S s
|
| j ¡ S r¼ rÈ rà r r r r¹ rº z@TestDataFrameSortKey.test_sort_values_key_axes.<locals>.<lambda>r» r( c S s | S r¼ r rà r r r r¹ rº rj rY r r r Útest_sort_values_key_axes s z.TestDataFrameSortKey.test_sort_values_key_axesc C s t ddgddgd}|jddd dd}|jd d
d d d
f }t ||¡ |jdd
|
d dd}|jd d
d d d
f }t ||¡ d S ) NrÇ r rÆ r rÏ c S s
|
| j ¡ S r¼ rÈ rà r r r r¹ rº zETestDataFrameSortKey.test_sort_values_key_dict_axis.<locals>.<lambda>)r´ r. r( c S s | S r¼ r rà r r r r¹ rº )r r r5 r r6 rY r r r Ú test_sort_values_key_dict_axis s z3TestDataFrameSortKey.test_sort_values_key_dict_axisrt TFc s~ dddg t dddgdddgd} fdd}|jdd g|d
|
}t dddgdddgdt d dd g¡d }t ||¡ d S )Nr r
|
r r )rs Úyc s" | j dkr t t| d¡S | S )NrÒ r )r r r r r» r r r Úsorter s
|
|
ÿzNTestDataFrameSortKey.test_sort_values_key_casts_to_categorical.<locals>.sorterrs rÒ rµ r r rW )r r r rX r r6 )r rt r rÓ rC r> r r r Ú)test_sort_values_key_casts_to_categorical s
|
ÿz>TestDataFrameSortKey.test_sort_values_key_casts_to_categoricalN)r r® r¯ r· rÀ rÁ rÅ rË rÌ rÎ rÐ rÑ r7 r° r± rÔ r r r r r³ s " r³ c
|
C s@ t ddddddgddddddgt ddd¡ddd d ddgd
|
S ) Nr r
|
r r r! r r( ZoneZtwo)ÚouterÚinnerr% ©r) r )r rH rQ r r r r Údf_none+ s üÿrØ rÕ rÖ )Úparamsc C s | j }| |¡S r¼ )ÚparamÚ set_index)ÚrequestrØ Úlevelsr r r Údf_idx7 s rÞ r% r× c C s | j S r¼ ©rÚ ©rÜ r r r Ú
|
sort_names= s rá TFc C s | j S r¼ rß rà r r r r' M s r' c @ sB e Zd Zdd Zdd Zdd Zej ddd d
|
d g¡d d Z dS )ÚTestSortValuesLevelAsStrc C s> |j j}|j||dd |¡}|j||dd}t ||¡ d S )Nr ©r r' r. )r$ Únamesr rÛ r r6 ©r rØ rÞ rá r' rÝ r> rC r r r Ú&test_sort_index_level_and_column_labelS s ÿþz?TestSortValuesLevelAsStr.test_sort_index_level_and_column_labelc C sB |j j}|j||dd |¡j}|jj||dd}t ||¡ d S )Nr rã r )r$ rä r rÛ ÚTr r6 rå r r r Ú&test_sort_column_level_and_index_labele s ÿÿz?TestSortValuesLevelAsStr.test_sort_column_level_and_index_labelc C sB t ddddgi}d}tjt|d |jddd W 5 Q R X d S ) NrE é rl r¨ z?For argument "ascending" expected type bool, received type str.r+ ÚFalser& rÍ )r r r? r r r Ú3test_sort_values_validate_ascending_for_value_error{ s zLTestSortValuesLevelAsStr.test_sort_values_validate_ascending_for_value_errorr' Fr r Tc C s` t ddddgi}|d ¡ j}|s2|d d d
}|j|j| }|jd|d}t ||¡ d S )NrE ré rl r¨ r( r& )r r3 r4 r5 r$ r r r6 )r r' r r= r> rC r r r Ú.test_sort_values_validate_ascending_functional s zGTestSortValuesLevelAsStr.test_sort_values_validate_ascending_functionalN)
|
r r® r¯ ræ rè rë r7 r° r± rì r r r r râ R s
|
râ )rI ÚnumpyrH r7 Zpandasr r r r r r Zpandas._testingZ_testingr r r³ ZfixturerØ rÞ rá r' râ r r r r Ú<module> s> t +
|
|
øÿ
|
|
|
