o
    ŀgY                     @   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sJJ |	j|jksRJ q3t	j
ttfdd |d W d    d S 1 smw   Y  d S )N
2000-01-01   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   pytestraises
IndexError
ValueError)
selfr   r	   
expected_0
expected_4
expected_1casesnexpectedresult r'   f/var/www/html/myenv/lib/python3.10/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s%J |jdks,J |j|jks4J t dddd|d}|d	}t|| |j|jksPJ |jdksWJ |j|jks_J 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   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 ]C\}}||}	t|	| |	j|jksHJ |	j|jksPJ |t	|d |d d }	t|	| |	j|jkslJ |	j|jkstJ q1d 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      )r2      r.   )   r   r   r   r   r   )
r   r   r   r   r   r   r   r   r   slice)
r   r   r	   expected_0_2expected_7_9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s7J |j|jks?J ||jg d	 j}|d d d
 	d }t|| |j|jksbJ |j|jksjJ |j|jksrJ d S )Nr+   r,   r-   r	   )r   r   r   r*   r   r   )indexr   )r   r5   r   r2   r.   r3   )
r   r   dropr;   r   r   r   r   r*   
_with_freq)r   r*   r   dtitsr&   r%   r'   r'   r(   test_delete_slice2v   s&   zTestDelete.test_delete_slice2N)
__name__
__module____qualname__r)   r   markparametrizer0   r:   r@   r'   r'   r'   r(   r      s    %
-r   )	r   pandasr   r   r   pandas._testing_testingr   r   r'   r'   r'   r(   <module>   s    