o
    թZh                     @   s:   d dl Zd dlmZmZmZ d dlmZ G dd dZ	dS )    N)NaT	Timedeltatimedelta_rangec                   @   s   e Zd Zdd Zdd ZdS )TestTimedeltaIndexArithmeticc                 C   s   t dddd}|d }|jd u sJ |d d d}t|| |d }|jd u s,J |}t|| |d }|jd u s?J |d d d}t|| d S )Nr   d   ns)Zperiodsfreq   2      )r   r   repeattmZassert_index_equal)selfZtdiresultexpectedZresult2Z	expected2Zresult3Z	expected3 r   f/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/indexes/timedeltas/test_arithmetic.pytest_arithmetic_zero_freq   s   z6TestTimedeltaIndexArithmetic.test_arithmetic_zero_freqc                 C   s   t dd}||||t ddd tgdd}|tdd	 }|ddd
tjg}t|| |tdd }|dddtjg}t|| d S )N   )days      )minutessecondszm8[ns])Zdtyper   Dg ?@si( i( )r   r   npZtimedelta64nanr   Zassert_equal)r   Zindex_or_seriesZscalartdr   r   r   r   r   test_tdi_division    s   
z.TestTimedeltaIndexArithmetic.test_tdi_divisionN)__name__
__module____qualname__r   r   r   r   r   r   r      s    r   )
numpyr   Zpandasr   r   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   