a
    Gb                      @   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                	   @   sX   e Zd Zejdg d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z0 daysz2 daysz4 daysfreq2D   Z48H0   c                 C   sH   t |}||j_|j|ksJ t|jts.J d |j_|jd u sDJ d S )N)r   _datar	   
isinstancer   )selfr   r	   idx r   Mlib/python3.9/site-packages/pandas/tests/indexes/timedeltas/test_freq_attr.pytest_freq_setter   s    zTestFreq.test_freq_setterc                 C   s   t g d}d}tjt|d d|j_W d    n1 s<0    Y  d}tjt|d d|j_W d    n1 sv0    Y  tjtdd d|j_W d    n1 s0    Y  d S )	Nr   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   sR   t g ddd}|j}t |d }|jd u s2J |jdks@J |jdksNJ d S )Nr   r
   )r	   )r   r   Z
_with_freqr	   )r   ZtdiZtdaZtdi2r   r   r   test_freq_view_safe1   s    zTestFreq.test_freq_view_safeN)__name__
__module____qualname__r   Z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   