o
    թZh                     @   s2   d dl Z d dlmZ d dlmZ G dd dZdS )    N)TimedeltaArrayc                   @   s   e Zd Zdd Zdd ZdS )TestAccumulatorc                 C   sP   t jddgdd}tjtdd |d W d    d S 1 s!w   Y  d S )N1D2Dzm8[ns]dtypezcumprod not supported)matchZcumprod)r   _from_sequencepytestZraises	TypeError_accumulate)selfarr r   e/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/arrays/timedeltas/test_cumulative.pytest_accumulators_disallowed   s   "z,TestAccumulator.test_accumulators_disallowedc                 C   sJ   d| d}t jddg|d}|d}t jddg|d}t|| d S )Nzm8[]r   r   r   ZcumsumZ3D)r   r	   r   tmZassert_timedelta_array_equal)r   unitr   r   resultexpectedr   r   r   test_cumsum   s
   
zTestAccumulator.test_cumsumN)__name__
__module____qualname__r   r   r   r   r   r   r      s    r   )r
   Zpandas._testingZ_testingr   Zpandas.core.arraysr   r   r   r   r   r   <module>   s    