U
    sVÞcæ  ã                   @   s6   d dl Zd dlmZmZ d dlmZ G dd„ dƒZdS )é    N)ÚCategoricalÚSeriesc                   @   s   e Zd Zdd„ Zdd„ ZdS )ÚTestCategoricalComparisonsc                 C   s>   t tdddtjgƒƒ}t ddddgƒ}||k}t ||¡ d S )NÚaÚbÚcTF)r   r   ÚnpÚnanÚtmÚassert_series_equal)ÚselfÚcatÚexpectedÚresult© r   úL/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/arithmetic/test_categorical.pyÚtest_categorical_nan_equality   s    z8TestCategoricalComparisons.test_categorical_nan_equalityc                 C   sV   t dddddgƒ}t dddddgƒ}|dk}t ||¡ | d¡dk}t ||¡ d S )N)r   r   )r   é   )r   r   )r   r   TFÚcategory)r   r
   r   Zastype)r   Zserr   r   r   r   r   Útest_categorical_tuple_equality   s    z:TestCategoricalComparisons.test_categorical_tuple_equalityN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   
   s   r   )	Znumpyr   Zpandasr   r   Zpandas._testingZ_testingr
   r   r   r   r   r   Ú<module>   s   