feat: introduce PolyWeather web map API for interactive weather data visualization, analysis, and market-related data collection.

2026-03-06 10:04:00 +08:00
parent d9876256b3
commit 86c15d2fd2
4 changed files with 756 additions and 1 deletions
@@ -0,0 +1,100 @@
+from src.analysis.market_alert_engine import build_trading_alerts
+
+
+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,
+        },
+        "metar_recent_obs": [
+            {"time": "10:30", "wdir": 180},
+            {"time": "10:00", "wdir": 60},
+        ],
+        "mgm_nearby": [
+            {
+                "name": "Ankara (Bölge/Center)",
+                "istNo": "17130",
+                "lat": 39.95,
+                "lon": 32.97,
+                "temp": 12.4,
+            },
+            {
+                "name": "Airport (MGM/17128)",
+                "istNo": "17128",
+                "lat": 40.1281,
+                "lon": 32.9951,
+                "temp": 11.2,
+            },
+        ],
+    }
+
+
+def _sample_market_snapshot():
+    return {
+        "city": "ankara",
+        "target_date": "2026-03-07",
+        "markets": [
+            {
+                "id": "m1",
+                "question": "Will temperature in Ankara exceed 11.5°C on March 7?",
+                "threshold": 11.5,
+                "threshold_unit": "C",
+                "contract_type": "exceed",
+                "outcomes": [
+                    {"name": "Yes", "buy_price": 0.73, "last_price": 0.72},
+                    {"name": "No", "buy_price": 0.27, "last_price": 0.28},
+                ],
+            }
+        ],
+    }
+
+
+def test_trading_alerts_all_core_rules_trigger():
+    out = build_trading_alerts(
+        city_weather=_sample_weather_payload(),
+        market_snapshot=_sample_market_snapshot(),
+        map_url="https://example.com/map",
+    )
+
+    assert out["trigger_count"] >= 3
+    assert out["rules"]["momentum_spike"]["triggered"] is True
+    assert out["rules"]["forecast_breakthrough"]["triggered"] is True
+    assert out["rules"]["kill_zone"]["triggered"] is True
+    assert out["rules"]["advection"]["triggered"] is True
+
+    msg = out["telegram"]["zh"]
+    assert "PolyWeather 异动预警" in msg
+    assert "动量突变" in msg
+    assert "No\" 单需谨慎" 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,
+        market_snapshot=_sample_market_snapshot(),
+    )
+    assert out["rules"]["forecast_breakthrough"]["triggered"] is False
+