U
    _{f                     @   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   s6   t ddg}tjtdd |d W 5 Q R X d S )N1D2Dzcumprod not supported)matchZcumprod)r   _from_sequence_not_strictpytestZraises	TypeError_accumulate)selfarr r   m/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/pandas/tests/arrays/timedeltas/test_cumulative.pytest_accumulators_disallowed   s    z,TestAccumulator.test_accumulators_disallowedc                 C   s6   t ddg}|d}t ddg}t|| d S )Nr   r   ZcumsumZ3D)r   r   r
   tmZassert_timedelta_array_equal)r   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   