3
Ud                 @   s:   d dl Zd dlZd dljZddlmZ G dd deZ	dS )    N   )BaseExtensionTestsc               @   sl   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd ZdS )BaseMissingTestsc             C   s~   t jddg}tj|}tj|| tj|j }tj|}| j|| tj|jddgj }tjg t	d}| j|| d S )NTFr   r   )dtype)
nparraypdisnatmZassert_numpy_array_equalSeriesassert_series_equalZdropbool)selfdata_missingexpectedresult r   W/var/www/html/virt/lib64/python3.6/site-packages/pandas/tests/extension/base/missing.py	test_isna
   s    

zBaseMissingTests.test_isnac             C   s"   |j  }|dg }| j|| d S )Nr   )dropnaassert_extension_array_equal)r   r   r   r   r   r   r   test_dropna_array   s    
z"BaseMissingTests.test_dropna_arrayc             C   s.   t j|}|j }|jdg }| j|| d S )Nr   )r   r   r   ilocr   )r   r   serr   r   r   r   r   test_dropna_series   s    
z#BaseMissingTests.test_dropna_seriesc             C   s   t jd|i}|j }|jdg }| j|| |jdd}t jddgd}| j|| t j|dtjgd}|j }|jd d }| j|| d S )NAr   columns)Zaxisr   )index)r   B)r   	DataFramer   r   assert_frame_equalr   nan)r   r   dfr   r   r   r   r   test_dropna_frame$   s    z"BaseMissingTests.test_dropna_framec             C   s,   |d }|j |}|j |}| j|| d S )Nr   )fillnar   )r   r   Zvalidr   r   r   r   r   test_fillna_scalar7   s    

z#BaseMissingTests.test_fillna_scalarc             C   sR   |j dddddg}tj|jddd}tj|j dddddg}| j|| d S )Nr   r   ffill   )methodlimit)taker   r   r$   r   )r   r   arrr   r   r   r   r   test_fillna_limit_pad=   s    z&BaseMissingTests.test_fillna_limit_padc             C   sR   |j dddddg}tj|jddd}tj|j dddddg}| j|| d S )Nr   r   Zbackfillr'   )r(   r)   )r*   r   r   r$   r   )r   r   r+   r   r   r   r   r   test_fillna_limit_backfillC   s    z+BaseMissingTests.test_fillna_limit_backfillc             C   sr   |d }t j|}|j|}t j|j||g|jd}| j|| |j|}| j|| |j|}| j|| d S )Nr   )r   )r   r   r$   _from_sequencer   r   )r   r   
fill_valuer   r   r   r   r   r   test_fillna_seriesI   s    



z#BaseMissingTests.test_fillna_seriesc             C   sZ   |d }|dkr|d d d }t j|j|d}t j|j||g|jd}| j|| d S )Nr   r&   )r(   )r   )r   r   r$   r.   r   r   )r   r   Zfillna_methodr/   r   r   r   r   r   test_fillna_series_method]   s    z*BaseMissingTests.test_fillna_series_methodc             C   sV   |d }t j|ddgdj|}t j|j||g|jdddgd}| j|| d S )Nr   r'   )r   r   )r   )r   r   r$   r.   r   r    )r   r   r/   r   r   r   r   r   test_fillna_framel   s    z"BaseMissingTests.test_fillna_framec             C   sP   t j|tjgt| djddi}t j|dgt| d}| j|| d S )N)r   r   r   g        )r   r   r   r!   lenr$   r    )r   datar   r   r   r   r   test_fillna_fill_other|   s    &z'BaseMissingTests.test_fillna_fill_otherc          
   C   sB   t j|}|j }t jdd |j }W d Q R X | j|| d S )Nzmode.use_inf_as_naT)r   r   r	   Zoption_contextr   )r   r   r   r   r   r   r   r   test_use_inf_as_na_no_effect   s
    
z-BaseMissingTests.test_use_inf_as_na_no_effectN)__name__
__module____qualname__r   r   r   r#   r%   r,   r-   r0   r2   r3   r6   r7   r   r   r   r   r   	   s   r   )
Znumpyr   Zpandasr   Zpandas._testingZ_testingr
   baser   r   r   r   r   r   <module>   s   
