U
|
ý°d¨o ã @ s d dl m Z d dlmZ d dlZd dlZd dlmZmZ d dl Z
|
d dl m Z m Z m Z mZmZmZmZmZmZmZmZmZ d dlmZ e ¡ dd ZG dd d Z ej d
|
d d ge fe
|
j d ddde
|
j!ddfg¡dd Z"dd Z#dd Z$dd Z%dd Z&d d Z'd d Z(d d! Z)dS )"é )Údatetime)ÚproductN)Úis_float_dtypeÚis_integer_dtype) Ú CategoricalÚCategoricalIndexÚ DataFrameÚIndexÚIntervalÚ IntervalIndexÚ
|
MultiIndexÚ
|
RangeIndexÚSeriesÚ TimestampÚcutÚ
|
date_rangec C s. ddgddgg} t ddgddggt | ¡d S )
|
NÚAÚ ÚBÚbr é é é ©Úcolumns)r r Ú from_tuples)Úlevels© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_reset_index.pyÚ multiindex_df s r c @ sj e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z dd Z dd Z ej dddgddgg¡dd Zej dddgdgg¡d d Zd d Zd d! Zd"d# Zej d$d%d&d'd(e d)¡ed*d+d,d-g¡d.d/ Zd0d1 Zd2d3 Zej d4d5d6g¡ej d7d5d6g¡d8d9 Zej d4d5d6g¡d:d; Zej d7d<g¡d=d> Zd?d@ Z dAdB Z dCdD Z dEdF Z ej dGej dfej d'fgej ej gdd'gdHdIgdJfej dfedKd'fgej edKgdd'gdHdIgdJfej dfe dLdM¡d'fgej e dLdM¡gdd'gdHdIgdJfg¡dNdO Z!ej dPddddgddddggdddQdgddddggf¡dRdS Z"d%S )TÚTestResetIndexc C sV t dgtjtdddddd}| |jdg¡}|jdgdd}tj||g dd d S )
|
NZbrandr r Úfoo©Úname)r ÚdtypeÚindexT©Údrop)Zcheck_index_type) r ÚnpÚint64r Ú set_indexr% Ú reset_indexÚtmÚassert_frame_equal)ÚselfÚdfÚdf2Úresultr r r Ú!test_reset_index_empty_rangeindex% s ÿz0TestResetIndex.test_reset_index_empty_rangeindexc C sd t dddgdd}tdddd gi|d
|
}| ¡ }|d jt d ¡ksHt| d¡}t |j |¡ d S ) Nl l l
|
r! r" r r r r ©r% Zuint64)
|
r r r+ r$ r( ÚAssertionErrorr* r, Úassert_index_equalr% ©r. Úidxr/ r1 r r r Útest_set_reset0 s
|
z TestResetIndex.test_set_resetc C sb t tdddddd}tddd d
|
gi|d }| ¡ }|d jd ksFt| d¡}t |j |¡ d S ) NÚ20130101r ú
|
US/Eastern©ÚperiodsÚtzr! r" r r r r r3 zdatetime64[ns, US/Eastern])
|
r r r r+ r$ r4 r* r, r5 r% r6 r r r Ú!test_set_index_reset_index_dt64tz; s
|
z0TestResetIndex.test_set_index_reset_index_dt64tzc s¾ | t ddd dd}ttdddddd gd
|
|d }ttd d d td d dtd d dtd d dtd d dgtdddddd gddddgd}|d fdd¡|d< t | ¡ |¡ d S )Nú1/1/2011é ÚDr7 ©r< Úfreqr= r# r r ÚCÚE©Úar r3 éÛ r r r é )r7 rG r rG r r c s t | dS ©N©r= ©r ©ÚdrK r r Ú<lambda>[ ó z4TestResetIndex.test_reset_index_tz.<locals>.<lambda>)r r Úranger Úapplyr, r- r+ )r. Ztz_aware_fixturer7 r/ Úexpectedr rK r Útest_reset_index_tzF s"
|
|
|
|
|
û ÷ ôz"TestResetIndex.test_reset_index_tzr= r: zdateutil/US/Easternc C sR t dd|d}ttj t|¡|}| ¡ d¡}|jj }|jj }||ksNt
|
d S )Nz
|
2012-06-02é
|
r; r% ) r r r( ÚrandomÚrandnÚlenr+ r* r% r= r4 )r. r= Zdrr/ Z roundtrippedÚxpÚrsr r r Ú$test_frame_reset_index_tzaware_index^ s z3TestResetIndex.test_frame_reset_index_tzaware_indexc C st t jt d¡dd}t|t d¡dddg }| d¡}tdt d¡i|d}t ||¡ | ¡ }t ||¡ d S )Né Úxr" rU )r] Úyr^ r3 ) r Z from_breaksr( Úaranger r* r, r- r+ )r. r7 Úoriginalr1 rS Zresult2r r r Ú test_reset_index_with_intervalsg s
|
z.TestResetIndex.test_reset_index_with_intervalsc C s¨ | ¡ d d d
}t||d}ddg}||j_| ¡ }tt|jj|jjD ]2\}\}}| |¡}|| } t
|
|t || ¡ qNd d g|j_| ¡ }
|
t
|
j |d |
|
d dd t
|
j |d |
|
d dd | ¡ } t|jjd d
|
} t
|
| d | ¡ | ¡ } d | d < | ¡ } t|jjdd
|
} t
|
| d | ¡ d |j_| ¡ }t
|
|d t|j¡ t
|
j |jt tt|d d d|j_| ¡ }|jjdkst| ¡ d ddg¡} | ddg¡}t
|
||¡ | d ddg¡}t
|
|| ¡ ¡ | d ddg¡}t
|
|| ¡ ¡ | d¡}| ¡ d dg¡}t
|
||¡ | ¡ } | ¡ }| jd d}|d ksLtt
|
| |¡ | ¡ d ddg¡} | jdd d}| ¡ }|d= |jdgd d}t
|
||¡ d S )Nr ©r! ÚbarÚfirstÚsecondÚlevel_0F©Z check_namesÚlevel_1r% r" r! T)Úexactr r r ©Zinplacer& ©Úappend)Ústackr r% Únamesr+ Ú enumerateÚzipr ÚcodesZtaker, r5 r Úassert_series_equalr ÚvaluesÚcopyr# rQ rX r r4 r* r- )r. Ú float_frameZstackedrn Ú deleveledÚiZlevZ level_codesrs r# Z
|
deleveled2ÚrdfÚexpr/ ÚresetrZ rY Ú return_valuer r r Útest_reset_indexr sv ÿ
|
ÿ ÿ
|
z TestResetIndex.test_reset_indexc C s t ddddgddddggd d
|
d d gttdd dd}| ¡ jjd ksJt|jddjjd ksbt|jdd}|d kszt|jjd kstd S )Nr r r rI r@ é é é r r rD rA r] r" ©r r% Tr& rj )r r rQ r+ r% r# r4 )r. r/ r{ r r r Útest_reset_index_name¾ s
|
ý z$TestResetIndex.test_reset_index_namer r r r r c C sN t ddddgddddggd d
|
d d gd }| d d
|
g¡j|d d}t || d
|
¡¡ | d d
|
g¡j|d d
d}t || d
|
¡¡ | d d
|
g¡j|d}t ||¡ | d d
|
g¡j|dd}t ||d d g ¡ | d ¡j|d d}t ||¡ | d ¡j|d d
d}t ||¡ | d g¡j|d dd}t ||d
|
d d g ¡ d S )Nr r r rI r@ r} r~ r r r rD rA r r ©ÚlevelT)r r' )r r* r+ r, r- )r. r r/ r1 r r r Útest_reset_index_levelÊ s ( z%TestResetIndex.test_reset_index_levelÚidx_levc C s t ddddgddddggd d
|
d d gd }tjtdd | |¡jd dgd W 5 Q R X tjtdd | |¡jdddgd W 5 Q R X d S )Nr r r rI r@ r} r~ r r r rD rA r z(L|l)evel \(?E\)?©ÚmatchrE r zToo many levelsr )r ÚpytestÚraisesÚKeyErrorr* r+ Ú
|
IndexError)r. r
r/ r r r Ú test_reset_index_level_missingå s
|
( z-TestResetIndex.test_reset_index_level_missingc C s~ t ddt d¡d ¡}td|d d t|dddd}t|}| ¡ }|d jt jks^t | ¡ }|d jt jkszt d S ) Nç rU r g
ëQ¸#@Útimer" Úspeed)r% r# )
|
r( r_ Úsqrtr r r r+ r$ Úfloat64r4 )r. r Ús1r/ rz r r r Ú test_reset_index_right_dtypeï s
|
ÿz+TestResetIndex.test_reset_index_right_dtypec C s t j dd¡ t¡}dddg}t dd |D |f¡}t|t|ddd d d
|
gd d d ggd }| ¡ }t|dd d d
|
gdd d d ggd }t
|
||¡ |jd d}t|dd d d
|
gdd d d ggd }t
|
||¡ |jddd}t|dd d d
|
gdd d d ggd }t
|
||¡ t|t j dddgdddggddgdd d d
|
gd d d ggd }| d¡}t|tdddgdddd d d
|
gdd d d ggd }t
|
||¡ |jdd d}t|tt ddddd d d
|
gdd d d ggd }t
|
||¡ |jdddd}t|tt ddddd d d
|
gdd d d ggd }t
|
||¡ d S )Nr r] r^ Úzc S s g | ]
|
}|gqS r r )Ú.0r] r r r Ú
|
<listcomp>ÿ s zBTestResetIndex.test_reset_index_multiindex_col.<locals>.<listcomp>rG r" r ÚcZmeanZmedianr r ©Úcol_fillr Zblah©Ú col_levelr r r rN ©rn )r r )r( rV rW ÚastypeÚobjectZhstackr r r+ r, r- r Ú from_arraysrQ )r. Úvalsr7 Úfullr/ rZ rY r r r Ú test_reset_index_multiindex_colü sh
|
|
ý ÿ ÿ ÿ ý
|
ý ý ýz.TestResetIndex.test_reset_index_multiindex_colc C s t dddgddtjgtj d¡d}| dd g¡ ¡ }t ||¡ t tjddgddd
|
gtj d¡d}| dd g¡ ¡ }t ||¡ t dddgddd
|
gtjd d gd}| dd g¡ ¡ }t ||¡ t dddgtjtjtjgtj d¡d}| dd g¡ ¡ }t ||¡ d S ) NrG r r r r r )r r rD r r r çñ?g@) r r( ÚnanrV Zrandr* r+ r, r- )r. r/ rZ r r r Ú test_reset_index_multiindex_nan4 s* ÿ ÿ $
|
ýÿz.TestResetIndex.test_reset_index_multiindex_nanr# Nr! r g @r} z
|
2012-12-30ÚUTCrK ú
|
2012-12-31c C sº t ddgddggtddddgd }||j_| ¡ }|d k r@|nd
|
}t|td ddtd ddg}t|tr|d kr| d ¡}n|j
|
t kst t dddgdddgg|d}t ||¡ d S )Nr r r rI z1/1/2013z1/2/2013r r r r% éÝ r§ údatetime64[ns]r )r r r% r# r+ r r Ú
|
isinstanceÚstrr r$ r r4 r, r- )r. r# r/ r1 Úitemr rS r r r Ú(test_reset_index_with_datetimeindex_colsQ s" ý þz7TestResetIndex.test_reset_index_with_datetimeindex_colsc C sv t ddgddggddgtddd}| ¡ }t|jts<tt dddgdddggdddgtddd}t ||¡ d S ) Nr r r r r )Ústopr r% )r r r+ rª r% r4 r, r- )r. r/ r1 rS r r r Útest_reset_index_rangeu s $ýz%TestResetIndex.test_reset_index_rangec C sÀ |dg d¡ ¡ }t ||¡ d}tjt|d | d¡ ¡ W 5 Q R X | dg¡ ¡ }t ||¡ tdgdggt
|
dg¡d }t j ||d
|
dg gdd }|jd
|
gd d ¡ }t ||¡ d}tjt|d | dg¡ ¡ W 5 Q R X dddgdddgg}tddgddggt
|
|¡d }tdgdggt
|
dg¡d }t j ||gdd }| dg¡jdd}t ||¡ tjtdd | dg¡jd d W 5 Q R X | dg¡jdd d }t ||¡ d S ) Nr r ú)cannot insert \('A', ''\), already existsr )r r r r )rf r r )r r )ZaxisTrk z0Item must have length equal to number of levels.)rD r rw rG r r rw r r )rD r Úii)rD r r± r zFcol_fill=None is incompatible with incomplete column name \('C', 'c'\))r r± rD r ) Ú rename_axisr+ r, r- r r Ú
|
ValueErrorr* r r r ÚpdÚconcat)r. r r1 ÚmsgZidx_colrS r r0 r r r Ú#test_reset_index_multiindex_columns sB
|
|
ÿ
|
|
ÿ ý z2TestResetIndex.test_reset_index_multiindex_columnsÚflagFTÚallow_duplicatesc C s¬ | d¡}|j|d}|rp|rp|j|d}ddgddgddgg}tdddgd d d
|
ggt |¡d }t ||¡ n8|s~|r~d }nd }tj t
|
|d |j|d W 5 Q R X d S )Nr ©Zallows_duplicate_labels©r¹ r r r r r r r r zYCannot specify 'allow_duplicates=True' when 'self.flags.allows_duplicate_labels' is Falser° r ) r² Ú set_flagsr+ r r r r, r- r r r³ ) r. r r¸ r¹ r/ r1 r rS r¶ r r r Ú(test_reset_index_duplicate_columns_allow³ s
|
ÿÿz7TestResetIndex.test_reset_index_duplicate_columns_allowc C s@ | d¡}|j|d}d}tjt|d | ¡ W 5 Q R X d S )Nr rº r° r )r² r¼ r r r³ r+ )r. r r¸ r/ r¶ r r r Ú*test_reset_index_duplicate_columns_defaultÎ s
|
|
z9TestResetIndex.test_reset_index_duplicate_columns_defaultz bad valuec C s* t jtdd |j|d W 5 Q R X d S )Nzexpected type boolr r» )r r r³ r+ )r. r r¹ r r r Ú'test_reset_index_allow_duplicates_check× s z6TestResetIndex.test_reset_index_allow_duplicates_checkc sÚ | t ddd dd}ttdddd}t ||g¡}ttjddd d
|
d d dd gd|d}ttdddtdddtdddtdddtdddgtjddd tjddd d
|
d d dd gdddddgd}|d fdd¡|d< t
|
| ¡ |¡ t ddd d d d}t |||g¡}ttjddd d
|
d d dd gd|d}ttdddtdddtdddtdddtdddgtjddd td ddtd ddtd ddtd ddtd ddgtjddd d
|
d d dd gd ddd ddgd}|d fd!d¡|d< |d d"d ¡|d < t
|
| ¡ |¡ t ddgt d#d d$g¡}ttjd%dd d%d¡dg|d&}td' ¡ td(ddtd(ddtd(ddgd tjd%dd d)d*d+dgd}|d+ fd,d¡|d+< | ¡ }t
|
||¡ d S )-Nr? r@ rA Úidx1rB Úidx2r) )r# r$ ©r$ r r rD rE rF r3 rH r r r rI )rÀ rÁ rG r rG r r c s t | dS rJ rL rM rK r r rO ö rP z:TestResetIndex.test_reset_index_datetime.<locals>.<lambda>z1/1/2012ZMSú Europe/ParisÚidx3iÜ )rÀ rÁ rÄ rG r c s t | dS rJ rL rM rK r r rO rP c S s t | ddS )Nrà rK rL rM r r r rO rP r9 r; r} r z a a a b b br¨ )rf rh rG rf rh c s t | dS rJ rL rM rK r r rO 4 rP )r r rQ r r r r( r_ r rR r, r- r+ Ú from_productÚreshapeÚsplit) r. Ztz_naive_fixturerÀ rÁ r7 r/ rS rÄ r1 r rK r Útest_reset_index_datetimeÜ s¤ þ
|
|
|
|
|
û ö
|
ó ÿ þ
|
|
|
|
|
û
|
|
|
|
|
û ï ìÿÿ ÿ
|
|
|
ýû ø
|
õ z(TestResetIndex.test_reset_index_datetimec C sÆ t jtjddddtdgddgd}ttjd d
|
d d d ¡|dgd}ttj dddgd tj dddgd tj dddgd dddgd tjd d
|
d ddddgd}|
|
¡ }t ||¡ d S )Nr9 r ÚM©r< rC ÚabcÚmonthÚfeaturer é r) r éÿÿÿÿr rG ©r% r z2013-01©rC z2013-02z2013-03r r )rÌ rÍ rG r ) r rÅ r´ Ú period_rangeÚlistr r( r_ rÆ ZPeriodr+ r, r- )r. r7 r/ rS r1 r r r Útest_reset_index_period8 s, þ ÿÿþ ù ö z&TestResetIndex.test_reset_index_periodc C sz t tddgddgddg}tj|ddd gd
|
}ttj d d ¡d ddg|d}| ¡ }t |d sft
|
t |d svt
|
d S )Nr! rc rU é ç ð?r£ Zprm0Zprm1Zprm2r r r r r rD r ) rÓ r r r r r( rV rW r+ r r4 r )r. Ztuplesr% r/ rv r r r Ú$test_reset_index_delevel_infer_dtypeQ s z3TestResetIndex.test_reset_index_delevel_infer_dtypec C s@ |}|j dd}t|jt|jks(t|jj|jjks<td S )NTr& )r+ rX r r4 r% r# )r. Z/multiindex_year_month_day_dataframe_random_dataZymdrv r r r Útest_reset_index_with_dropY s z)TestResetIndex.test_reset_index_with_dropzix_data, exp_datar\ é )rG r r] ú
|
2020-01-01é{ rN c C sF t j|ddgd}tdddgi|d}| ¡ }t|}t ||¡ d S )NrG r r r] r\ rÙ r3 )r r r r+ r, r- )r. Zix_dataZexp_dataÚixr1 rS r r r Ú test_reset_index_nat_multiindexb s
|
z.TestResetIndex.test_reset_index_nat_multiindexrq rÏ c C s¨ t tddgtddgg|}dtt|i}t||d}ttj|d ddgdtj|d ddgdtd
|
d }| ¡ }t ||¡ |
|
d d g¡ ¡ }t ||¡ d S )Nr r rG r Úcol)Údatar% r )Ú
|
categoriesr rI )rf rh rÞ rf rh ) r r rQ rX r r Z
|
from_codesr+ r, r- r* )r. rq r% rß r/ rS Úresr r r Ú:test_rest_index_multiindex_categorical_with_missing_values| s ÿ ýÿ zITestResetIndex.test_rest_index_multiindex_categorical_with_missing_values)#Ú__name__Ú
|
__module__Ú __qualname__r2 r8 r> rT r ÚmarkÚ parametrizer[ ra r| r r r r r¢ r¥ r´ Z Timedeltar r r¯ r· r½ r¾ r¿ rÈ rÔ r× rØ ÚNaTrÝ râ r r r r r $ sx
|
L
|
|
8
|
ùþ
|
2
|
|
\ þ þ þ÷þ
|
.ÿr z array, dtyperG r z 12-1-2000r zQ-DECrÊ rÑ c C sT t ddgddg| g¡}t|dd d
¡ j}ttjtj|d}t
|
||¡ d S )Nr r g à?rÖ r3 )rf rh Zlevel_2) r rÅ r r+ Zdtypesr r( r) r r, rr )Úarrayr$ r7 r1 rS r r r Ú6test_reset_index_dtypes_on_empty_frame_with_multiindex s rê c C s t tdgddggg g gddgd} t| ddgd }| ¡ }ttd
|
td d d d d}|d d¡|d< |d d¡|d< t ||¡ d S )Nz2020-07-20 00:00:00r rI rG r )r rq rn r rN rÐ Zabcdr r ©Ústartr® Ústepr r© r) ) r r r r+ rÓ r r r, r- )r7 r/ r1 rS r r r Ú7test_reset_index_empty_frame_with_datetime64_multiindexª s ý ÿrî c C s t dgdgt d¡d} | d¡ ddg¡dg ¡ } | ¡ }t dddgtddd d
|
d }|d d ¡|d< |d d ¡|d< t
|
||¡ d S )Ng $@rG rÚ )Úc1Úc2Úc3r rð rñ rï r rë r r© r ) r r´ Ú to_datetimeÚheadÚgroupbyÚsumr+ r r r, r- ©r/ r1 rS r r r ÚDtest_reset_index_empty_frame_with_datetime64_multiindex_from_groupby» s ÿr÷ c C s t d} tdddd}t| |tdd}tj|jd< | dd g¡ d¡}tt dtdd
|
tj dd d gd d d}t
|
||¡ d S )Nr z
|
2015-07-01rA )rC r< rË )ÚidÚtstamprG )r rù rø rù )rø rG z
|
2015-07-02rè r" r3 ) rQ r r rÓ r´ rè Úlocr* r+ Z DatetimeIndexr, r- )r7 rù r/ r1 rS r r r Ú test_reset_index_multiindex_natÈ s þrû c C sz t t d¡tddgddtddgdddgd} | ¡ }t dddgdddggtdtddtddgd }t ||¡ d S )
|
Nr r ZYearr" r rÐ rÖ r r ) r r( Zeyer r r+ r
|
r, r- rö r r r Ú-test_reset_index_interval_columns_object_castÖ s ÿþrü c C s\ | j dd}t| jjdd}t |d |¡ | j dd}t| jjdd}t |d |¡ d S )NÚnew_namer r" rÛ )r+ r r% rs r, rr )ru r1 rS r r r Útest_reset_index_renameã s rþ c C s| | ¡ d d d
}t||d}ddg}||j_| ¡ }|jddgd}tj|d |d dd tj|d |d dd d S )
|
Nr rb rd re Ú new_firstÚ
|
new_secondr Frg )rm r r% rn r+ r, rr )ru Ú
|
stacked_dfrn r1 rS r r r Ú"test_reset_index_rename_multiindexî s r c C sx | ¡ d d d
}t||d}tjtdd |jdddd W 5 Q R X tjtdd |jdgd W 5 Q R X d S ) Nr )rd re z-Index names must be str or 1-dimensional listr rÿ r r zlist index out of range)rm r r r r³ r+ r )ru r r r r Ú test_errorreset_index_renameü s ÿ r )*r Ú itertoolsr Únumpyr( r Zpandas.core.dtypes.commonr r Zpandasr´ r r r r r
|
r r r r r r r Zpandas._testingZ_testingr, Zfixturer r ræ rç r rÒ Z PeriodDtyperê rî r÷ rû rü rþ r r r r r r Ú<module> s< 8
|
x
|
|
þþþ
|
|
|
