a
    Pf                     @   s>   d dl Zd dlZd dlmZmZ d dlmZ G dd dZ	dS )    N)PeriodIndexperiod_rangec                
   @   sZ   e Zd Zejdddgejdedddded	dd
deg dddgdd ZdS )
TestRepeat	use_numpyTFindexz
2000-01-01   D)Zperiodsfreqz
2001-01-01Z2D)z2001-01ZNaTz2003-01M)r	   c                 C   sL   t dd |D }|r"t|dn|d}t|| |j|jksHJ d S )Nc                 S   s   g | ]}t d D ]}|qqS )r   )range).0Zper_ r   h/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexes/period/methods/test_repeat.py
<listcomp>       z2TestRepeat.test_repeat_freqstr.<locals>.<listcomp>r   )r   nprepeattmZassert_index_equalZfreqstr)selfr   r   expectedresultr   r   r   test_repeat_freqstr   s    zTestRepeat.test_repeat_freqstrN)	__name__
__module____qualname__pytestmarkZparametrizer   r   r   r   r   r   r   r      s   r   )
numpyr   r   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   