a
    PfY                     @   s:   d dl Z d dlmZmZmZ d dlmZ G dd dZdS )    N)DatetimeIndexSeries
date_rangec                   @   sP   e Zd Zdd Zejdg ddd Zdd Zejdg dd	d
 Z	dS )
TestDeletec           
      C   s   t dddd|d}t dddd|d}t dddd|d}tg dd dd	|}|||||d
}| D ]>\}}||}	t|	| |	j|jksJ |	j|jksfJ qft	j
ttfdd |d W d    n1 s0    Y  d S )N
2000-01-01   ZMEidxstartperiodsfreqnameunitz
2000-02-01   )z
2000-01-31z
2000-03-31z
2000-04-30z
2000-05-31r   r   )r   r      zout of bounds)match)r   r   as_unititemsdeletetmassert_index_equalr   r   pytestZraises
IndexError
ValueError)
selfr   r   Z
expected_0Z
expected_4Z
expected_1casesnexpectedresult r"   k/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/datetimes/methods/test_delete.pytest_delete   s:    



zTestDelete.test_deletetz)Nz
Asia/Tokyoz
US/Pacificc                 C   s   t dddd|d}t dddd|d}|d}t|| |j|jksJJ |jdksXJ |j|jkshJ t dddd|d}|d	}t|| |j|jksJ |jdksJ |j|jksJ d S )
N2000-01-01 09:00
   hr   )r
   r   r   r   r%   z2000-01-01 10:00	   r   r   )r   r   r   r   r   Zfreqstrr%   )r   r%   r   r    r!   r"   r"   r#   test_delete21   s&    




zTestDelete.test_delete2c           
      C   s   t dddd|d}t dddd|d}t dddd|d}tg dd dd	|}|||d
}| D ]\}}||}	t|	| |	j|jksJ |	j|jksJ |t	|d |d d }	t|	| |	j|jksJ |	j|jksbJ qbd S )Nr   r'   Dr   r	   z
2000-01-04   )r   z
2000-01-02z
2000-01-03z
2000-01-07z
2000-01-08z
2000-01-09z
2000-01-10r   ))r   r      )r,      r)   )   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   slice)
r   r   r   Zexpected_0_2Zexpected_7_9Zexpected_3_5r   r   r    r!   r"   r"   r#   test_delete_sliceI   s:    


	
zTestDelete.test_delete_slicec                 C   s   t dddd||d}td|d}||jd d j}|dd  }t|| |j|jks^J |j|jksnJ |j|jks~J ||jg d	 j}|d d d
 	d }t|| |j|jksJ |j|jksJ |j|jksJ d S )Nr&   r'   r(   r   )r   r   r   r%   r   r   )indexr   )r   r/   r   r,   r)   r-   )
r   r   Zdropr2   r   r   r   r   r%   Z
_with_freq)r   r%   r   Zdtitsr!   r    r"   r"   r#   test_delete_slice2v   s&    zTestDelete.test_delete_slice2N)
__name__
__module____qualname__r$   r   markZparametrizer*   r1   r4   r"   r"   r"   r#   r      s   %
-r   )	r   Zpandasr   r   r   Zpandas._testingZ_testingr   r   r"   r"   r"   r#   <module>   s   