U
|
ý°d= ã @ sZ d dl mZ d dlZd dlZd dlZd dlmZmZm Z m
|
Z
|
d dl m Z G dd dZdS )é )ÚChainMapN)Ú DataFrameÚIndexÚ
|
MultiIndexÚmergec @ s> e Zd Zdd Zdd Zdd Zej de dd id
|
d ifd d ifdd e dd id
|
d iifg¡dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zej d d d d d d d
|
d!d"gfdd id#d d
|
d!d"gfejd#d d d$d%gfg¡d&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Zd6d7 Zd8d9 Z d:d; Z d<S )=Ú
|
TestRenamec
|
C s8 t tj¡}t|j}|ddddddddd h ks4td S )
|
NÚselfÚmapperÚindexÚcolumnsÚaxisÚinplaceÚcopyÚlevelÚerrors)ÚinspectÚ signaturer ÚrenameÚsetÚ
|
parametersÚAssertionError)r Úsigr © r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_rename.pyÚtest_rename_signature s
|
÷z TestRename.test_rename_signaturec C s2 |dddgt dd dD ¡d}| tj¡ d S )Né é é c S s g | ] }d |fqS )ÚAr )Ú.0Úxr r r Ú
|
<listcomp># s z-TestRename.test_rename_mi.<locals>.<listcomp>)ÚaÚBÚc©r
|
)r Ú from_tuplesr ÚstrÚlower)r Zframe_or_seriesÚobjr r r Útest_rename_mi s
|
þzTestRename.test_rename_mic C s ddddd}|j |d}|j tjd}t ||¡ tj|j tjd|dd d d
|
d d i}t|}|j d dd d}t |jt dd g¡ |j tjd}t |jt ddg¡ t
|
j t dd | ¡ W 5 Q R X |j dd dd}t |j t d ddd g¡ |jj dd dd}t |jt d ddd g¡ t dd gdd}t||d}|j d dd d}t |jt d dgdd¡ |jj|jjkstd S )Nr" Úbr$ Úd)r r# ÚCÚD©r F)Z check_namesr r é )ÚfooÚbarr2 r1 r% ZBARÚFOOú must pass an index to rename©Úmatch)r- r. r# Úname)r7 )r r' r( ÚtmÚassert_frame_equalÚupperr Úassert_index_equalr
|
r ÚpytestÚraisesÚ TypeErrorr ÚTr7 r ) r Ú float_frameÚmappingÚrenamedZrenamed2ÚdataÚdfr
|
Zrenamerr r r Ú test_rename' s4 ÿ zTestRename.test_renamez args,kwargsr r" r# r+ r r r c C sN t dd}tj d¡}t||d}|j||}t||d}t ||¡ d S )Nr0 r é
|
©r r# ©r" r+ )ÚrangeÚnpÚrandomZrandnr r r8 r9 )r ÚargsÚkwargsZcolADataZcolBdatarD ÚresultÚexpectedr r r Útest_rename_chainmapP s
|
z TestRename.test_rename_chainmapc
|
C sb ddg}ddg}t j|ddgd}t j|dd gd}td
|
d g||d }|jd dddddd }t jddgddgd}t jddgdd gd}t |j|¡ t |j|¡ |jj|jjks¼t |jj|jjksÐt t jddgdd gd}|jddddd}t |j|¡ |jddddd}t |j|¡ t jddgdd gd}|jddddd}t |j|¡ |jdddd d}t |j|¡ t
|
j } t jddgdd gd}|j| dd}t |j|¡ |j| dd}t |j|¡ t jd d gdd gd}|j| dd}t |j|¡ |j| d d}t |j|¡ t jddgddgd}|jd dddd }t |j|¡ d S ) N)Úfoo1Úbar1)Úfoo2Úbar2)Úfizz1Úbuzz1)Úfizz2Úbuzz2r1 r2 ©ÚnamesZfizzZbuzz©r r ©r0 r0 ©r
|
r Úfoo3Úbar3)rQ rT Úfizz3Úbuzz3)rU rX )r^ rR )rS r_ )r` rV )rW ra r )r r r0 )ZFIZZ1rV )ZFIZZ2rX )rU ZBUZZ1)rW ZBUZZ2)r
|
r ) r r& r r r8 r; r
|
r rZ r r' r: )
|
r Z tuples_indexZtuples_columnsr
|
r rD rB Z new_indexZ new_columnsÚfuncr r r Útest_rename_multiindexb sx þ ÿ ÿ ÿ ÿ ÿ ÿ ÿz!TestRename.test_rename_multiindexc C sr |j ddidd}t |d j|d j¡s.td|jd d
df< |rZ|d dk ¡ rntn|d dk ¡ sntd S )Nr- r1 F)r r g ð?)r rJ Z shares_memoryZ_valuesr ÚlocÚall)r r@ Zusing_copy_on_writerB r r r Útest_rename_nocopy¨ s z TestRename.test_rename_nocopyc C s |j ddid d|ks td|ks(t|d }| ¡ }|j ddidd}|d ksVtd|ksbtd|ksnt|d |k s~td S )Nr- r1 r/ T)r r )r r r )r r@ Zc_valuesZ return_valuer r r Útest_rename_inplace³ s z TestRename.test_rename_inplacec C s t ddgddgddgd}|jdd id
|
}|jdd id
|
}| d d g¡}d g|_t dgdggtjd dgd d gdd gd}t ||¡ d S )Nr1 r2 ÚbahÚbasr0 é ©r r0 rj r r" r/ r+ z
|
2001-01-01)r1 rh )r2 ri rY r] )r r Ú set_indexr r r& r8 r9 ©r rD rO r r r Útest_rename_bugà s
|
ÿûzTestRename.test_rename_bugc C sZ t t d¡dddgdgd}|jddidd }t t d¡dddgdgd}t ||¡ d S )
|
Né r[ r\ )rj rj r" )rC r
|
r )é é r
|
©r )r rJ Úaranger r8 r9 rm r r r Útest_rename_bug2Ô s ÿzTestRename.test_rename_bug2c C sD t ddddgd}tjtdd |jdd d
|
d d W 5 Q R X d S ) Nr r# r- r. r/ z'E'] not found in axisr5 r" Úe©r ÚEÚraise©r r )r r< r= ÚKeyErrorr )r rD r r r Útest_rename_errors_raisesß s z$TestRename.test_rename_errors_raisesz mapper, errors, expected_columnsru rv Úignorer- r. rx r$ r, c C s: t ddddgd}|j||d}t |d}t ||¡ d S )Nr r# r- r. r/ ry )r r r8 r9 )r r r Zexpected_columnsrD rN rO r r r Útest_rename_errorsä s
|
z TestRename.test_rename_errorsc C s* |j tjd}d|kstd|ks&td S )Nr/ r3 r1 )r r' r: r )r Zfloat_string_framerB r r r Útest_rename_objectsô s z TestRename.test_rename_objectsc C sj t ddgddgdddgd}t ddgddgdddgd}|jtjdd}t ||¡ |jtjd d}t ||¡ |jd
|
d ddd}t ||¡ |jd
|
d dd d}t ||¡ t ddgddgdd d gd}|jtjdd}t ||¡ |jtjdd}t ||¡ |jd d ddd}t ||¡ |jd d ddd}t ||¡ |jtjdd}t ||¡ d S )Nr0 rj rG ÚXÚYr% rH rr r r" r+ r Úyr r
|
)r r )r r )r r r' r( r8 r9 )r rD rO rN r r r Útest_rename_axis_styleú s* z!TestRename.test_rename_axis_stylec C sP t ddgddgddgd dd g¡}| tj¡}|jtjd
|
}t ||¡ d S ) Nr" r+ r$ r, r0 rj )r r# r- r r# r% )r rl r r' r: r8 r9 ©r rD rN rO r r r Útest_rename_mapper_multi s ÿ z#TestRename.test_rename_mapper_multic C s^ t ddgddgdddgd}|jtjtjd}t ddgddgdd d
|
gd}t ||¡ d S ) Nr0 rj rH r r r% r] rG r r )r r r' r( r: r8 r9 r r r r Ú test_rename_positional_named$ s z'TestRename.test_rename_positional_namedc C sX t ddgddgdddgd}d}tjt|d |jtjdd W 5 Q R X tjt|d |jtjd
|
d W 5 Q R X tjt|d |jtjd
|
d W 5 Q R X tjt|d |jtjd d W 5 Q R X tjt|d |jtjtjd
|
d W 5 Q R X d }tjt|d |jtjtjtjd W 5 Q R X tjtdd |jttd W 5 Q R X d S )Nr0 rj rG Ú0Ú1r% z:Cannot specify both 'axis' and any of 'index' or 'columns'r5 )r
|
r r )r r r ú<Cannot specify both 'mapper' and any of 'index' or 'columns'r] zmultiple values)r )r r< r= r> r r' r( Úid)r rD Z over_spec_msgr r r Ú test_rename_axis_style_raises+ s" z(TestRename.test_rename_axis_style_raisesc C s> t ddgd}d}tjt|d | d tj¡ W 5 Q R X d S )Nr r# r/ z1rename\(\) takes from 1 to 2 positional argumentsr5 )r r< r= r> r r' r( ©r rD Úmsgr r r Ú test_rename_positional_raisesJ s z(TestRename.test_rename_positional_raisesc C s° t dgg}d}tjt|d | ¡ W 5 Q R X tjt|d |jd d d W 5 Q R X tjt|d |jd d d W 5 Q R X tjt|d |jd d d d W 5 Q R X d S )Nr0 r4 r5 r% r/ ©r r
|
©r r< r= r> r r r r r Ú test_rename_no_mappings_raisesR s z)TestRename.test_rename_no_mappings_raisesc C s t dgg}d}tjt|d |ji i d W 5 Q R X tjt|d |ji i d W 5 Q R X tjt|d |ji i i d W 5 Q R X d S )Nr0 r r5 r% r/ r r r r r r Ú2test_rename_mapper_and_positional_arguments_raisesb s z=TestRename.test_rename_mapper_and_positional_arguments_raisesc C sæ t dgdgdgdtjdgddgdd }t d
|
d d gd gd dddgdddgdtjdddgddgdd }t||dddd}|jdddd }t| |j t dddd d ddggd dd ddd dgd jddgd d!}t ||¡ d S )"Ngz6«>§?g
ëQ¸ 6@g^KÈ=¥?)ÚRTÚTCloseÚTExg)éé* é*3ÚSTK_IDÚRPT_DaterY r% éb3é3r r ro u 饡驦gfffffC@g®GázÔD@gÃõ(\>@)r r ÚSTK_Namer )r r )r r ÚinnerT)ÚhowZ
|
left_indexZ right_indexr ZQT_Close)ZTClose_xZTClose_yr/ r r r F)Zdrop)
|
r r r& r r r' Zdtypesrl r8 r9 )r Zdf4Zdf5ÚkrN rO r r r Ú"test_rename_with_duplicate_columnso sJ ÿþüþù ùþ õ z-TestRename.test_rename_with_duplicate_columnsc C sx t t d¡ dd¡dddddgd}d d
|
d d }|j|d }t t d¡ dd¡dddddgd d
|
d gd}t ||¡ d S )Né ro rp FTrj rq r/ r1 r2 rh rk r% r )r rJ rs Zreshaper r8 r9 )r rD r ÚresÚexpr r r Útest_rename_boolean_index s $ ýz$TestRename.test_rename_boolean_indexN) Ú__name__Ú
|
__module__Ú __qualname__r r* rE r< ÚmarkZ parametrizer rP rc rf rg rn rt r{ r' r( r} r~ r r r
r r r r r r£ r r r r r sD )þþ
|
F ýþ
|
" )r )Ú collectionsr r ÚnumpyrJ r< Zpandasr r r r Zpandas._testingZ_testingr8 r r r r r Ú<module> s
|
