diff --git a/tests/test_timefuncs.py b/tests/test_timefuncs.py index 208e6c60..1d7a11bf 100644 --- a/tests/test_timefuncs.py +++ b/tests/test_timefuncs.py @@ -753,7 +753,8 @@ def test_some_offsets() -> None: def test_timestampseries_offset() -> None: """Test that adding an offset to a timestamp series works.""" vv = pd.bdate_range("2024-09-01", "2024-09-10") - vv + pd.tseries.offsets.YearEnd(0) + shifted_vv = vv + pd.tseries.offsets.YearEnd(0) + check(assert_type(shifted_vv, pd.DatetimeIndex), pd.DatetimeIndex) def test_types_to_numpy() -> None: