2569718930@qq.com
47 lines
2.0 KiB
Python
47 lines
2.0 KiB
Python
from src.data_collection.nws_open_meteo_sources import (
|
|
OPEN_METEO_MULTI_MODEL_ORDER,
|
|
_parse_open_meteo_multi_model_daily,
|
|
)
|
|
|
|
|
|
def test_multi_model_parser_exposes_open_recommended_models():
|
|
daily = {
|
|
"time": ["2026-04-17", "2026-04-18"],
|
|
"temperature_2m_max_ecmwf_ifs025": [20.1, 21.1],
|
|
"temperature_2m_max_ecmwf_aifs025_single": [20.2, 21.2],
|
|
"temperature_2m_max_icon_eu": [20.3, 21.3],
|
|
"temperature_2m_max_icon_d2": [20.4, None],
|
|
"temperature_2m_max_gem_global": [19.8, 20.8],
|
|
"temperature_2m_max_gem_regional": [21.0, 22.0],
|
|
"temperature_2m_max_gem_hrdps_continental": [21.5, None],
|
|
}
|
|
|
|
dates, forecasts, metadata, model_keys = _parse_open_meteo_multi_model_daily(daily)
|
|
|
|
assert dates == ["2026-04-17", "2026-04-18"]
|
|
assert forecasts["2026-04-17"]["ECMWF"] == 20.1
|
|
assert forecasts["2026-04-17"]["ECMWF AIFS"] == 20.2
|
|
assert forecasts["2026-04-17"]["ICON-EU"] == 20.3
|
|
assert forecasts["2026-04-17"]["ICON-D2"] == 20.4
|
|
assert forecasts["2026-04-17"]["GDPS"] == 19.8
|
|
assert forecasts["2026-04-17"]["RDPS"] == 21.0
|
|
assert forecasts["2026-04-17"]["HRDPS"] == 21.5
|
|
assert "ICON-D2" not in forecasts["2026-04-18"]
|
|
assert metadata["ECMWF AIFS"]["provider"] == "ECMWF"
|
|
assert metadata["HRDPS"]["resolution_km"] == 2.5
|
|
assert model_keys["RDPS"] == "gem_regional"
|
|
|
|
|
|
def test_multi_model_order_includes_legacy_and_new_sources():
|
|
assert "ecmwf_ifs025" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "ecmwf_aifs025_single" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "gfs_seamless" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "icon_seamless" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "icon_eu" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "icon_d2" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "gem_seamless" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "gem_global" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "gem_regional" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "gem_hrdps_continental" in OPEN_METEO_MULTI_MODEL_ORDER
|
|
assert "jma_seamless" in OPEN_METEO_MULTI_MODEL_ORDER
|