a
    Pf                     @   s2   d dl Z d dlZd dlmZmZ G dd dZdS )    N)SeriesTimedeltaIndexc                   @   s>   e Zd Zdd Zejdddgdd Zdd	 Zd
d Z	dS )TestTimedeltaIndexRenderingc                 C   sL   t dgddd}t|}d}||ks,J tt|}d}||ksHJ d S )N1 daysDfreqsz<TimedeltaIndex(['1 days'], dtype='timedelta64[s]', freq='D')z 0   1 days
dtype: timedelta64[s])r   Zas_unitreprr   )selfZtdiresultexpectedZresult2Z	expected2 r   e/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/timedeltas/test_formats.pytest_repr_round_days_non_nano   s    z9TestTimedeltaIndexRendering.test_repr_round_days_non_nanomethod__repr____str__c                 C   s   t g dd}t dgdd}t ddgdd}t g ddd}t g d}d}d}d	}	d
}
d}tddP t|||||g|||	|
|gD ] \}}t|| }||ksJ qW d    n1 s0    Y  d S )Nr   r   r   2 daysr   r   3 daysz1 days 00:00:01r   r   z5TimedeltaIndex([], dtype='timedelta64[ns]', freq='D')z=TimedeltaIndex(['1 days'], dtype='timedelta64[ns]', freq='D')zGTimedeltaIndex(['1 days', '2 days'], dtype='timedelta64[ns]', freq='D')zQTimedeltaIndex(['1 days', '2 days', '3 days'], dtype='timedelta64[ns]', freq='D')zmTimedeltaIndex(['1 days 00:00:01', '2 days 00:00:00', '3 days 00:00:00'], dtype='timedelta64[ns]', freq=None)display.width,  )r   pdoption_contextzipgetattr)r   r   idx1idx2idx3idx4idx5exp1exp2exp3exp4exp5idxr   r   r   r   r   test_representation   s$    z/TestTimedeltaIndexRendering.test_representationc                 C   s   t g dd}t dgdd}t ddgdd}t g ddd}t g d}d}d}d	}d
}	d}
tddP t|||||g||||	|
gD ] \}}tt|}||ksJ qW d    n1 s0    Y  d S )Nr   r   r   r   r   r   z"Series([], dtype: timedelta64[ns])z!0   1 days
dtype: timedelta64[ns]z,0   1 days
1   2 days
dtype: timedelta64[ns]z70   1 days
1   2 days
2   3 days
dtype: timedelta64[ns]zR0   1 days 00:00:01
1   2 days 00:00:00
2   3 days 00:00:00
dtype: timedelta64[ns]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   test_representation_to_series7   s"    z9TestTimedeltaIndexRendering.test_representation_to_seriesc                 C   s   t g dd}t dgdd}t ddgdd}t g ddd}t g d}d}d}d	}d
}	d}
t|||||g||||	|
gD ]\}}| }||ksxJ qxd S )Nr   r   r   r   r   r   z!TimedeltaIndex: 0 entries
Freq: Dz3TimedeltaIndex: 1 entries, 1 days to 1 days
Freq: Dz3TimedeltaIndex: 2 entries, 1 days to 2 days
Freq: Dz3TimedeltaIndex: 3 entries, 1 days to 3 days
Freq: Dz=TimedeltaIndex: 3 entries, 1 days 00:00:01 to 3 days 00:00:00)r   r   Z_summaryr*   r   r   r   test_summaryT   s    z(TestTimedeltaIndexRendering.test_summaryN)
__name__
__module____qualname__r   pytestmarkZparametrizer)   r+   r,   r   r   r   r   r   
   s
   
r   )r0   Zpandasr   r   r   r   r   r   r   r   <module>   s   