U
    >vfV                     @   sN   d dl Zd dlZd dlmZmZmZmZmZm	Z	 d dl
mZ G dd dZdS )    N)DatetimeIndexIntervalIndexNaTPeriodSeries	Timestampc                   @   sV   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Ze	j
dddgdd ZdS )
TestDropnac              	   C   sv   t g td}t| dks t|jdd}|d ks8tt|dksHtd}tjt|d |jdd W 5 Q R X d S )	N)dtyper   Tinplacez&No axis named 1 for object type Series)match   )Zaxis)r   objectlendropnaAssertionErrorpytestZraises
ValueError)selfserreturn_valuemsg r   K/tmp/pip-unpacked-wheel-vdrwu74i/pandas/tests/series/methods/test_dropna.pytest_dropna_empty   s    zTestDropna.test_dropna_emptyc                 C   s^   t j|d d< | }|j|jks&t|j}| }|jdd}|d ksLt|j|ksZtd S )N   Tr
   )npnanr   namer   copy)r   Zdatetime_seriesresultr   tsr   r   r   r   test_dropna_preserve_name   s    z$TestDropna.test_dropna_preserve_namec                 C   sz   t dddgddt dddgddfD ]P}| }t|| ||k	sHt| }|jdd}|d kshtt|| q$d S )	Nr         x)r   FTr
   )r   r   tmassert_series_equalr   r   )r   r   r    s2r   r   r   r   test_dropna_no_nan'   s    zTestDropna.test_dropna_no_nanc              	   C   sV   t tjdddgttjdddgtjdddg}| }|jdd  }t|| d S )Nr   r#   r$   r   )	r   r   r   r   Zfrom_arraysr   Zilocr&   r'   r   r   r    expectedr   r   r   test_dropna_intervals5   s    z TestDropna.test_dropna_intervalsc                 C   sF   t tdddtdddg}| }t tdddg}t|| d S )Nz2011-01M)freqr   )r   r   r   r&   r'   r*   r   r   r   test_dropna_period_dtype?   s    z#TestDropna.test_dropna_period_dtypec                 C   s   t tdttdtg}| }t tdtdgddgd}t|| tdtdtgdd}t |}|jdksrt| }t tdddtdddgddgd}|jdkstt|| d S )	Nz2011-01-01 10:00z2011-01-03 10:00r   r#   indexz
Asia/Tokyo)tzzdatetime64[ns, Asia/Tokyo])	r   r   r   r   r&   r'   r   r	   r   )r   r   r    r+   idxr   r   r   test_datetime64_tz_dropnaG   s8     
 

z$TestDropna.test_datetime64_tz_dropnavalr   g      ?c                 C   s\   t dd|gdddgd}|jdd}t dd|g}t|| |jddd t|| d S )Nr   r#   r$   r0   T)ignore_index)r6   r   )r   r   r&   r'   )r   r5   r   r    r+   r   r   r   test_dropna_ignore_indexh   s    z#TestDropna.test_dropna_ignore_indexN)__name__
__module____qualname__r   r"   r)   r,   r/   r4   r   markZparametrizer7   r   r   r   r   r      s   

!r   )Znumpyr   r   Zpandasr   r   r   r   r   r   Zpandas._testingZ_testingr&   r   r   r   r   r   <module>   s    