Skip to content

Commit 6786a4c

Browse files
committed
update forecast test with no ts spelling
1 parent df662f3 commit 6786a4c

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/unit/test_forecast_no_ts.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,19 +10,19 @@ def test_run_forecast_no_ts():
1010
current_ts = pd.Timestamp.now()
1111

1212
# run gradient boosting model with no ts
13-
predications_df = run_forecast(site=site, model="gb")
13+
predictions_df = run_forecast(site=site, model="gb")
1414
# check current ts agrees with dataset
15-
assert predications_df.index.min() >= current_ts - pd.Timedelta(hours=1)
15+
assert predictions_df.index.min() >= current_ts - pd.Timedelta(hours=1)
1616

17-
print(predications_df)
17+
print(predictions_df)
1818
print(f"Current time: {current_ts}")
19-
print(f"Max: {predications_df['power_kw'].max()}")
19+
print(f"Max: {predictions_df['power_kw'].max()}")
2020

2121
# run xgb model with no ts
22-
predications_df = run_forecast(site=site, model="xgb")
22+
predictions_df = run_forecast(site=site, model="xgb")
2323
# check current ts agrees with dataset
24-
assert predications_df.index.min() >= current_ts - pd.Timedelta(hours=1)
24+
assert predictions_df.index.min() >= current_ts - pd.Timedelta(hours=1)
2525

26-
print(predications_df)
26+
print(predictions_df)
2727
print(f"Current time: {current_ts}")
28-
print(f"Max: {predications_df['power_kw'].max()}")
28+
print(f"Max: {predictions_df['power_kw'].max()}")

0 commit comments

Comments
 (0)