U
    +if                     @   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                
   @   s\   e Zd Zejdddgejdedddded	dd
dedddgddgdd ZdS )
TestRepeat	use_numpyTFindexz
2000-01-01   D)Zperiodsfreqz
2001-01-01Z2Dz2001-01ZNaTz2003-01M)r	   c                 C   sL   t dd |D }|r"t|dn|d}t|| |j|jksHtd S )Nc                 S   s   g | ]}t d D ]}|qqS )r   )range).0Zper_ r   b/home/mars/bis/venv/lib/python3.8/site-packages/pandas/tests/indexes/period/methods/test_repeat.py
<listcomp>   s     
  z2TestRepeat.test_repeat_freqstr.<locals>.<listcomp>r   )r   nprepeattmZassert_index_equalZfreqstrAssertionError)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   