U
|
®ý°d ã @ s² d dl Z d dlmZ d dlmZ e j deddddgeddddgdddddgd g¡d
|
d Ze j d d dg¡dd Z dd Z
|
dd Z dd Z e j ddddg¡dd Z dS ) é N)Ú CategoricalÚcé é é é é ©Ú
|
categoriesc C s t | | ¡ d S )N)ÚtmÚassert_categorical_equal)r © r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/util/test_assert_categorical_equal.pyÚtest_categorical_equal s r Úcheck_category_orderTFc C s t ddddgddddgd}t ddddgddddgd}d| i}| rtd}tjt|d tj||f| W 5 Q R X ntj||f| d S ) Nr r r r r r zÄCategorical\.categories are different
|
|
Categorical\.categories values are different \(100\.0 %\)
|
\[left\]: Index\(\[1, 2, 3, 4\], dtype='int64'\)
|
\[right\]: Index\(\[4, 3, 2, 1\], dtype='int64'\)©Úmatch©r ÚpytestZraisesÚAssertionErrorr r )r Úc1Úc2ÚkwargsÚmsgr r r Ú%test_categorical_equal_order_mismatch s r c C sN d} t ddddg}t ddddg}tjt| d t ||¡ W 5 Q R X d S )NzÃCategorical\.categories are different
|
|
Categorical\.categories values are different \(25\.0 %\)
|
\[left\]: Index\(\[1, 2, 3, 4\], dtype='int64'\)
|
\[right\]: Index\(\[1, 2, 3, 5\], dtype='int64'\)r r r r r r r )r r r r r r Ú*test_categorical_equal_categories_mismatch! s
|
r c C sb ddddg} d}t ddddg| d}t ddddg| d}tjt|d t ||¡ W 5 Q R X d S )Nr r r r zCategorical\.codes are different
|
|
Categorical\.codes values are different \(50\.0 %\)
|
\[left\]: \[0, 1, 3, 2\]
|
\[right\]: \[0, 1, 2, 3\]r r r )r
|
r r r r r r Ú%test_categorical_equal_codes_mismatch/ s r c C sR ddddg} d}t | dd}t | dd}tjt|d t ||¡ W 5 Q R X d S )
|
Nr r r r z]Categorical are different
|
|
Attribute "ordered" are different
|
\[left\]: False
|
\[right\]: TrueF©ZorderedTr r )Údatar r r r r r Ú'test_categorical_equal_ordered_mismatch> s r ÚobjÚindexZfooÚpandasc C s\ ddddg}| d}t |dd}t |dd}tjt|d tj||| d
|
W 5 Q R X d S ) Nr r r r zR are different
|
|
Attribute "ordered" are different
|
\[left\]: False
|
\[right\]: TrueFr Tr )r r )r r r r r r r r Ú&test_categorical_equal_object_overrideM s
|
r# )r r" r Zpandas._testingZ_testingr ÚmarkZ parametrizer r r r r r# r r r r Ú<module> s ,þ
|
|
|
