U
    ¾mœd‡  ã                   @   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   úg/home/sam/Atlas/atlas_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   