a
    Pf                     @   s:   d dl mZ d dlmZ G dd deZG dd dZdS )    )CategoricalNc                   @   s   e Zd ZdS )SubclassedCategoricalN)__name__
__module____qualname__ r   r   f/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/arrays/categorical/test_subclass.pyr      s   r   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestCategoricalSubclassingc                 C   s2   t g d}t|t sJ t|tg d d S )Nabc)r   
isinstancetmassert_categorical_equalr   )selfscr   r   r   test_constructor
   s    z+TestCategoricalSubclassing.test_constructorc                 C   sF   t g dg d}t|t s"J tg dg d}t|| d S )N)   r      r
   )r   Z
from_codesr   r   r   r   )r   r   expr   r   r   test_from_codes   s    z*TestCategoricalSubclassing.test_from_codesc                 C   sH   t g d}|jdd d d}t|t s,J tg d}t|| d S )Nr
   c                 S   s   |   S )N)upper)xr   r   r   <lambda>       z5TestCategoricalSubclassing.test_map.<locals>.<lambda>)Z	na_action)ABC)r   mapr   r   r   r   )r   r   resr   r   r   r   test_map   s
    z#TestCategoricalSubclassing.test_mapN)r   r   r   r   r   r!   r   r   r   r   r	   	   s   r	   )Zpandasr   Zpandas._testingZ_testingr   r   r	   r   r   r   r   <module>   s   