a
    Gb^	                     @   s.   d dl mZmZ d dlmZ G dd dZdS )    )TimedeltaIndextimedelta_rangeNc                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestTimedeltaIndexDeletec           	      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sZJ qZtt	t
f |d W d    n1 s0    Y  d S )Nz1 Days   DidxstartZperiodsfreqname2 Days   )z1 dayz3 dayz4 dayz5 dayr
   r   )r   r      )r   r   itemsdeletetmassert_index_equalr   r
   Zexternal_error_raised
IndexError
ValueError)	selfr   Z
expected_0Z
expected_4Z
expected_1casesnexpectedresult r   Jlib/python3.9/site-packages/pandas/tests/indexes/timedeltas/test_delete.pytest_delete	   s&    

z$TestTimedeltaIndexDelete.test_deletec           	      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sVJ qVd S )Nz1 days
   r   r   r   z4 days   )z1 dz2 dz3 dz7 dz8 dz9 dZ10dr   ))r   r      )r!      	   )   r   r   r   r   r   )	r   r   r   r   r   r   r   r
   slice)	r   r   Zexpected_0_2Zexpected_7_9Zexpected_3_5r   r   r   r   r   r   r   test_delete_slice&   s&    

z*TestTimedeltaIndexDelete.test_delete_slicec                 C   s(   t g d}|d}|jd u s$J d S )N)z1 Dayr   Nz3 Daysz4 Daysr"   )r   r   r
   )r   Ztdir   r   r   r   test_delete_doesnt_infer_freqB   s    
z6TestTimedeltaIndexDelete.test_delete_doesnt_infer_freqN)__name__
__module____qualname__r   r'   r(   r   r   r   r   r      s   r   )Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   