2569718930@qq.com
139 lines
4.9 KiB
Python
139 lines
4.9 KiB
Python
|
|
import time
|
|
|
|
from src.database.runtime_state import (
|
|
DailyRecordRepository,
|
|
OpenMeteoCacheRepository,
|
|
ProbabilitySnapshotRepository,
|
|
RuntimeStateDB,
|
|
TelegramAlertStateRepository,
|
|
TrainingFeatureRecordRepository,
|
|
TruthRecordRepository,
|
|
TruthRevisionRepository,
|
|
)
|
|
|
|
|
|
def test_daily_record_repository_roundtrip(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
repo = DailyRecordRepository(RuntimeStateDB(str(tmp_path / 'polyweather.db')))
|
|
repo.upsert_record('ankara', '2026-03-20', {'actual_high': 15.2, 'deb_prediction': 14.8, 'mu': 15.0})
|
|
data = repo.load_all()
|
|
assert data['ankara']['2026-03-20']['actual_high'] == 15.2
|
|
|
|
|
|
def test_telegram_alert_state_repository_roundtrip(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
repo = TelegramAlertStateRepository(RuntimeStateDB(str(tmp_path / 'polyweather.db')))
|
|
state = {
|
|
'last_by_city': {
|
|
'ankara': {
|
|
'signature': 'sig-1',
|
|
'trigger_key': 'mkt:test',
|
|
'severity': 'medium',
|
|
'ts': 123,
|
|
'active': True,
|
|
'evidence': {'x': 1},
|
|
}
|
|
},
|
|
'by_signature': {'sig-1': 123},
|
|
}
|
|
repo.save_state(state)
|
|
loaded = repo.load_state()
|
|
assert loaded == state
|
|
|
|
|
|
def test_probability_snapshot_repository_recent_rows(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
repo = ProbabilitySnapshotRepository(RuntimeStateDB(str(tmp_path / 'polyweather.db')))
|
|
repo.append_snapshot({
|
|
'city': 'ankara',
|
|
'date': '2026-03-20',
|
|
'timestamp': '2026-03-20T12:00:00Z',
|
|
'raw_mu': 15.2,
|
|
'raw_sigma': 1.1,
|
|
'max_so_far': 14.9,
|
|
'peak_status': 'before',
|
|
'probability_mode': 'emos_shadow',
|
|
'prob_snapshot': [{'v': 15, 'p': 0.6}],
|
|
'shadow_prob_snapshot': [{'v': 15, 'p': 0.4}],
|
|
})
|
|
rows = repo.load_recent_rows('ankara', '2026-03-20', 5)
|
|
assert len(rows) == 1
|
|
assert rows[0]['raw_mu'] == 15.2
|
|
|
|
|
|
def test_open_meteo_cache_repository_roundtrip(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
repo = OpenMeteoCacheRepository(RuntimeStateDB(str(tmp_path / 'polyweather.db')))
|
|
payload = {
|
|
'forecast': {'ankara': {'t': time.time(), 'temp': 15}},
|
|
'ensemble': {'ankara': {'t': time.time(), 'spread': 1.5}},
|
|
'multi_model': {},
|
|
'saved_at': 1000,
|
|
}
|
|
repo.replace_payload(payload, 86400)
|
|
loaded = repo.load_payload(86400)
|
|
assert loaded['forecast']['ankara']['temp'] == 15
|
|
assert loaded['ensemble']['ankara']['spread'] == 1.5
|
|
|
|
|
|
def test_truth_record_repository_tracks_revisions(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
db = RuntimeStateDB(str(tmp_path / 'polyweather.db'))
|
|
truth_repo = TruthRecordRepository(db)
|
|
revision_repo = TruthRevisionRepository(db)
|
|
|
|
changed = truth_repo.upsert_truth(
|
|
city='taipei',
|
|
target_date='2026-04-01',
|
|
actual_high=18.0,
|
|
settlement_source='wunderground',
|
|
settlement_station_code='RCSS',
|
|
settlement_station_label='Taipei Songshan Airport Station',
|
|
truth_version='v1',
|
|
updated_by='test:first',
|
|
source_payload={'raw_max_temp_c': 18.4},
|
|
reason='initial',
|
|
)
|
|
assert changed is True
|
|
assert revision_repo.load_revisions('taipei', '2026-04-01') == []
|
|
|
|
changed = truth_repo.upsert_truth(
|
|
city='taipei',
|
|
target_date='2026-04-01',
|
|
actual_high=19.0,
|
|
settlement_source='wunderground',
|
|
settlement_station_code='RCSS',
|
|
settlement_station_label='Taipei Songshan Airport Station',
|
|
truth_version='v1',
|
|
updated_by='test:second',
|
|
source_payload={'raw_max_temp_c': 19.1},
|
|
reason='correction',
|
|
)
|
|
assert changed is True
|
|
loaded = truth_repo.get_record('taipei', '2026-04-01')
|
|
assert loaded['actual_high'] == 19.0
|
|
revisions = revision_repo.load_revisions('taipei', '2026-04-01')
|
|
assert len(revisions) == 1
|
|
assert revisions[0]['previous_actual_high'] == 18.0
|
|
assert revisions[0]['next_actual_high'] == 19.0
|
|
|
|
|
|
def test_training_feature_record_repository_roundtrip(tmp_path, monkeypatch):
|
|
monkeypatch.setenv('POLYWEATHER_DB_PATH', str(tmp_path / 'polyweather.db'))
|
|
db = RuntimeStateDB(str(tmp_path / 'polyweather.db'))
|
|
repo = TrainingFeatureRecordRepository(db)
|
|
repo.upsert_record(
|
|
'ankara',
|
|
'2026-03-20',
|
|
{
|
|
'forecasts': {'ECMWF': 12.3},
|
|
'deb_prediction': 12.1,
|
|
'mu': 12.0,
|
|
'probability_features': {'ens_median': 12.2},
|
|
},
|
|
)
|
|
loaded = repo.get_record('ankara', '2026-03-20')
|
|
assert loaded['forecasts']['ECMWF'] == 12.3
|
|
assert loaded['deb_prediction'] == 12.1
|