2569718930@qq.com
94 lines
3.0 KiB
Python
94 lines
3.0 KiB
Python
import src.models.lgbm_daily_high as runtime
|
|
|
|
|
|
class _FakeBooster:
|
|
best_iteration = 7
|
|
|
|
def predict(self, rows, num_iteration=None):
|
|
assert len(rows) == 1
|
|
return [14.36]
|
|
|
|
|
|
def test_predict_lgbm_daily_high_skips_when_disabled(monkeypatch):
|
|
monkeypatch.setenv("POLYWEATHER_LGBM_ENABLED", "false")
|
|
prediction, meta = runtime.predict_lgbm_daily_high(
|
|
city_name="ankara",
|
|
current_forecasts={"Open-Meteo": 12.4},
|
|
deb_prediction=12.3,
|
|
current_temp=11.0,
|
|
max_so_far=11.4,
|
|
humidity=62.0,
|
|
wind_speed_kt=8.0,
|
|
visibility_mi=6.0,
|
|
local_hour=10,
|
|
local_date="2026-03-24",
|
|
peak_status="before",
|
|
history_data={},
|
|
)
|
|
assert prediction is None
|
|
assert meta["reason"] == "disabled"
|
|
|
|
|
|
def test_predict_lgbm_daily_high_returns_prediction(monkeypatch):
|
|
monkeypatch.setenv("POLYWEATHER_LGBM_ENABLED", "true")
|
|
monkeypatch.setenv("POLYWEATHER_LGBM_MIN_HISTORY_POINTS", "3")
|
|
monkeypatch.setattr(runtime, "_load_schema", lambda path: {"feature_names": runtime.FEATURE_NAMES})
|
|
monkeypatch.setattr(runtime, "_load_booster", lambda path: _FakeBooster())
|
|
|
|
history_data = {
|
|
"ankara": {
|
|
"2026-03-20": {"actual_high": 10.0},
|
|
"2026-03-21": {"actual_high": 11.0},
|
|
"2026-03-22": {"actual_high": 13.0},
|
|
"2026-03-23": {"actual_high": 12.0},
|
|
}
|
|
}
|
|
prediction, meta = runtime.predict_lgbm_daily_high(
|
|
city_name="ankara",
|
|
current_forecasts={"Open-Meteo": 12.4, "ECMWF": 12.1, "GFS": 11.9},
|
|
deb_prediction=12.3,
|
|
current_temp=11.0,
|
|
max_so_far=11.4,
|
|
humidity=62.0,
|
|
wind_speed_kt=8.0,
|
|
visibility_mi=6.0,
|
|
local_hour=10,
|
|
local_date="2026-03-24",
|
|
peak_status="before",
|
|
history_data=history_data,
|
|
)
|
|
assert prediction == 14.4
|
|
assert meta["reason"] == "ok"
|
|
assert meta["history_count"] == 4
|
|
|
|
|
|
def test_predict_lgbm_daily_high_requires_min_history(monkeypatch):
|
|
monkeypatch.setenv("POLYWEATHER_LGBM_ENABLED", "true")
|
|
monkeypatch.setenv("POLYWEATHER_LGBM_MIN_HISTORY_POINTS", "5")
|
|
monkeypatch.setattr(runtime, "_load_schema", lambda path: {"feature_names": runtime.FEATURE_NAMES})
|
|
monkeypatch.setattr(runtime, "_load_booster", lambda path: _FakeBooster())
|
|
|
|
history_data = {
|
|
"ankara": {
|
|
"2026-03-21": {"actual_high": 11.0},
|
|
"2026-03-22": {"actual_high": 13.0},
|
|
"2026-03-23": {"actual_high": 12.0},
|
|
}
|
|
}
|
|
prediction, meta = runtime.predict_lgbm_daily_high(
|
|
city_name="ankara",
|
|
current_forecasts={"Open-Meteo": 12.4},
|
|
deb_prediction=12.3,
|
|
current_temp=11.0,
|
|
max_so_far=11.4,
|
|
humidity=62.0,
|
|
wind_speed_kt=8.0,
|
|
visibility_mi=6.0,
|
|
local_hour=10,
|
|
local_date="2026-03-24",
|
|
peak_status="before",
|
|
history_data=history_data,
|
|
)
|
|
assert prediction is None
|
|
assert meta["reason"] == "insufficient_history"
|