2569718930@qq.com
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
from src.analysis.deb_hourly_correction import (
|
|
DEB_HOURLY_PEAK_CORRECTED_VERSION,
|
|
build_deb_hourly_path,
|
|
build_hourly_peak_corrector,
|
|
)
|
|
|
|
|
|
def test_hourly_peak_corrector_learns_peak_window_shape_from_snapshots():
|
|
snapshots = [
|
|
{
|
|
"city": "ankara",
|
|
"target_date": "2026-05-20",
|
|
"deb_prediction": 30.0,
|
|
"deb_base_path": {
|
|
"times": ["10:00", "12:00", "14:00", "16:00"],
|
|
"temps": [25.0, 27.0, 30.0, 28.0],
|
|
},
|
|
"settlement_today_obs": [
|
|
{"time": "12:00", "temp": 27.5},
|
|
{"time": "14:00", "temp": 32.0},
|
|
{"time": "16:00", "temp": 28.5},
|
|
],
|
|
"peak": {"first_h": 13, "last_h": 15},
|
|
},
|
|
{
|
|
"city": "ankara",
|
|
"target_date": "2026-05-21",
|
|
"deb_prediction": 29.0,
|
|
"deb_base_path": {
|
|
"times": ["10:00", "12:00", "14:00", "16:00"],
|
|
"temps": [24.0, 26.0, 29.0, 27.0],
|
|
},
|
|
"settlement_today_obs": [
|
|
{"time": "12:00", "temp": 26.5},
|
|
{"time": "14:00", "temp": 31.0},
|
|
{"time": "16:00", "temp": 27.5},
|
|
],
|
|
"peak": {"first_h": 13, "last_h": 15},
|
|
},
|
|
]
|
|
|
|
corrector = build_hourly_peak_corrector(snapshots, min_samples=2)
|
|
result = corrector.apply(
|
|
"ankara",
|
|
["10:00", "12:00", "14:00", "16:00"],
|
|
[24.0, 26.0, 29.0, 27.0],
|
|
peak_first_h=13,
|
|
peak_last_h=15,
|
|
deb_prediction=30.0,
|
|
)
|
|
|
|
assert result["version"] == DEB_HOURLY_PEAK_CORRECTED_VERSION
|
|
assert result["samples"] == 6
|
|
assert result["temps"][2] == 30.0
|
|
assert result["temps"][1] < result["temps"][2]
|
|
assert result["temps"][3] < result["temps"][2]
|
|
assert result["phase_adjustments"]["peak_window"]["samples"] == 2
|
|
|
|
|
|
def test_build_deb_hourly_path_anchors_corrected_curve_to_deb_prediction():
|
|
corrector = build_hourly_peak_corrector(
|
|
[
|
|
{
|
|
"city": "shanghai",
|
|
"target_date": "2026-05-20",
|
|
"deb_base_path": {
|
|
"times": ["10:00", "14:00", "18:00"],
|
|
"temps": [25.0, 28.0, 24.0],
|
|
},
|
|
"settlement_today_obs": [
|
|
{"time": "10:00", "temp": 24.0},
|
|
{"time": "14:00", "temp": 30.0},
|
|
{"time": "18:00", "temp": 23.0},
|
|
],
|
|
"peak": {"first_h": 13, "last_h": 15},
|
|
},
|
|
{
|
|
"city": "shanghai",
|
|
"target_date": "2026-05-21",
|
|
"deb_base_path": {
|
|
"times": ["10:00", "14:00", "18:00"],
|
|
"temps": [26.0, 29.0, 25.0],
|
|
},
|
|
"settlement_today_obs": [
|
|
{"time": "10:00", "temp": 25.0},
|
|
{"time": "14:00", "temp": 31.0},
|
|
{"time": "18:00", "temp": 24.0},
|
|
],
|
|
"peak": {"first_h": 13, "last_h": 15},
|
|
},
|
|
],
|
|
min_samples=2,
|
|
)
|
|
|
|
path = build_deb_hourly_path(
|
|
city="shanghai",
|
|
hourly_times=["10:00", "14:00", "18:00"],
|
|
hourly_temps=[24.0, 28.0, 23.0],
|
|
deb_prediction=29.0,
|
|
peak_first_h=13,
|
|
peak_last_h=15,
|
|
corrector=corrector,
|
|
)
|
|
|
|
assert path["source"] == DEB_HOURLY_PEAK_CORRECTED_VERSION
|
|
assert path["base_source"] == "hourly_plus_deb_offset"
|
|
assert max(path["temps"]) == 29.0
|
|
assert path["temps"][1] == 29.0
|
|
assert path["temps"][0] < 25.5
|
|
assert path["correction"]["samples"] == 6
|
|
|
|
|
|
def test_build_deb_hourly_path_preserves_consensus_base_source():
|
|
corrector = build_hourly_peak_corrector([], min_samples=2)
|
|
|
|
path = build_deb_hourly_path(
|
|
city="wuhan",
|
|
hourly_times=["09:00", "15:00"],
|
|
hourly_temps=[24.0, 30.0],
|
|
deb_prediction=30.0,
|
|
peak_first_h=15,
|
|
peak_last_h=15,
|
|
corrector=corrector,
|
|
base_source="deb_hourly_consensus",
|
|
)
|
|
|
|
assert path["base_source"] == "deb_hourly_consensus"
|
|
assert path["temps"] == [24.0, 30.0]
|