U
|
ý°d] ã @ s: d dl Zd dlZd dlmZmZmZmZ G dd dZdS )é N)Ú CategoricalÚCategoricalIndexÚIndexÚ
|
MultiIndexc @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú
|
TestEqualsc C s" t ddgddgdd}t ddgdddgdd}| |¡s<t| |¡rJt| | t¡¡s^t| t¡ |¡srt||k ¡ st||k ¡ rt||k ¡ r¢t||k ¡ r²t||k ¡ sÂt||k ¡ sÒt|dk ¡ rât|tddgk ¡ sút||jk ¡ sttj t
|
dd |tdddgk W 5 Q R X d }tj t |d ||k W 5 Q R X tj t |d |t |jd
|
d k W 5 Q R X tj t |d |t |jt d d k W 5 Q R X t t ddddgd }| t d¡ræt| t t d¡sþt| t t ddd ¡rt| | ¡ ¡s.tt t dtjg dddgd }| t d¡r`t| t t d¡rxt| | ¡ ¡stt t dtjg dddgd }| t dtjg ¡rÆt| t t dtjg ¡sæt| t t dtjg dd ¡r
|
t| | ¡ ¡s td S )NÚaÚbT)Ú
|
categoriesÚorderedÚcé zLengths must match)Úmatchz>Categoricals can only be compared if 'categories' are the sameF)r
|
Úabc©r Zaabca)r ÚequalsÚAssertionErrorÚastypeÚobjectÚallr ÚvaluesÚpytestZraisesÚ
|
ValueErrorÚ TypeErrorr ÚlistÚcopyÚnpÚnan)ÚselfZci1Zci2ÚmsgÚci© r úcd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/categorical/test_equals.pyÚtest_equals_categorical sL $z"TestEquals.test_equals_categoricalc C sd t dgddgd}t dgddgd}t dgddgd}| |¡sDt| |¡rRt| |¡r`td S )NÚAÚBr ÚC)r r r )r r r r r r r! Ú!test_equals_categorical_unorderedC s z,TestEquals.test_equals_categorical_unorderedc C s8 t ddtjtjg}tdddtjg}| |¡r4td S )Nr# r$ ÚD)r r r r r r )r r Úotherr r r! Útest_equals_non_categoryL s z#TestEquals.test_equals_non_categoryc C s: t ddddgtdg¡}| ¡ d¡}| |¡r6td S )Nr# r$ r% r' é Úcategory)r Z from_arraysÚrangeZ to_flat_indexr r r )r Úmir r r r! Útest_equals_multiindexT s z!TestEquals.test_equals_multiindexN)Ú__name__Ú
|
__module__Ú __qualname__r" r& r) r. r r r r! r s 6 r ) Únumpyr r Zpandasr r r r r r r r r! Ú<module> s
|
