2569718930@qq.com
261 lines
8.0 KiB
Python
261 lines
8.0 KiB
Python
from src.analysis.market_alert_engine import build_trading_alerts
|
|
from src.utils.telegram_push import (
|
|
_build_focus_digest_message,
|
|
_shortlist_focus_payloads,
|
|
build_market_monitor_digest,
|
|
)
|
|
|
|
|
|
def _sample_weather_payload():
|
|
return {
|
|
"name": "ankara",
|
|
"display_name": "Ankara",
|
|
"lat": 40.1281,
|
|
"lon": 32.9951,
|
|
"temp_symbol": "°C",
|
|
"current": {
|
|
"temp": 11.3,
|
|
"wind_dir": 180.0,
|
|
"wind_speed_kt": 11.0,
|
|
},
|
|
"trend": {
|
|
"recent": [
|
|
{"time": "10:30", "temp": 11.3},
|
|
{"time": "10:00", "temp": 10.3},
|
|
{"time": "09:30", "temp": 9.9},
|
|
]
|
|
},
|
|
"multi_model": {
|
|
"MGM": 10.8,
|
|
"GFS": 10.4,
|
|
"ECMWF": 10.6,
|
|
},
|
|
"deb": {
|
|
"prediction": 11.8,
|
|
},
|
|
"metar_recent_obs": [
|
|
{"time": "10:30", "wdir": 180},
|
|
{"time": "10:00", "wdir": 60},
|
|
],
|
|
"mgm_nearby": [
|
|
{
|
|
"name": "Airport (MGM/17128)",
|
|
"istNo": "17128",
|
|
"lat": 39.95,
|
|
"lon": 32.97,
|
|
"temp": 12.4,
|
|
},
|
|
{
|
|
"name": "Ankara (Bölge/Center)",
|
|
"istNo": "17130",
|
|
"lat": 40.1281,
|
|
"lon": 32.9951,
|
|
"temp": 11.2,
|
|
},
|
|
],
|
|
}
|
|
|
|
|
|
def test_trading_alerts_all_core_rules_trigger():
|
|
out = build_trading_alerts(
|
|
city_weather=_sample_weather_payload(),
|
|
map_url="https://example.com/map",
|
|
)
|
|
|
|
assert out["trigger_count"] >= 2
|
|
assert out["rules"]["momentum_spike"]["triggered"] is True
|
|
assert out["rules"]["forecast_breakthrough"]["triggered"] is True
|
|
assert out["rules"]["advection"]["triggered"] is True
|
|
|
|
msg = out["telegram"]["zh"]
|
|
assert "PolyWeather 市场提醒" in msg
|
|
assert "动量突变" in msg
|
|
assert "https://example.com/map" in msg
|
|
|
|
|
|
def test_forecast_breakthrough_not_triggered_when_current_not_above_margin():
|
|
city_weather = _sample_weather_payload()
|
|
city_weather["current"]["temp"] = 11.0
|
|
|
|
out = build_trading_alerts(
|
|
city_weather=city_weather,
|
|
)
|
|
assert out["rules"]["forecast_breakthrough"]["triggered"] is False
|
|
|
|
|
|
def test_ankara_center_hits_deb_triggers_force_push():
|
|
city_weather = _sample_weather_payload()
|
|
city_weather["current"]["temp"] = 10.7
|
|
city_weather["deb"]["prediction"] = 11.2
|
|
city_weather["trend"]["recent"] = [
|
|
{"time": "10:30", "temp": 10.7},
|
|
{"time": "10:00", "temp": 10.7},
|
|
{"time": "09:30", "temp": 10.6},
|
|
]
|
|
city_weather["multi_model"] = {"MGM": 11.2, "GFS": 11.2, "ECMWF": 11.2}
|
|
|
|
out = build_trading_alerts(
|
|
city_weather=city_weather,
|
|
)
|
|
|
|
center_rule = out["rules"]["ankara_center_deb_hit"]
|
|
assert center_rule["triggered"] is True
|
|
assert center_rule["force_push"] is True
|
|
assert out["severity"] in ("medium", "high")
|
|
assert "Center信号" in out["telegram"]["zh"]
|
|
|
|
|
|
def test_ankara_center_signal_only_uses_official_center_station():
|
|
city_weather = _sample_weather_payload()
|
|
city_weather["deb"]["prediction"] = 11.2
|
|
city_weather["current"]["temp"] = 10.7
|
|
city_weather["mgm_nearby"] = [
|
|
{
|
|
"name": "Etimesgut",
|
|
"istNo": "17069",
|
|
"lat": 39.95,
|
|
"lon": 32.68,
|
|
"temp": 12.6,
|
|
},
|
|
{
|
|
"name": "Airport (MGM/17128)",
|
|
"istNo": "17128",
|
|
"lat": 39.95,
|
|
"lon": 32.97,
|
|
"temp": 11.3,
|
|
},
|
|
]
|
|
|
|
out = build_trading_alerts(city_weather=city_weather)
|
|
|
|
center_rule = out["rules"]["ankara_center_deb_hit"]
|
|
assert center_rule["triggered"] is True
|
|
assert center_rule["center_station"]["istNo"] == "17128"
|
|
assert center_rule["center_station"]["name"] == "Airport (MGM/17128)"
|
|
assert "Airport (MGM/17128)" in out["telegram"]["zh"]
|
|
assert "Etimesgut" not in out["telegram"]["zh"]
|
|
|
|
|
|
def test_peak_passed_guard_suppresses_late_day_cooldown_alerts():
|
|
city_weather = {
|
|
"name": "wellington",
|
|
"display_name": "Wellington",
|
|
"temp_symbol": "°C",
|
|
"local_time": "16:40",
|
|
"current": {
|
|
"temp": 19.0,
|
|
"max_so_far": 20.2,
|
|
"max_temp_time": "15:20",
|
|
"wind_dir": 220.0,
|
|
"wind_speed_kt": 8.0,
|
|
},
|
|
"trend": {
|
|
"recent": [
|
|
{"time": "16:40", "temp": 19.0},
|
|
{"time": "16:10", "temp": 20.0},
|
|
{"time": "15:40", "temp": 20.5},
|
|
]
|
|
},
|
|
"multi_model": {
|
|
"MGM": 18.2,
|
|
"GFS": 18.4,
|
|
"ECMWF": 18.5,
|
|
},
|
|
"deb": {"prediction": 18.7},
|
|
"metar_recent_obs": [
|
|
{"time": "16:40", "wdir": 220},
|
|
{"time": "16:10", "wdir": 210},
|
|
],
|
|
"mgm_nearby": [],
|
|
}
|
|
|
|
out = build_trading_alerts(city_weather=city_weather)
|
|
|
|
assert out["suppression"]["suppressed"] is True
|
|
assert out["severity"] == "none"
|
|
assert out["trigger_count"] == 0
|
|
assert out["rules"]["momentum_spike"]["raw_triggered"] is True
|
|
assert out["rules"]["forecast_breakthrough"]["raw_triggered"] is True
|
|
assert "高温已过（暂停推送）" in out["telegram"]["zh"]
|
|
assert "暂停主动推送" in out["telegram"]["zh"]
|
|
|
|
|
|
def test_market_monitor_digest_skips_non_tradable_market(monkeypatch):
|
|
payload = build_trading_alerts(
|
|
city_weather=_sample_weather_payload(),
|
|
map_url="https://example.com/map",
|
|
)
|
|
payload["market_snapshot"]["available"] = True
|
|
payload["market_snapshot"]["market_closed"] = True
|
|
payload["market_snapshot"]["market_tradable"] = False
|
|
payload["market_snapshot"]["market_accepting_orders"] = False
|
|
payload["market_snapshot"]["market_tradable_reason"] = "in_review"
|
|
|
|
monkeypatch.setattr(
|
|
"src.utils.telegram_push.build_trade_alert_for_city",
|
|
lambda city, config, force_refresh=False: payload,
|
|
)
|
|
monkeypatch.setenv("TELEGRAM_ALERT_CITIES", "ankara")
|
|
|
|
digest = build_market_monitor_digest({}, slot_label="当前市场概览")
|
|
assert digest == "ℹ️ 当前没有可用的市场监控摘要。"
|
|
|
|
|
|
def _focus_payload(*, local_time: str = "14:00", peak_time: str = "12:00"):
|
|
return {
|
|
"city": "madrid",
|
|
"severity": "medium",
|
|
"trigger_count": 1,
|
|
"rules": {},
|
|
"market_snapshot": {
|
|
"available": True,
|
|
"forecast_bucket": {
|
|
"label": "30°C",
|
|
"value": 30,
|
|
"yes_buy": 0.18,
|
|
"market_url": "https://example.com/market",
|
|
},
|
|
"market_url": "https://example.com/market",
|
|
"confidence": "medium",
|
|
"edge_percent": 5.0,
|
|
},
|
|
"evidence": {
|
|
"generated_local_time": local_time,
|
|
"inputs": {
|
|
"current_temp": 29.0,
|
|
"deb_prediction": 30.2,
|
|
},
|
|
"trigger_summary": {
|
|
"suppression_snapshot": {
|
|
"max_temp_time": peak_time,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
def test_focus_digest_push_window_does_not_cut_off_after_2pm_spanish_time(monkeypatch):
|
|
monkeypatch.delenv("TELEGRAM_MARKET_FOCUS_PUSH_AFTER_PEAK_MIN", raising=False)
|
|
|
|
shortlisted = _shortlist_focus_payloads(
|
|
[_focus_payload(local_time="14:00", peak_time="12:00")],
|
|
top_n=5,
|
|
for_push=True,
|
|
)
|
|
|
|
assert len(shortlisted) == 1
|
|
|
|
|
|
def test_focus_digest_footer_explains_no_daily_signal_cap(monkeypatch):
|
|
monkeypatch.setenv("TELEGRAM_ALERT_PUSH_INTERVAL_SEC", "300")
|
|
monkeypatch.setenv("TELEGRAM_MARKET_FOCUS_DIGEST_INTERVAL_SEC", "1800")
|
|
|
|
message = _build_focus_digest_message(
|
|
[_focus_payload(local_time="14:00", peak_time="12:00")],
|
|
slot_label="白天关注",
|
|
top_n=5,
|
|
)
|
|
|
|
assert "没有每日信号次数上限" in message
|
|
assert "观察窗口" in message
|