
    i
                     D    d dl mZmZmZmZmZmZ d dlmZ	  G d d      Z
y)    )	TimedeltaTimedeltaIndex	Timestamp
date_rangeoffsetstimedelta_rangeNc                   *    e Zd Zd Zd Zd Zd Zd Zy)TestDatetimeIndexArithmeticc                     d}t        t        d|      t        d|      d      }|t        d      z   }|j                  |j                  k(  sJ y )	NzCanada/Easternz2019-03-26 00:00:00-0400tzz2020-10-17 00:00:00-040012h)startendfreq   days)r   r   r   r   )selfr   dtiresults       k/app/cer_product_mecsu/.venv/lib/python3.12/site-packages/pandas/tests/indexes/datetimes/test_arithmetic.py!test_add_timedelta_preserves_freqz=TestDatetimeIndexArithmetic.test_add_timedelta_preserves_freq   sS     62>4<

 ya(({{chh&&&    c                     t        dd|t        j                  d            }||d   z
  }t        ddt        j                  d            }t	        j
                  ||       |j                  |j                  k(  sJ y )	Nz
2016-01-01      )periodsr   r   r   z0 Daysz11 Days)r   )r   r   Hourr   tmassert_index_equalr   )r   tz_naive_fixturer   resexpecteds        r    test_sub_datetime_preserves_freqz<TestDatetimeIndexArithmetic.test_sub_datetime_preserves_freq   so    
 b!	
 CFl"8YW\\"=MN
c8,xx8==(((r   c           	          t        dd      }t        |dd      }||d   z
  }t        t        d      t        d	      t        d
      t        d
d      gd      }t	        j
                  ||       |j                  |j                  k(  sJ y )Nz
2016-03-11z
US/Pacificr      ns)r   unitr   r   r         )r   hourszm8[ns])dtype)r   r   r   r   r    r!   r   )r   tsr   r#   r$   s        r   +test_sub_datetime_preserves_freq_across_dstzGTestDatetimeIndexArithmetic.test_sub_datetime_preserves_freq_across_dst.   s    |5QT2CFl!q!q!q!q+	 
 	c8,xx8==(((r   c                 |    t        dddd      }||j                  z   d d }|dd  }t        j                  ||       y )Nz
2020-03-28r'   DzEurope/Berlin)r   r   r   r   )r   r   r    r!   )r   r   r   r$   s       r   test_add_dti_dayz,TestDatetimeIndexArithmetic.test_add_dti_day@   sA    qsO.#2&qr7
fh/r   c                     t        ddd      }t        d      }||z
  }t        |t              sJ |j                  |j                  k(  sJ y )Nz
2021-01-01   r1   )r   r   z
2020-01-01)r   r   
isinstancer   r   )r   r   r.   r   s       r   %test_sub_timestamp_preserves_day_freqzATestDatetimeIndexArithmetic.test_sub_timestamp_preserves_day_freqG   sK    qs;|$r &.111{{chh&&&r   N)__name__
__module____qualname__r   r%   r/   r3   r7    r   r   r
   r
      s    
')")$0	'r   r
   )pandasr   r   r   r   r   r   pandas._testing_testingr    r
   r;   r   r   <module>r?      s    
  @' @'r   