a
    Pf:                     @   s:   d dl Zd dlZd dlmZ d dlmZ G dd dZdS )    N)CategoricalIndexc                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestFillNAc                 C   s   t dtjddgdd}t g ddd}t|d| |j}d}tjt	|d |d W d    n1 sp0    Y  |d}|
td}t|| d S )	N      ?      @x)name)r   r   r   r   3Cannot setitem on a Categorical with a new categorymatchg       @)r   npnantmassert_index_equalfillna_datapytestraises	TypeErrorZastypeobject)selfidxexpcatmsgresultexpected r   e/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/categorical/test_fillna.pytest_fillna_categorical	   s    (
z"TestFillNA.test_fillna_categoricalc                 C   sv   t g d}|d}||us"J t||s2J |j}|d}|j|jusRJ |jjd u sbJ t||rrJ d S )N)r      r   r   )r   r   r   Zshares_memoryr   Z_ndarraybase)r   cir   r   r   r   r   test_fillna_copies_with_no_nas   s    

z)TestFillNA.test_fillna_copies_with_no_nasc                 C   sh   t g d}|j}d}|d}t|| tjt|d |d W d    n1 sZ0    Y  d S )N)      r$   r   Fr	   )r   r   r   r   r   r   r   r   )r   r!   r   r   resr   r   r   !test_fillna_validates_with_no_nas*   s    
z,TestFillNA.test_fillna_validates_with_no_nasN)__name__
__module____qualname__r   r"   r&   r   r   r   r   r      s   r   )	numpyr   r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   