From f4a37e4bdd6f99562e5eaaed002a6fd7850fbe3e Mon Sep 17 00:00:00 2001 From: "2569718930@qq.com" <2569718930@qq.com> Date: Thu, 14 May 2026 18:42:41 +0800 Subject: [PATCH] =?UTF-8?q?=E7=A7=BB=E9=99=A4=20market=5Falert=5Fengine=20?= =?UTF-8?q?=E5=AF=B9=E5=BA=94=E7=9A=84=E6=B5=8B=E8=AF=95=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 该测试文件引用已删除的 src.analysis.market_alert_engine 模块。 --- tests/test_market_alert_engine.py | 291 ------------------------------ 1 file changed, 291 deletions(-) delete mode 100644 tests/test_market_alert_engine.py diff --git a/tests/test_market_alert_engine.py b/tests/test_market_alert_engine.py deleted file mode 100644 index bb56988b..00000000 --- a/tests/test_market_alert_engine.py +++ /dev/null @@ -1,291 +0,0 @@ -from src.analysis.market_alert_engine import build_trading_alerts -from src.utils.telegram_push import _run_market_monitor_cycle - - -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"]["forecast_breakthrough"]["triggered"] is True - assert out["rules"]["advection"]["triggered"] is True - - msg = out["telegram"]["zh"] - assert "PolyWeather 市场提醒" 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"]["forecast_breakthrough"]["raw_triggered"] is True - assert "高温已过（暂停推送）" in out["telegram"]["zh"] - assert "暂停主动推送" in out["telegram"]["zh"] - - -class _DigestBot: - def __init__(self): - self.messages = [] - - def send_message(self, chat_id, text, **kwargs): - self.messages.append( - { - "chat_id": chat_id, - "text": text, - "kwargs": kwargs, - } - ) - - -def _focus_payload( - *, - city: str = "madrid", - local_time: str = "14:00", - peak_time: str = "12:00", - severity: str = "medium", - trigger_count: int = 1, - edge_percent: float = 5.0, - signal_label: str = "MONITOR", - yes_buy: float = 0.18, -): - return { - "city": city, - "severity": severity, - "trigger_count": trigger_count, - "rules": { - "momentum_spike": { - "triggered": trigger_count > 0, - }, - }, - "market_snapshot": { - "available": True, - "forecast_bucket": { - "label": "30°C", - "value": 30, - "yes_buy": yes_buy, - "yes_sell": min(0.99, yes_buy + 0.02), - "market_url": "https://example.com/market", - }, - "market_url": "https://example.com/market", - "confidence": "medium", - "edge_percent": edge_percent, - "signal_label": signal_label, - "market_active": True, - "market_closed": False, - "market_accepting_orders": True, - "market_tradable": True, - }, - "evidence": { - "generated_local_time": local_time, - "inputs": { - "current_temp": 29.0, - "deb_prediction": 30.2, - }, - "trigger_summary": { - "trigger_types": ["momentum_spike"] if trigger_count > 0 else [], - "suppression_snapshot": { - "max_temp_time": peak_time, - }, - }, - "market": { - "low_yes_signal": { - "should_push": signal_label in {"BUY YES", "BUY NO"}, - }, - }, - }, - "triggered_alerts": [ - { - "type": "momentum_spike", - } - ] - if trigger_count > 0 - else [], - "telegram": { - "zh": f"CRITICAL {city}", - }, - } - - -def test_market_monitor_cycle_restores_critical_alert_push(monkeypatch): - payload = _focus_payload( - city="ankara", - severity="high", - trigger_count=2, - signal_label="BUY YES", - yes_buy=0.05, - edge_percent=12.0, - ) - monkeypatch.setattr( - "src.utils.telegram_push.build_trade_alert_for_city", - lambda city, config: payload, - ) - bot = _DigestBot() - state = {} - - dirty = _run_market_monitor_cycle( - bot=bot, - config={}, - chat_ids=["chat"], - cities=["ankara"], - state=state, - alert_cooldown_sec=1800, - min_severity="medium", - min_trigger_count=2, - sleep_between_cities_sec=0, - ) - - assert dirty is True - assert [row["text"] for row in bot.messages] == ["CRITICAL ankara"] - assert state["last_by_city"]["ankara"]["active"] is True - -