o
    �¨Êh:  ã                   @   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 s8w   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Ú	TypeErrorÚastypeÚobject)ÚselfÚidxÚexpÚcatÚmsgÚresultÚexpected© r   ú^/var/www/html/env/lib/python3.10/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 ||¡sJ ‚|j}| d¡}|j|jus)J ‚|jjd u s1J ‚t ||¡r9J ‚d S )N)r   é   r    r   )r   r   r   Úshares_memoryr   Ú_ndarrayÚbase)r   Úcir   r   r   r   r   Útest_fillna_copies_with_no_nas   s   

z)TestFillNA.test_fillna_copies_with_no_nasc                 C   sj   t g d¢ƒ}|j}d}| d¡}t ||¡ tjt|d�� | d¡ W d   ƒ d S 1 s.w   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   Úpandasr   Úpandas._testingÚ_testingr   r   r   r   r   r   Ú<module>   s
    