separate DatetimeIndex timezone tests (pandas-dev#19545)

Author: jbrockmendel

pandas/tests/indexes/datetimes/test_arithmetic.py

@@ -412,6 +412,14 @@ def test_dti_shift_no_freq(self):
  with pytest.raises(NullFrequencyError):
  dti.shift(2)
 
+ @pytest.mark.parametrize('tzstr', ['US/Eastern', 'dateutil/US/Eastern'])
+ def test_dti_shift_localized(self, tzstr):
+ dr = date_range('2011/1/1', '2012/1/1', freq='W-FRI')
+ dr_tz = dr.tz_localize(tzstr)
+
+ result = dr_tz.shift(1, '10T')
+ assert result.tz == dr_tz.tz
+
  # -------------------------------------------------------------
  # Binary operations DatetimeIndex and timedelta-like
 
@@ -767,6 +775,24 @@ def test_dti_with_offset_series(self, tz, names):
  res3 = dti - other
  tm.assert_series_equal(res3, expected_sub)
 
+ def test_dti_add_offset_tzaware(self):
+ dates = date_range('2012-11-01', periods=3, tz='US/Pacific')
+ offset = dates + pd.offsets.Hour(5)
+ assert dates[0] + pd.offsets.Hour(5) == offset[0]
+
+ # GH#6818
+ for tz in ['UTC', 'US/Pacific', 'Asia/Tokyo']:
+ dates = date_range('2010-11-01 00:00', periods=3, tz=tz, freq='H')
+ expected = DatetimeIndex(['2010-11-01 05:00', '2010-11-01 06:00',
+ '2010-11-01 07:00'], freq='H', tz=tz)
+
+ offset = dates + pd.offsets.Hour(5)
+ tm.assert_index_equal(offset, expected)
+ offset = dates + np.timedelta64(5, 'h')
+ tm.assert_index_equal(offset, expected)
+ offset = dates + timedelta(hours=5)
+ tm.assert_index_equal(offset, expected)
+
 
 @pytest.mark.parametrize('klass,assert_func', [
  (Series, tm.assert_series_equal),