U
    ÏøïcT  ã                   @   s*   d dl mZ d dlmZ G dd„ dƒZdS )é    )ÚCategoricalNc                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestCategoricalSubclassingc                 C   s:   t  dddg¡}t|t jƒs t‚t  |tdddgƒ¡ d S )NÚaÚbÚc)ÚtmÚSubclassedCategoricalÚ
isinstanceÚAssertionErrorÚassert_categorical_equalr   )ÚselfÚsc© r   úQ/tmp/pip-unpacked-wheel-g7fro6k3/pandas/tests/arrays/categorical/test_subclass.pyÚtest_constructor   s    z+TestCategoricalSubclassing.test_constructorc                 C   sR   t j dddgdddg¡}t|t jƒs*t‚t dddgdddg¡}t  ||¡ d S )Né   r   é   r   r   r   )r   r   Z
from_codesr	   r
   r   r   )r   r   Úexpr   r   r   Útest_from_codes   s    z*TestCategoricalSubclassing.test_from_codesc                 C   sL   t  dddg¡}| dd„ ¡}t|t jƒs.t‚tdddgƒ}t  ||¡ d S )	Nr   r   r   c                 S   s   |   ¡ S )N)Úupper)Úxr   r   r   Ú<lambda>   ó    z5TestCategoricalSubclassing.test_map.<locals>.<lambda>ÚAÚBÚC)r   r   Úmapr	   r
   r   r   )r   r   Úresr   r   r   r   Útest_map   s
    z#TestCategoricalSubclassing.test_mapN)Ú__name__Ú
__module__Ú__qualname__r   r   r   r   r   r   r   r      s   r   )Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   