U
    +if                      @   s:   d dl Z d dlmZ d dlmZmZmZ G dd dZdS )    N)TimedeltaIndex)
DateOffsetDayHourc                	   @   sZ   e Zd Zejddddgg gejddedded	gd
d Zdd Z	dd Z
dS )TestFreqvalues0 days2 days4 daysfreq2D   Z48H0   c                 C   sH   t |}||j_|j|kstt|jts.td |j_|jd ksDtd S )N)r   _datar   AssertionError
isinstancer   )selfr   r   idx r   a/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/indexes/timedeltas/test_freq_attr.pytest_freq_setter   s    zTestFreq.test_freq_setterc              	   C   s   t dddg}d}tjt|d d|j_W 5 Q R X d}tjt|d d|j_W 5 Q R X tjtd	d d
|j_W 5 Q R X d S )Nr   r	   r
   zPInferred frequency 2D from passed values does not conform to passed frequency 5D)matchZ5Dz,<2 \* BusinessDays> is a non-fixed frequencyZ2BzInvalid frequencyZfoo)r   pytestZraises
ValueErrorr   r   )r   r   msgr   r   r   test_freq_setter_errors   s    z TestFreq.test_freq_setter_errorsc                 C   sT   t dddgdd}|j}t |d }|jd ks4t|jdksBt|jdksPtd S )Nr   r	   r
   r   )r   )r   r   Z
_with_freqr   r   )r   ZtdiZtdaZtdi2r   r   r   test_freq_view_safe1   s    zTestFreq.test_freq_view_safeN)__name__
__module____qualname__r   markZparametrizer   r   r   r   r   r   r   r   r   r      s
   r   )r   Zpandasr   Zpandas.tseries.offsetsr   r   r   r   r   r   r   r   <module>   s   