def test_seasonality_time_index(time_index):
tt = Seasonality.from_index(time_index)
assert tt.period == 5
fcast = tt.out_of_sample(steps=12, index=time_index)
new_idx = DeterministicTerm._extend_index(time_index, 12)
pd.testing.assert_index_equal(fcast.index, new_idx)
def extend_index(steps: int, index: pd.Index) -> pd.Index:
return DeterministicTerm._extend_index(index, steps)
def test_index_like():
idx = np.empty((100, 2))
with pytest.raises(TypeError, match="index must be a pandas"):
DeterministicTerm._index_like(idx)