
    KEf                     D    d dl Zd dlZd dlmZmZ d dlmZ  G d d      Z	y)    N)
MultiIndexSeriesc                       e Zd Zd Zd Zd Zy)
TestRepeatc                 B   t        t        j                  j                  d      j	                  d      g d      }|j                  d      }t        |j                  j                  d      |j                  j                  j                  d            }t        j                  ||       g d}|j                  |      }t        |j                  j                  |      |j                  j                  j                  |            }t        j                  ||       y )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        `/var/www/VTS_Report/venv/lib/python3.12/site-packages/pandas/tests/series/methods/test_repeat.pytest_repeatzTestRepeat.test_repeat   s    RYY**1-==a@Xzz!}SZZ&&q)1A1A1H1H1KL
tS)zz&!SZZ&&v.cii6F6F6M6Mf6UV
tS)    c                    t        t        j                  d      d      }t        |j                  j	                  d      d|j
                  j                  j	                  d            }t        j                  t        j                  |d      |       d}t        j                  t        |      5  t        j                  |dd	       d d d        y # 1 sw Y   y xY w)
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    RYYq\,JJa s#))2B2B2I2I!2L
 	ryya0(;5]]:S1 	&IIc11%	& 	& 	&s   2CCc                     t        j                  g d      }g d}t        ||      }|j                  d      j                  dt        |      z  fk(  sJ y )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#   sK    &&'GH#d%({{1~##CI'7777r    N)__name__
__module____qualname__r   r,   r9    r    r   r   r      s    
*	&8r    r   )
numpyr   r'   pandasr   r   pandas._testing_testingr   r   r=   r    r   <module>rB      s      8 8r    