def test_align(self):
dates = pd.date_range("2000-01-01", "2010-01-01", freq="M")
columns = ["h." + "{0:>02}".format(h + 1) for h in range(10)]
forecasts = pd.DataFrame(self.rng.randn(120, 10), index=dates, columns=columns)
aligned = _align_forecast(forecasts.copy(), align="origin")
assert_frame_equal(aligned, forecasts)
aligned = _align_forecast(forecasts.copy(), align="target")
direct = forecasts.copy()
for i in range(10):
direct.iloc[(i + 1) :, i] = direct.iloc[: (120 - i - 1), i].values
direct.iloc[: (i + 1), i] = np.nan
assert_frame_equal(aligned, direct)
with pytest.raises(ValueError):
_align_forecast(forecasts, align="unknown")
def test_align(self):
dates = pd.date_range('2000-01-01', '2010-01-01', freq='M')
columns = ['h.' + '{0:>02}'.format(str(h + 1)) for h in range(10)]
forecasts = pd.DataFrame(self.rng.randn(120, 10),
index=dates,
columns=columns)
aligned = _align_forecast(forecasts.copy(), align='origin')
assert_frame_equal(aligned, forecasts)
aligned = _align_forecast(forecasts.copy(), align='target')
direct = forecasts.copy()
for i in range(10):
direct.iloc[(i + 1):, i] = direct.iloc[:(120 - i - 1), i].values
direct.iloc[:(i + 1), i] = np.nan
assert_frame_equal(aligned, direct)
with pytest.raises(ValueError):
_align_forecast(forecasts, align='unknown')
def test_align(self):
dates = pd.date_range('2000-01-01', '2010-01-01', freq='M')
columns = ['h.' + '{0:>02}'.format(str(h + 1)) for h in range(10)]
forecasts = pd.DataFrame(np.random.randn(120, 10),
index=dates,
columns=columns)
aligned = _align_forecast(forecasts.copy(), align='origin')
assert_frame_equal(aligned, forecasts)
aligned = _align_forecast(forecasts.copy(), align='target')
direct = forecasts.copy()
for i in range(10):
direct.iloc[(i + 1):, i] = direct.iloc[:(120 - i - 1), i].values
direct.iloc[:(i + 1), i] = np.nan
assert_frame_equal(aligned, direct)
with pytest.raises(ValueError):
_align_forecast(forecasts, align='unknown')
