U
    sVc^	                     @   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dd	d
dgd dd}|||||d}| D ]>\}}||}t|| |j|jkst|j|jks^tq^t	t
tf |d W 5 Q R X 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   AssertionErrorr
   Zexternal_error_raised
IndexError
ValueError)	selfr   Z
expected_0Z
expected_4Z
expected_1casesnexpectedresult r   O/tmp/pip-unpacked-wheel-xj8nt62q/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dd	d
ddddgd dd}|||d}| D ]\}}||}t|| |j|jkst|j|jkst|t	|d |d d }t|| |j|jkst|j|jks`tq`d 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   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 ddd ddg}|d}|jd ks*td S )Nz1 Dayr   z3 Daysz4 Daysr#   )r   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   