def test_concat_same_type_invalid(self, datetime_index):
# different timezones
dti = datetime_index
arr = DatetimeArray(dti)
if arr.tz is None:
other = arr.tz_localize('UTC')
else:
other = arr.tz_localize(None)
with pytest.raises(AssertionError):
arr._concat_same_type([arr, other])
def test_concat_same_type_invalid(self, datetime_index):
# different timezones
dti = datetime_index
arr = DatetimeArray(dti)
if arr.tz is None:
other = arr.tz_localize('UTC')
else:
other = arr.tz_localize(None)
with pytest.raises(AssertionError):
arr._concat_same_type([arr, other])
def test_concat_same_type_different_freq(self):
# we *can* concatentate DTI with different freqs.
a = DatetimeArray(pd.date_range('2000', periods=2, freq='D',
tz='US/Central'))
b = DatetimeArray(pd.date_range('2000', periods=2, freq='H',
tz='US/Central'))
result = DatetimeArray._concat_same_type([a, b])
expected = DatetimeArray(pd.to_datetime([
'2000-01-01 00:00:00', '2000-01-02 00:00:00',
'2000-01-01 00:00:00', '2000-01-01 01:00:00',
]).tz_localize("US/Central"))
tm.assert_datetime_array_equal(result, expected)
def test_concat_same_type_different_freq(self):
# we *can* concatentate DTI with different freqs.
a = DatetimeArray(pd.date_range('2000', periods=2, freq='D',
tz='US/Central'))
b = DatetimeArray(pd.date_range('2000', periods=2, freq='H',
tz='US/Central'))
result = DatetimeArray._concat_same_type([a, b])
expected = DatetimeArray(pd.to_datetime([
'2000-01-01 00:00:00', '2000-01-02 00:00:00',
'2000-01-01 00:00:00', '2000-01-01 01:00:00',
]).tz_localize("US/Central"))
tm.assert_datetime_array_equal(result, expected)
