3
UdQ	                 @   s,   d dl mZmZ d dlj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|d|d|d|i}xJ|j D ]>\}}|j|}tj|| |j|jkst|j|jkshtqhW tj	t
tf |jd W d Q R X d S )Nz1 Days   Didx)startperiodsfreqnamez2 Days   z1 dayz3 dayz4 dayz5 day)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   _/var/www/html/virt/lib64/python3.6/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dd	d
ddddgd dd}d|d|d|i}x|j D ]\}}|j|}tj|| |j|jkst|j|jkst|jt	|d |d d }tj|| |j|jkst|j|jksftqfW d S )Nz1 days
   r   r   )r   r	   r
   r   z4 days   z1 dz2 dz3 dz7 dz8 dz9 dZ10d)r
   r   r   r         	      r   r   )r   r   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}|jd}|jd ks*td S )Nz1 Dayz2 Daysz3 Daysz4 Daysr!   )r   r   r
   r   )r   Ztdir   r   r   r   test_delete_doesnt_infer_freq?   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   
