
    bMh                     L    d dl Zd dlZd dlmZmZ d dlmZ  G d d          Z	dS )    N)
MultiIndexSeriesc                        e Zd Zd Zd Zd ZdS )
TestRepeatc                 T   t          t          j                            d                              d          g d          }|                    d          }t          |j                            d          |j        j                            d                    }t          j	        ||           g d}|                    |          }t          |j                            |          |j        j                            |                    }t          j	        ||           d S )N      )abcindex   )r   r	      )
r   nprandomdefault_rngstandard_normalrepeatvaluesr   tmassert_series_equal)selfserrepsexpto_reps        g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/series/methods/test_repeat.pytest_repeatzTestRepeat.test_repeat   s    RY**1--==a@@XXXzz!}}SZ&&q))1A1H1H1K1KLLL
tS)))zz&!!SZ&&v..ci6F6M6Mf6U6UVVV
tS)))))    c                    t          t          j        d          d          }t          |j                            d          d|j        j                            d                    }t          j        t          j        |d          |           d}t          j	        t          |          5  t          j        |dd	           d d d            d S # 1 swxY w Y   d S )
Nr	   x)namer   )r#   r   z%the 'axis' parameter is not supported)matchr   )axis)r   r   aranger   r   r   r   r   pytestraises
ValueError)r   r   expectedmsgs       r   test_numpy_repeatzTestRepeat.test_numpy_repeat   s   RYq\\,,,Ja  s#)2B2I2I!2L2L
 
 
 	rya00(;;;5]:S111 	& 	&Ic11%%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&s   0CCCc                     t          j        g d          }g d}t          ||          }|                    d          j        dt          |          z  fk    sJ d S )N))   r   )r	   r   )r      )      )r
   r   r   dr   r	   )r   from_tuplesr   r   shapelen)r   m_idxdatam_dfs       r   test_repeat_with_multiindexz&TestRepeat.test_repeat_with_multiindex#   sg    &'G'G'GHH###d%((({{1~~#CII'7777777r    N)__name__
__module____qualname__r   r,   r9    r    r   r   r      sA        
* 
* 
*	& 	& 	&8 8 8 8 8r    r   )
numpyr   r'   pandasr   r   pandas._testing_testingr   r   r=   r    r   <module>rB      s                     8 8 8 8 8 8 8 8 8 8r    