From 6786a4c4ed8b6c7afc8d46282a085667be5e722e Mon Sep 17 00:00:00 2001 From: Megawattz Date: Fri, 24 Jan 2025 12:06:10 +0100 Subject: [PATCH] update forecast test with no ts spelling --- tests/unit/test_forecast_no_ts.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/unit/test_forecast_no_ts.py b/tests/unit/test_forecast_no_ts.py index a66b7b1c..0df021b6 100644 --- a/tests/unit/test_forecast_no_ts.py +++ b/tests/unit/test_forecast_no_ts.py @@ -10,19 +10,19 @@ def test_run_forecast_no_ts(): current_ts = pd.Timestamp.now() # run gradient boosting model with no ts - predications_df = run_forecast(site=site, model="gb") + predictions_df = run_forecast(site=site, model="gb") # check current ts agrees with dataset - assert predications_df.index.min() >= current_ts - pd.Timedelta(hours=1) + assert predictions_df.index.min() >= current_ts - pd.Timedelta(hours=1) - print(predications_df) + print(predictions_df) print(f"Current time: {current_ts}") - print(f"Max: {predications_df['power_kw'].max()}") + print(f"Max: {predictions_df['power_kw'].max()}") # run xgb model with no ts - predications_df = run_forecast(site=site, model="xgb") + predictions_df = run_forecast(site=site, model="xgb") # check current ts agrees with dataset - assert predications_df.index.min() >= current_ts - pd.Timedelta(hours=1) + assert predictions_df.index.min() >= current_ts - pd.Timedelta(hours=1) - print(predications_df) + print(predictions_df) print(f"Current time: {current_ts}") - print(f"Max: {predications_df['power_kw'].max()}") \ No newline at end of file + print(f"Max: {predictions_df['power_kw'].max()}") \ No newline at end of file