2569718930@qq.com
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from scripts.fit_probability_calibration import _extract_samples
|
|
|
|
|
|
def test_extract_samples_prefers_snapshot_rows_for_same_city_day():
|
|
history = {
|
|
"ankara": {
|
|
"2026-03-19": {
|
|
"actual_high": 11.0,
|
|
"mu": 10.8,
|
|
"deb_prediction": 10.9,
|
|
"forecasts": {"ECMWF": 10.5, "GFS": 11.2},
|
|
"probability_features": {
|
|
"ens_median": 10.7,
|
|
"ensemble_spread": 0.8,
|
|
"peak_status": "before",
|
|
},
|
|
}
|
|
}
|
|
}
|
|
snapshot_rows = [
|
|
{
|
|
"city": "ankara",
|
|
"date": "2026-03-19",
|
|
"timestamp": "2026-03-19T12:00:00+03:00",
|
|
"raw_mu": 11.2,
|
|
"raw_sigma": 1.1,
|
|
"deb_prediction": 11.0,
|
|
"ensemble": {"p10": 10.0, "median": 11.1, "p90": 12.2},
|
|
"multi_model": {"ECMWF": 10.5, "GFS": 11.2},
|
|
"max_so_far": 10.9,
|
|
"peak_status": "in_window",
|
|
}
|
|
]
|
|
truth_history = {
|
|
"ankara": {
|
|
"2026-03-19": {
|
|
"actual_high": 11.0,
|
|
"settlement_source": "metar",
|
|
"settlement_station_code": "LTAC",
|
|
"truth_version": "v1",
|
|
"updated_by": "test",
|
|
"truth_updated_at": 123.0,
|
|
}
|
|
}
|
|
}
|
|
|
|
samples, filled = _extract_samples(
|
|
history,
|
|
truth_history=truth_history,
|
|
settlement_history={},
|
|
snapshot_rows=snapshot_rows,
|
|
)
|
|
|
|
assert filled == 0
|
|
assert len(samples) == 1
|
|
assert samples[0]["sample_source"] == "snapshot"
|
|
assert samples[0]["raw_mu"] == 11.2
|
|
assert samples[0]["peak_flag"] == 0.5
|
|
assert samples[0]["settlement_source"] == "metar"
|
|
assert samples[0]["settlement_station_code"] == "LTAC"
|
|
assert samples[0]["truth_version"] == "v1"
|