U
    sVÞcá  ã                   @   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)Ú
MultiIndexÚSeriesc                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )Ú
TestRepeatc                 C   s   t tj d¡dddgd}| d¡}t |j d¡|jj d¡d}t ||¡ dddg}| |¡}t |j |¡|jj |¡d}t ||¡ d S )	Né   ÚaÚbÚc©Úindexé   é   é   )	r   ÚnpÚrandomZrandnÚrepeatÚvaluesr
   ÚtmÚassert_series_equal)ÚselfÚserZrepsÚexpZto_rep© r   úK/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/series/methods/test_repeat.pyÚtest_repeat   s    


zTestRepeat.test_repeatc              	   C   sx   t t d¡dd}t |j d¡d|jj d¡d}t t |d¡|¡ d}tj	t
|d tj|ddd	 W 5 Q R X d S )
Nr   Úx)Únamer   )r   r
   z%the 'axis' parameter is not supported)Úmatchr   )Zaxis)r   r   Zaranger   r   r
   r   r   ÚpytestZraisesÚ
ValueError)r   r   ÚexpectedÚmsgr   r   r   Útest_numpy_repeat   s    
  ÿzTestRepeat.test_numpy_repeatc                 C   sL   t  ddddg¡}ddddg}t||d	}| d
¡jd
t|ƒ fksHt‚d S )N)é   r   )r   r   )r   é   )é   é   r   r   r   Údr	   r   )r   Úfrom_tuplesr   r   ÚshapeÚlenÚAssertionError)r   Zm_idxÚdataZm_dfr   r   r   Útest_repeat_with_multiindex#   s    z&TestRepeat.test_repeat_with_multiindexN)Ú__name__Ú
__module__Ú__qualname__r   r!   r,   r   r   r   r   r      s   r   )
Znumpyr   r   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   