o
    d-                     @   s  d dl mZ d dlZd dlZd dlmZmZmZ ej	
dddgej	
deeded	eed
deddgej	
ded
ded
ed
dgdd Zej	
deddeddgej	
ded
ded
ed
dgdd Zej	
deejegdd ZdS )    )	timedeltaN)Interval	Timedelta	Timestampmethod__add____sub__intervalz2017-01-01 00:00:00z2018-01-01 00:00:00   )days   deltaDc                 C   sH   t | ||}t | j||}t | j||}t||}||ks"J d S )N)getattrleftrightr   )r	   r   r   resultr   r   expected r   `/app/.heroku/python/lib/python3.10/site-packages/pandas/tests/scalar/interval/test_arithmetic.py)test_time_interval_add_subtract_timedelta   s
   
r         g      ?g       @c                 C   s   d g d}tjttf|d | |  W d    n1 sw   Y  tjttf|d ||   W d    d S 1 s=w   Y  d S )N|)zunsupported operandzcannot use operandsz;Only numeric, Timestamp and Timedelta endpoints are allowed)match)joinpytestZraises	TypeError
ValueError)r	   r   msgr   r   r   *test_numeric_interval_add_timedelta_raises"   s   

"r    klassc                 C   sF   | d}t tdtd}|| }||ksJ || }||ks!J d S )Nr   z
2020-01-01z
2020-02-01)r   r   )r!   r   r   r   r   r   r   %test_timedelta_add_timestamp_interval6   s   r"   )datetimer   numpynpr   Zpandasr   r   r   markZparametrizeZtimedelta64r   r    r"   r   r   r   r   <module>   s,    
