U
    sVc                     @   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                   @   sD   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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Z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-xj8nt62q/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_intervals6   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_dropnaH   s8     
 

z$TestDropna.test_datetime64_tz_dropnac              	   C   sT   t dddg}d}tjt|d |d}W 5 Q R X t dddg}t|| d S )Nr   r#   r$   zRIn a future version of pandas all arguments of Series\.dropna will be keyword-onlyr   r   )r   r&   Zassert_produces_warningFutureWarningr   r'   )r   r   r   r    r+   r   r   r    test_dropna_pos_args_deprecationi   s    z+TestDropna.test_dropna_pos_args_deprecationN)
__name__
__module____qualname__r   r"   r)   r,   r/   r3   r5   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    