o
    g                     @   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   s   t g td}t| dksJ |jdd}|d u sJ t|dks$J d}tjt|d |jdd W d    d S 1 s?w   Y  d S )	N)dtyper   Tinplacez&No axis named 1 for object type Seriesmatch   )axis)r   objectlendropnapytestraises
ValueError)selfserreturn_valuemsg r   e/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/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J |j}| }|jdd}|d u s&J |j|ks-J d S )N   Tr
   )npnanr   namecopy)r   datetime_seriesresultr    tsr   r   r   r   test_dropna_preserve_name   s   z$TestDropna.test_dropna_preserve_namec                 C   sv   t g dddt g dddfD ](}| }t|| ||us"J | }|jdd}|d u s2J t|| qd S )Nr         x)r    )FTFTr
   )r   r   tmassert_series_equalr!   )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   from_arraysr   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+   r0   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s9J | }t tdddtdddgddgd}|jdksVJ 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#   r1   idxr   r   r   test_datetime64_tz_dropnaH   s4   

z$TestDropna.test_datetime64_tz_dropnac                 C   sd   t g d}d}tjt|d |d}W d    n1 sw   Y  t g d}t|| d S )Nr&   zRIn a future version of pandas all arguments of Series\.dropna will be keyword-onlyr   r   )r   r*   assert_produces_warningFutureWarningr   r+   )r   r   r   r#   r1   r   r   r    test_dropna_pos_args_deprecationi   s   z+TestDropna.test_dropna_pos_args_deprecationN)
__name__
__module____qualname__r   r%   r-   r2   r5   r9   r<   r   r   r   r   r      s    

!r   )numpyr   r   pandasr   r   r   r   r   r   pandas._testing_testingr*   r   r   r   r   r   <module>   s
     