U
|
¬ý°dK ã @ sÊ d dl Z d dlZd dlmZ d dlmZ e j ddddddgdfdddddgdfdddddgdfd d
|
dddgdfdgddddgdfddgddddgdfddgddddgdfd d d d dgdfd ddddgdfdgdgdddgdfddgd dgd ddgdfdddgdddgdddgdfdddddgdfdej dddgdfdd ddd gdfdddgddddgdfg¡dd Z
|
e j dddddgdfddddgdfddddgdfddddgdfg¡dd Z d d Z d d Z dS ) é N)Ú Categoricalz)to_replace,value,expected,flip_categoriesé é é Fé é é )r r r )r r Ú4Ú3Ú5Tc C s t jdddgdd}| | |¡}t j|dd}|j| |dd |r\|j |jjd d d
¡}tj||d d
|
tj||d d
|
d S ) Nr r r Úcategory©ÚdtypeT©ZinplaceéÿÿÿÿF©Zcheck_category_order)ÚpdÚSeriesÚreplaceÚcatZset_categoriesÚ
|
categoriesÚtmÚassert_series_equal)Ú
|
to_replaceÚvalueÚexpectedZflip_categoriesÚserÚresult© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_replace.pyÚ test_replace_categorical_series s r z-to_replace, value, result, expected_error_msgÚbÚcÚaz$Categorical.categories are differentÚdz+Categorical.categories length are differentc C s t ddg}t |}tj|dd | |¡j}t ||¡ | dkrftjt |d t ||¡ W 5 Q R X tj|ddj| |dd t ||¡ d S )Nr# r! F)Úcopy)ÚmatchTr )
|
r r r r Ú_valuesr Úassert_categorical_equalÚpytestZraisesÚAssertionError)r r r Zexpected_error_msgr r r r r Útest_replace_categorical1 s r+ c C s\ t tjddgdd} t | ¡ ddgdtjg¡j}t tjdtjgdd}t ||¡ d S )Nr# r! Ústringr r" ) r r Úarrayr r ÚNAr' r r( )r r r r r r Ú!test_replace_categorical_ea_dtypeK s r/ c C sn t jdddgdd} t jdddg| d}| dd¡}t jddgdd}t jdddg|d}tj||dd d S )Nr r r T)Zorderedr r )r ZCategoricalDtyper r r r )r r r Zexpected_dtyper r r r Ú test_replace_maintain_orderingS s r0 )r) Zpandasr r Zpandas._testingZ_testingr ÚmarkZ parametrizer. r r+ r/ r0 r r r r Ú<module> sF ëþ
|
ûþ
|
|
|
