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
U
®ý°dã@sVddlmZddlZddlZddlZddlmZmZmZddl    m
Z Gdd„dƒZ dS)é)ÚdatetimeN)ÚIndexÚ
MultiIndexÚSeriesc@sŒeZdZdd„Zdd„Zdd„Zdd„Zd    d
„Zd d „Zd d„Z    dd„Z
dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd „Zd!S)"Ú
TestRenamecCs^|}dd„}| |¡}|jd||jdƒks2t‚tt|j|jƒƒ}| |¡}t ||¡dS)NcSs
| d¡S©Nz%Y%m%d©Ústrftime©Úx©r ú^d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_rename.pyÚ<lambda>óz(TestRename.test_rename.<locals>.<lambda>r)ÚrenameÚindexÚAssertionErrorÚdictÚzipÚtmÚassert_series_equal)ÚselfÚdatetime_seriesÚtsÚrenamerÚrenamedZ rename_dictZrenamed2r r r Ú test_renames
 
zTestRename.test_renamecCsJtt d¡ddddgdd}| dd    d
œ¡}t |jtdddd    gƒ¡dS) NéÚaÚbÚcÚdÚint64©rÚdtypeÚfooÚbar)rr!)rÚnpÚarangerrZassert_index_equalrr)rÚserrr r r Útest_rename_partial_dictsz#TestRename.test_rename_partial_dictcCsFtt d¡tddddgdddd    }| i¡}|jj|jjksBt‚dS)
Nrrrr r!Úname©r+r"r#)rr'r(rrrr+r)rrrr r r Útest_rename_retain_index_name!sÿ
z(TestRename.test_rename_retain_index_namecCsTttdƒdd}tdddœƒ}| |¡}ttdƒddddd    gdd
}t ||¡dS) Nér%r,é
é)éérér)rr+)rÚrangerrr)rr)rÚresultÚexpectedr r r Útest_rename_by_series)s
 
z TestRename.test_rename_by_seriescCsnttdƒtdƒd}dddtdddƒd    fD]>}| |¡}|j|ksFt‚t |j    j
|j    j
¡|jdks*t‚q*dS)
NrZabcd©rr%é{çÀ^@éÑé ©r%) rr4Úlistrrr+rrÚassert_numpy_array_equalrÚvalues)rr)r+r5r r r Útest_rename_set_name0s 
zTestRename.test_rename_set_namecCsvttdƒtdƒd}dddtdddƒd    fD]F}|j|d
d |j|ksJt‚tjd d dgtj    d}t
  |j j |¡q*dS)Nr3Úabcr8r%r9r:r;r<r=T©Zinplacerrr ©r$)rr4r>rrr+rr'ÚarrayZobject_rr?rr@)rr)r+Úexpr r r Útest_rename_set_name_inplace8s z'TestRename.test_rename_set_name_inplacec    CsTttdƒƒ}|jidd|jiddtjtdd|jiddW5QRXdS)Nr.r)ZaxisrzNo axis named 5©Úmatch)rr4rÚpytestÚraisesÚ
ValueError)rr)r r r Útest_rename_axis_supportedAs
 z%TestRename.test_rename_axis_supportedcCs:dd„}||jdƒ}|j|dd|jd|ks6t‚dS)NcSs
| d¡Srrr
r r r rKrz0TestRename.test_rename_inplace.<locals>.<lambda>rTrC)rrr)rrrr6r r r Útest_rename_inplaceJszTestRename.test_rename_inplacecCs:Gdd„dƒ}|ƒ}tdddgƒ |¡}|j|ks6t‚dS)Nc@s eZdZdS)z=TestRename.test_rename_with_custom_indexer.<locals>.MyIndexerN©Ú__name__Ú
__module__Ú __qualname__r r r r Ú    MyIndexerSsrSr1r2r3©rrr+r©rrSÚixr)r r r Útest_rename_with_custom_indexerQsz*TestRename.test_rename_with_custom_indexercCsBGdd„dƒ}|ƒ}tdddgƒ}|j|dd|j|ks>t‚dS)Nc@s eZdZdS)zETestRename.test_rename_with_custom_indexer_inplace.<locals>.MyIndexerNrOr r r r rS\srSr1r2r3TrCrTrUr r r Ú'test_rename_with_custom_indexer_inplaceZs
z2TestRename.test_rename_with_custom_indexer_inplacecCsXttddƒttddƒddd}| t¡}| dd    „¡}t ||¡|j|jksTt‚dS)
Nr1ér2éZIntIndexr,r8cSst|ƒS)N)Ústr)Úir r r rhrz1TestRename.test_rename_callable.<locals>.<lambda>)    rr4rrr[rrr+r©rr)r5r6r r r Útest_rename_callableds
 
 zTestRename.test_rename_callablecCs6tddgdd}| d¡}tddgƒ}t ||¡dS)Nr1r2r%r,)rrrrr]r r r Útest_rename_nonems
 zTestRename.test_rename_nonecCs¤dddddgdddddgg}tj|ddgd    }tt d
¡|d }|jdd idd d}dddddgd d ddd gg}tj|ddgd    }tt d
¡|d }t ||¡dS)Nr&Zbazr%ZquxZoneZtwoÚfirstÚsecond©Únamesr.r8ÚyesÚraise)rÚlevelÚerrors)rÚ from_arraysrr'Zonesrrr)rÚarraysrr)r5Úarrays_expectedZindex_expectedZseries_expectedr r r Ú"test_rename_series_with_multiindexts  þ  þÿz-TestRename.test_rename_series_with_multiindexcCs²tdddgdd d¡tdddgddg}tj|ddgd    }td|d
}|jdd idd }tdddgdd d¡td ddgddg}tj|ddgd    }td|d
}t ||¡dS) Nr1r2r3ZInt64rDÚcategoryÚAÚBrbr8r)rf)rZastyperrhrrrr)rriÚmir)r5rjZ mi_expectedr6r r r Ú2test_rename_series_with_multiindex_keeps_ea_dtypes‹sþ þ z=TestRename.test_rename_series_with_multiindex_keeps_ea_dtypesc    CsFtddgƒ}t d¡}tjt|d|jddiddW5QRXdS)    Nr%r&z[2] not found in axisrHr2é    re)rg)rÚreÚescaperJrKÚKeyErrorr)rr)rIr r r Útest_rename_error_argžs 
z TestRename.test_rename_error_argcCsˆtddgƒ}| ¡}|jddidd}d|d<|r\|d|dksFt‚|d|dks„t‚n(|d|dkspt‚|d|dks„t‚dS)    Nr%r&r1rqF)ÚcopyZfoobarr)rrvrr)rZusing_copy_on_writer)Zser_origZ shallow_copyr r r Útest_rename_copy_false¥s z!TestRename.test_rename_copy_falseN)rPrQrRrr*r-r7rArGrMrNrWrXr^r_rkrprurwr r r r rs              
    r) rrrÚnumpyr'rJZpandasrrrZpandas._testingZ_testingrrr r r r Ú<module>s