a
    Pf                     @   s>   d dl Z d dlmZ d dlmZmZmZmZ G dd dZdS )    N)TimedeltaIndex)
DateOffsetDayHourMonthEndc                	   @   s`   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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   g/var/www/ai-form-bot/venv/lib/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 }td}d}tjt|d || W d    n1 sB0    Y  tjt|d |j| W d    n1 s|0    Y  d S )N   z(TimedeltaArray/Index freq must be a Tickmatch)r   r   pytestraises	TypeError
_with_freqr   )r   r   offmsgr   r   r   "test_with_freq_empty_requires_tick   s    (z+TestFreq.test_with_freq_empty_requires_tickc                 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 5Dr   Z5Dz,<2 \* BusinessDays> is a non-fixed frequencyZ2BzInvalid frequencyZfoo)r   r   r   
ValueErrorr   r
   )r   r   r   r   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   r   r
   )r   ZtdiZtdaZtdi2r   r   r   test_freq_view_safe<   s    zTestFreq.test_freq_view_safeN)__name__
__module____qualname__r   markZparametrizer   r   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   r   <module>   s   