U
|
¬ý°dÐ ã @ s¬ d dl Zd dlZd dlZd dlmZ ej dddg¡ej ddddgdddd gg¡d
|
d Z
|
d d Z dd Z dd Z ej dg ejede g ¡g¡dd Zdd ZdS )é NÚorderedTFÚ
|
categoriesÚbÚaÚcÚdc C st t jddddd g| |d}t |¡\}}tjdddddgtjd }t jdddg| |d}t ||¡ t ||¡ d S )
|
Nr r r ©r r r é é éÿÿÿÿ©Zdtype© ÚpdÚ CategoricalZ factorizeÚnpÚarrayZintpÚtmÚassert_numpy_array_equalZassert_categorical_equal)r r ÚcatÚcodesÚuniquesÚexpected_codesÚexpected_uniques© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/arrays/categorical/test_algos.pyÚtest_factorize s ÿ ÿ r c C sf t ddd dg¡} t j| dd\}}tjddddgtjd}t ddg¡}t ||¡ t ||¡ d S ) Nr r T©Úsortr r r r r ©r r r r r r r r Útest_factorized_sort s r c C s~ t jddd dgdddgdd} t j| dd\}}tjdddd gtjd
|
}t jddgdddgdd}t ||¡ t ||¡ d S ) Nr r r Tr r r r r r r r r r r Ú test_factorized_sort_ordered" s
|
ÿ ÿ r c C st t ddtjg¡} | dtjg¡}tjdddgtd}t ||¡ | ddg¡}tjdddgtd}t ||¡ d S )Nr r TFr r ) r r r ÚnanÚisinr Úboolr r )r ÚresultÚexpectedr r r Útest_isin_cats1 s r& Úemptyr c C s: t ddg¡}tjddgtd}| | ¡}t ||¡ d S )Nr r Fr )r r r r r# r" r r )r' Úsr% r$ r r r Útest_isin_empty> s
|
r) c C sl t jdddgdd} d}tjt|d | ¡ W 5 Q R X | jdd }tjt|d | ¡ W 5 Q R X d S )
|
Nr r
|
é Úcategoryr zConvert to a suitable dtype)ÚmatchÚA)Úname)r ÚSeriesÚpytestZraisesÚ TypeErrorZdiffZto_frame)ZserÚmsgZdfr r r Ú test_diffG s r3 )Únumpyr r0 Zpandasr Zpandas._testingZ_testingr ÚmarkZ parametrizer r r r& r/ Úobjectr r) r3 r r r r Ú<module> s
|
|
|
