a
    é´«jÝ  ã                   @   s.   d dl Z d dlZd dlmZ G dd„ dƒZdS )é    Nc                   @   sR   e Zd ZdZejeedœdd„Zejeedœdd„Z	e
j dd	d
g¡dd„ ƒZdS )ÚBaseAccumulateTestszj
    Accumulation specific tests. Generally these only
    make sense for numeric/boolean operations.
    )ÚserÚop_nameÚreturnc                 C   s   dS )NF© )Úselfr   r   r   r   úe/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/extension/base/accumulate.pyÚ_supports_accumulation   s    z*BaseAccumulateTests._supports_accumulation)r   r   Úskipnac              	   C   sd   z|  d¡}W n  ttfy.   |  t¡}Y n0 t||ƒ|d�}t||ƒ|d�}tj||dd� d S )NZfloat64©r
   F)Zcheck_dtype)ZastypeÚ	TypeErrorÚ
ValueErrorÚobjectÚgetattrÚtmZassert_series_equal)r   r   r   r
   ZaltÚresultÚexpectedr   r   r   Úcheck_accumulate   s    z$BaseAccumulateTests.check_accumulater
   TFc                 C   sl   |}t  |¡}|  ||¡r*|  |||¡ n>t ttf¡�  t||ƒ|d� W d   ƒ n1 s^0    Y  d S )Nr   )	ÚpdÚSeriesr	   r   ÚpytestZraisesÚNotImplementedErrorr   r   )r   ÚdataZall_numeric_accumulationsr
   r   r   r   r   r   Útest_accumulate_series   s    
z*BaseAccumulateTests.test_accumulate_seriesN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚstrÚboolr	   r   r   ÚmarkZparametrizer   r   r   r   r   r      s
   r   )r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   Ú<module>   s   