Skip to content

Commit 95f2fbe

Browse files
committed
Add test
1 parent ac3056f commit 95f2fbe

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

pandas/tests/arithmetic/test_interval.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,3 +271,21 @@ def test_index_series_compat(self, op, constructor, expected_type, assert_func):
271271
result = op(index, other)
272272
expected = expected_type(self.elementwise_comparison(op, index, other))
273273
assert_func(result, expected)
274+
275+
276+
@pytest.mark.parametrize("add", [True, False])
277+
def test_timestamp_interval_can_add_timedelta(add):
278+
# https://github.com/pandas-dev/pandas/issues/32023
279+
interval = pd.Interval(
280+
pd.Timestamp("2017-01-01 00:00:00"), pd.Timestamp("2018-01-01 00:00:00")
281+
)
282+
delta = pd.Timedelta(days=7)
283+
284+
if add:
285+
result = interval + delta
286+
expected = pd.Interval(interval.left + delta, interval.right + delta)
287+
else:
288+
result = interval - delta
289+
expected = pd.Interval(interval.left - delta, interval.right - delta)
290+
291+
assert result == expected

0 commit comments

Comments
 (0)