U
|
ý°dj ã @ sB d dl Zd dlZd dlmZmZmZ d dlmZ G dd dZ
|
dS )é N)Ú DataFrameÚIndexÚ
|
MultiIndexc @ s e Zd Zdd Zdd Zdd Zej ddd d
|
d d fdd d
|
d d fddid d fddid d fdddd d fi d d fg¡dd Z dS )ÚTestDataFrameRenameAxisc C s | d¡}| ¡ }|j ddd }}|d ks0t|d ks<tt ||¡ |j ddd}| ¡ }|j dddd }}|d ks~t|d kstt ||¡ d S )NÚfooT)ÚinplaceÚbaré ©Úaxis)r r )Ú rename_axisÚcopyÚAssertionErrorÚtmÚassert_frame_equal)ÚselfZ float_frameÚexpectedÚresultZ return_valueZ no_return© r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_rename_axis.pyÚtest_rename_axis_inplace s
|
z0TestDataFrameRenameAxis.test_rename_axis_inplacec C sÀ t ddgddgd}tjtdd |jtdd W 5 Q R X tjtdd |jdd d
|
dd W 5 Q R X tjtdd |jtdd W 5 Q R X tjtdd |d t¡ W 5 Q R X d S ) Nr é )ÚAÚBz Use `.rename`©Úmatchr r
|
é
|
é )r r r )r ÚpytestÚraisesÚ
|
ValueErrorr Úid)r Údfr r r Útest_rename_axis_raises s z/TestDataFrameRenameAxis.test_rename_axis_raisesc C sÖ t jdddgddggddgd}tttt|d d
|
tt|D d |d }|jd dd}t |j t
|
ddgd d¡ |jd didd}t |j t
|
ddgdd¡ |jddid }|j j ddgksÈt |jtjdd}|j j ddgksìt |jddgd }|j j ddgkst | ¡ d¡jddgd}|jddd }|j jdksLt |j jdks^t tjtd d |jd d W 5 Q R X tjtd d |jd gd W 5 Q R X tjtd!d |jd d" W 5 Q R X d S )#NÚaÚbÚcr r ZllÚnn)Únamesc S s g | ] }|d qS )r r )Ú.0Úir r r Ú
|
<listcomp>2 s zCTestDataFrameRenameAxis.test_rename_axis_mapper.<locals>.<listcomp>)ÚxÚy)ÚindexÚcolsr
|
r, r- ©ÚnameÚnew)Úcolumnsr r r )r. r ZLLZNNZgoo)r3 Zmeh©r. r3 z Must passr ZwrongzLength of namesÚbogus)r5 )r Z from_productr ÚlistÚrangeÚlenr r Zassert_index_equalr3 r r. r( r ÚstrÚupperZ reset_indexZ set_indexZdropr1 r r Ú TypeErrorr )r Úmir" r Zsdfr r r Útest_rename_axis_mapper. s2 & ÿz/TestDataFrameRenameAxis.test_rename_axis_mapperz$kwargs, rename_index, rename_columnsNr )Zmapperr TFr r. r3 r4 c C s t tddd}t ddgdd}t d¡ dd ¡}t|||}|jf |}|rX| d ¡n|} |rj| d ¡n|}
|
t|| |
|
} t || ¡ d S )
|
NÚabcr r0 Zcol1Zcol2r é é r )
|
r r6 ÚnpZarangeZreshaper r Úrenamer r ) r ÚkwargsZ rename_indexZrename_columnsr. r3 Údatar" r Zexpected_indexZexpected_columnsr r r r Útest_rename_axis_noneY s z-TestDataFrameRenameAxis.test_rename_axis_none)
|
Ú__name__Ú
|
__module__Ú __qualname__r r# r= r ÚmarkZ parametrizerE r r r r r s + úþ r ) ÚnumpyrA r Zpandasr r r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
