zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
U
®ý°dWã    @sàddlmZddlZddlZddlZddlmZmZmZm    Z    m
Z
m Z ddl m ZGdd„dƒZej dddgefejd    d
d d ejd d fg¡dd„ƒZej dddgddgfddgddgfg¡ej dddg¡dd„ƒƒZdS)é)ÚdatetimeN)Ú    DataFrameÚIndexÚ
MultiIndexÚ
RangeIndexÚSeriesÚ
date_rangec@sLeZdZdd„Zdd„Zdd„Zdd„Zd    d
„Zd d „Zd d„Z    dd„Z
dS)ÚTestResetIndexcCsÄtdddd d¡}tdtj t|ƒ¡i|d}| ¡}|jdt     d¡ksPt
‚|  d    ¡}t j ||d
d td d dƒ}t|dggddgd}|  d¡}|jd|ks¨t
‚| ¡dd|ksÀt
‚dS)Nz1/1/2001z6/1/2001ÚD)ÚstartÚendÚfreqÚv©ÚindexrzM8[ns]rF)Z check_namesiÜé ég333333(@ÚDateÚValue©Úcolumns)rZ
_with_freqrÚnpÚrandomZrandÚlenÚ reset_indexÚdtypesÚdtypeÚAssertionErrorÚ    set_indexÚtmÚassert_frame_equalrr)ÚselfZdtiÚd1Zd2Zd3ZstampÚdf©r$úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_reset_index.pyÚtest_reset_index_dti_round_trips
 
z.TestResetIndex.test_reset_index_dti_round_tripc
CsFt ¡dd…}| ¡}ddg|j_d|_| ¡}d|ks>t‚|jdd}d|ksVt‚|jdd}|}|jddd    }|dks€t‚t ||¡t    d
gd d d gddggddddddgddddddgddddddggd}t
t j   d¡|d}|jdd}t|jƒdkst‚|jddgdd}t |jt| d¡ƒ¡t|t
ƒsBt‚dS)NéÚhashÚcategoryÚvalueZvalue2©ÚnameT©Údrop)r.ÚinplaceÚbarZoneZtwoZthreeréé)ÚlevelsÚcodesér©Úlevel©r7r.)rZ makeDataFrameÚstackrÚnamesr,rrÚassert_series_equalrrrrZrandnrrZassert_index_equalrZget_level_valuesÚ
isinstance)r!r#ÚserÚsÚs2Z return_valuerÚrsr$r$r%Útest_reset_index#s.       ,þ zTestResetIndex.test_reset_indexcCsNtdddgttdƒddd}| ¡jjdks2t‚|jddjjdksJt‚dS)    Nr1r2éÚxr+rTr-)rrÚrangerrr,r©r!r>r$r$r%Útest_reset_index_nameCsz$TestResetIndex.test_reset_index_namec
CsÚtdddgdddggddd    gd
}ddgd dgfD]r}| ddg¡d    }|j|d d }t || d¡¡|j|dd…d }t || d¡¡|j|d }t ||¡| ddg¡j|d d}t ||d    g¡tjtdd|jddgd W5QRX| d¡d}|j|d d }t ||ddg¡|j|dd…d }t ||ddg¡|j|d d d}t ||d¡tjt    dd|jd ddgd W5QRXq2t
t dƒƒjgd d}t
t dƒƒ}t ||¡dS)Nr1r2rBér'r5ÚAÚBÚCrrr6Tr8zLevel E ©ÚmatchÚEzToo many levelsr-) rrrrr ÚpytestÚraisesÚKeyErrorr;Ú
IndexErrorrrD)r!r#r3r>ÚresultÚexpectedr$r$r%Útest_reset_index_levelHs2"   z%TestResetIndex.test_reset_index_levelcCs^ttdƒddd}| ¡}t|jtƒs*t‚tddgddggddgtddd    }t     ||¡dS)
Nr2rHÚint64)r,rrr1r)Ústop)rr)
rrDrr<rrrrrr )r!r>Z series_resultZseries_expectedr$r$r%Útest_reset_index_rangersÿz%TestResetIndex.test_reset_index_rangec    Cs¤ttdƒƒ}tjtdd|jdddW5QRXtjtdd| d¡W5QRXttdƒt ddggd¡d    }tjtd
d|jdddW5QRXdS) NrGzdoes not match index namerKZwrongTr-r1r2rz    not found)rrDrNrOrPrrÚ from_productrEr$r$r%Útest_reset_index_drop_errors|s  z+TestResetIndex.test_reset_index_drop_errorscCs~|}| ¡}t|tƒst‚t|jƒt|jjƒdks8t‚|jj|jjksLt‚|jdd}t|t    ƒsft‚|jj|jjkszt‚dS)Nr1Tr-)
rr<rrrrrr3r,r)r!Zseries_with_multilevel_indexr=Z    deleveledr$r$r%Útest_reset_index_with_drop‹s z)TestResetIndex.test_reset_index_with_dropcCs@ttdƒdd}|jddddttdƒdd}t ||¡dS)Nr2Úoldr+ÚnewT)r,r.r/)rrDrrr;)r!r=rSr$r$r%Ú-test_reset_index_inplace_and_drop_ignore_name—sz<TestResetIndex.test_reset_index_inplace_and_drop_ignore_nameN) Ú__name__Ú
__module__Ú __qualname__r&rArFrTrWrYrZr]r$r$r$r%r    s *
 r    z array, dtypeÚaÚbz    12-1-2000r2zQ-DEC)Zperiodsr )r cCsXt ddgddg|g¡}tt|ddd… ¡j}ttjtj|tdœƒ}t     
||¡dS)Nrr1gà?gð?)rr)Zlevel_0Úlevel_1Zlevel_2r) rrXrÚobjectrrrrUÚfloat64rr;)ÚarrayrÚidxrRrSr$r$r%Ú7test_reset_index_dtypes_on_empty_series_with_multiindexŸs ÿrhznames, expected_namesrHrcÚallow_duplicatesFTc    Cs~tdgtjdgdgg|dd}|rX|jdd}tdddgg|dgd}t ||¡n"tjt    dd    | ¡W5QRXdS)
Nr1)r:rT)rirrz cannot insertrK)
rrZ from_arraysrrrr rNrOÚ
ValueError)r:Zexpected_namesrir>rRrSr$r$r%Útest_column_name_duplicates³s
  rk)rÚnumpyrrNZpandasÚpdrrrrrrZpandas._testingZ_testingrr    ÚmarkZ parametrizerdZ period_rangeZ PeriodDtyperhrkr$r$r$r%Ú<module>s0   
 
þþþ
 
 
þþ