o
    ҷhd                     @   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   s6   t g d}t|t jsJ t |tg d d S )Nabc)tmSubclassedCategorical
isinstanceassert_categorical_equalr   )selfsc r   `/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/arrays/categorical/test_subclass.pytest_constructor   s   z+TestCategoricalSubclassing.test_constructorc                 C   sJ   t jg dg d}t|t jsJ tg dg d}t || d S )N)   r      r   )r   r	   
from_codesr
   r   r   )r   r   expr   r   r   test_from_codes   s   z*TestCategoricalSubclassing.test_from_codesc                 C   sL   t g d}|jdd d d}t|t jsJ tg d}t || d S )Nr   c                 S   s   |   S )N)upper)xr   r   r   <lambda>   s    z5TestCategoricalSubclassing.test_map.<locals>.<lambda>)	na_action)ABC)r   r	   mapr
   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   )pandasr   pandas._testing_testingr   r   r   r   r   r   <module>   s    