U
    b:                     @   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 ddddgdd}t|d| |j}d}tjt	|d |d W 5 Q R X |d}|
td}t|| d S )Ng      ?g      @x)name3Cannot 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   P/tmp/pip-unpacked-wheel-ck39h295/pandas/tests/indexes/categorical/test_fillna.pytest_fillna_categorical	   s    
z"TestFillNA.test_fillna_categoricalc                 C   sx   t dddg}|d}||k	s$tt||s4t|j}|d}|j|jk	sTt|jjd ksdtt||rttd S )Nr      )r   r   AssertionErrorr   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   sV   t dddg}|j}d}|d}t|| tjt|d |d W 5 Q R X d S )N      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   )	Znumpyr	   r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   