U
|
ý°d§ ã @ s< d dl Z d dlmZ d dlmZ dd Zdd Zdd ZdS ) é N)ÚIndexc C s´ | d¡}tdd | jD }t |j|¡ | d¡}tdd | jD }t |j|¡ | d¡}tdd | jD }t |j|¡ | d¡}tdd | jD }t |j|¡ d S ) Núfoo#c S s g | ]}d | qS ©r © ©Ú.0Úcr r úhd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_add_prefix_suffix.pyÚ
|
<listcomp> s z*test_add_prefix_suffix.<locals>.<listcomp>ú#fooc S s g | ]}| d qS ©r r r r r r r
|
s ú%c S s g | ]}d | qS ©r r r r r r r
|
s c S s g | ]}| d qS r r r r r r r
|
s )Ú
|
add_prefixr ÚcolumnsÚtmÚassert_index_equalÚ
|
add_suffix)Ú float_frameÚ with_prefixÚexpectedÚ with_suffixZwith_pct_prefixÚwith_pct_suffixr r r Útest_add_prefix_suffix s
|
|
|
|
r c C sÄ | j ddd}tdd | jD }t |j|¡ | j ddd}tdd | jD }t |j|¡ | jddd}td d | jD }t |j|¡ | jddd}td
|
d | jD }t |j|¡ d S ) Nr r ©Zaxisc S s g | ]}d | qS r r r r r r r
|
s z/test_add_prefix_suffix_axis.<locals>.<listcomp>é c S s g | ]}d | qS r r r r r r r
|
s r c S s g | ]}| d qS r r r r r r r
|
$ s c S s g | ]}| d qS r r r r r r r
|
( s )r r Úindexr r r r )r r r r r r r Útest_add_prefix_suffix_axis s r c C sT t jtdd | jddd W 5 Q R X t jtdd | jddd W 5 Q R X d S )Nz)No axis named 2 for object type DataFrame)Úmatchr é r )ÚpytestZraisesÚ
|
ValueErrorr r )r r r r Ú#test_add_prefix_suffix_invalid_axis, s r" ) r Zpandasr Zpandas._testingZ_testingr r r r" r r r r Ú<module> s
|
|
