Jyotisha/tests/smoke_test_runner.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Jyotish 事件应期回归测试框架 v1.0
自动验证 Dasha 推运精度：名人已知人生事件 vs 引擎预测

用法:
  python3 tests/smoke_test_runner.py
  python3 tests/smoke_test_runner.py --case einstein,obama
  python3 tests/smoke_test_runner.py --report json

依赖: 需要 engin 可运行（pyswisseph 已安装）
"""

import json, os, sys, subprocess, argparse, time
from datetime import datetime
from pathlib import Path

SCRIPTS_DIR = str(Path(__file__).resolve().parent.parent / "scripts")
if SCRIPTS_DIR not in sys.path:
    sys.path.insert(0, SCRIPTS_DIR)

CASES_FILE = Path(__file__).resolve().parent / "prediction_regression_cases.json"

# ============================================================
# 测试用例 — 名人已知事件时间线
# 每个事件可验证 Dasha 推运系统的精度
# ============================================================
CELEBRITY_EVENTS = {
    "obama": {
        "name": "Barack Obama",
        "birth": {"year": 1961, "month": 8, "day": 4, "hour": 19, "minute": 24,
                  "lat": 21.3, "lon": -157.8, "tz": -10.0},
        "known_lagna": "Capricorn",
        "events": [
            {"date": "1992-10-03", "type": "marriage", "desc": "与 Michelle 结婚", "target_houses": [7]},
            {"date": "2008-11-04", "type": "career_peak", "desc": "当选美国总统", "target_houses": [10, 1]},
            {"date": "2012-11-06", "type": "career_peak", "desc": "连任美国总统", "target_houses": [10, 1]},
            {"date": "2020-11-17", "type": "publication", "desc": "回忆录《应许之地》出版", "target_houses": [3, 10]},
        ]
    },
    "trump": {
        "name": "Donald Trump",
        "birth": {"year": 1946, "month": 6, "day": 14, "hour": 10, "minute": 54,
                  "lat": 40.7, "lon": -73.8, "tz": -5.0},
        "known_lagna": "Leo",
        "events": [
            {"date": "1977-04-07", "type": "marriage", "desc": "与 Ivana 结婚", "target_houses": [7]},
            {"date": "1990-04-04", "type": "business", "desc": "Taj Mahal 赌场开业", "target_houses": [10, 11]},
            {"date": "2016-11-08", "type": "career_peak", "desc": "当选美国总统", "target_houses": [10, 1]},
            {"date": "2024-11-05", "type": "career_peak", "desc": "再次当选总统", "target_houses": [10, 1]},
        ]
    },
    "jobs": {
        "name": "Steve Jobs",
        "birth": {"year": 1955, "month": 2, "day": 24, "hour": 19, "minute": 15,
                  "lat": 37.8, "lon": -122.4, "tz": -8.0},
        "known_lagna": "Leo",
        "events": [
            {"date": "1976-04-01", "type": "career_start", "desc": "Apple 公司成立", "target_houses": [10, 11]},
            {"date": "1985-09-17", "type": "career_loss", "desc": "被逐出 Apple", "target_houses": [10, 8]},
            {"date": "1997-07-09", "type": "career_return", "desc": "回归 Apple 任 CEO", "target_houses": [10, 1]},
            {"date": "2007-01-09", "type": "career_peak", "desc": "发布第一代 iPhone", "target_houses": [10, 3]},
            {"date": "2011-10-05", "type": "death", "desc": "因胰腺癌去世", "target_houses": [8, 1]},
        ]
    },
    "einstein": {
        "name": "Albert Einstein",
        "birth": {"year": 1879, "month": 3, "day": 14, "hour": 11, "minute": 30,
                  "lat": 48.4, "lon": 9.98, "tz": 0.89},
        "known_lagna": "Gemini",
        "events": [
            {"date": "1905-06-30", "type": "career_peak", "desc": "奇迹年：发表狭义相对论等四篇论文", "target_houses": [10, 5, 9]},
            {"date": "1915-11-25", "type": "career_peak", "desc": "完成广义相对论", "target_houses": [10, 9]},
            {"date": "1921-04-02", "type": "honor", "desc": "诺贝尔物理学奖", "target_houses": [10, 5]},
            {"date": "1955-04-18", "type": "death", "desc": "在普林斯顿去世", "target_houses": [8, 1]},
        ]
    },
    "monroe": {
        "name": "Marilyn Monroe",
        "birth": {"year": 1926, "month": 6, "day": 1, "hour": 9, "minute": 30,
                  "lat": 34.1, "lon": -118.3, "tz": -8.0},
        "known_lagna": "Cancer",
        "events": [
            {"date": "1942-06-19", "type": "marriage", "desc": "与 Jim Dougherty 结婚", "target_houses": [7]},
            {"date": "1953-00-00", "type": "career_peak", "desc": "《绅士爱美人》上映，成为巨星", "target_houses": [10, 5]},
            {"date": "1962-08-05", "type": "death", "desc": "在洛杉矶去世", "target_houses": [8, 1]},
        ]
    },
    "dicaprio": {
        "name": "Leonardo DiCaprio",
        "birth": {"year": 1974, "month": 11, "day": 11, "hour": 2, "minute": 47,
                  "lat": 34.1, "lon": -118.3, "tz": -8.0},
        "known_lagna": "Virgo",
        "events": [
            {"date": "1997-12-19", "type": "career_peak", "desc": "《泰坦尼克号》上映", "target_houses": [10, 5]},
            {"date": "2016-02-28", "type": "honor", "desc": "凭《荒野猎人》获奥斯卡最佳男主角", "target_houses": [10, 1]},
        ]
    },
    "mjackson": {
        "name": "Michael Jackson",
        "birth": {"year": 1958, "month": 8, "day": 29, "hour": 19, "minute": 33,
                  "lat": 41.6, "lon": -87.3, "tz": -6.0},
        "known_lagna": "Pisces",
        "events": [
            {"date": "1982-11-30", "type": "career_peak", "desc": "《Thriller》专辑发行", "target_houses": [10, 5]},
            {"date": "2009-06-25", "type": "death", "desc": "在洛杉矶去世", "target_houses": [8, 1]},
        ]
    },
    "indira": {
        "name": "Indira Gandhi",
        "birth": {"year": 1917, "month": 11, "day": 19, "hour": 23, "minute": 11,
                  "lat": 25.5, "lon": 81.9, "tz": 5.5},
        "known_lagna": "Leo",
        "events": [
            {"date": "1966-01-24", "type": "career_peak", "desc": "就任印度总理", "target_houses": [10, 1]},
            {"date": "1984-10-31", "type": "death", "desc": "遇刺身亡", "target_houses": [8, 1]},
        ]
    },
    "presley": {
        "name": "Elvis Presley",
        "birth": {"year": 1935, "month": 1, "day": 8, "hour": 4, "minute": 35,
                  "lat": 34.3, "lon": -88.4, "tz": -6.0},
        "known_lagna": "Scorpio",
        "events": [
            {"date": "1956-01-27", "type": "career_start", "desc": "首张专辑《Heartbreak Hotel》", "target_houses": [10, 5]},
            {"date": "1977-08-16", "type": "death", "desc": "在 Graceland 去世", "target_houses": [8, 1]},
        ]
    },
    "curie": {
        "name": "Marie Curie",
        "birth": {"year": 1867, "month": 11, "day": 7, "hour": 12, "minute": 0,
                  "lat": 52.2, "lon": 21.0, "tz": 1.0},
        "known_lagna": "Sagittarius",
        "events": [
            {"date": "1903-12-10", "type": "honor", "desc": "诺贝尔物理学奖（与 Pierre Curie 共享）", "target_houses": [10, 5]},
            {"date": "1911-12-10", "type": "honor", "desc": "诺贝尔化学奖（唯一两次获奖女性）", "target_houses": [10, 5]},
            {"date": "1934-07-04", "type": "death", "desc": "因再生障碍性贫血去世", "target_houses": [8, 1]},
        ]
    },
    "hanks": {
        "name": "Tom Hanks",
        "birth": {"year": 1956, "month": 7, "day": 9, "hour": 11, "minute": 17,
                  "lat": 37.9, "lon": -122.1, "tz": -8.0},
        "known_lagna": "Virgo",
        "events": [
            {"date": "1994-03-21", "type": "honor", "desc": "凭《费城故事》获奥斯卡影帝", "target_houses": [10, 1]},
            {"date": "1995-03-27", "type": "honor", "desc": "凭《阿甘正传》再获奥斯卡影帝", "target_houses": [10, 1]},
        ]
    },
    "jolie": {
        "name": "Angelina Jolie",
        "birth": {"year": 1975, "month": 6, "day": 4, "hour": 9, "minute": 9,
                  "lat": 34.1, "lon": -118.3, "tz": -8.0},
        "known_lagna": "Cancer",
        "events": [
            {"date": "2000-03-26", "type": "honor", "desc": "凭《移魂女郎》获奥斯卡女配", "target_houses": [10, 5]},
            {"date": "2013-05-14", "type": "health", "desc": "预防性双乳切除术公告", "target_houses": [8, 6]},
        ]
    },
}


# ============================================================
# 测试运行器
# ============================================================

class SmokeTestRunner:
    def __init__(self, engine_path=None):
        self.engine = engine_path or os.path.join(SCRIPTS_DIR, "jyotish_engine.py")
        self.results = {"total": 0, "passed": 0, "failed": 0, "errors": [], "details": []}
    
    def _run_engine(self, birth, command, extra_args=None):
        """调用 jyotish_engine.py CLI"""
        cmd = [
            sys.executable, self.engine, command,
            "--year", str(birth["year"]), "--month", str(birth["month"]),
            "--day", str(birth["day"]), "--hour", str(birth["hour"]),
            "--minute", str(birth["minute"]),
            "--lat", str(birth["lat"]), "--lon", str(birth["lon"]),
            "--tz", str(birth["tz"]), "--node-mode", "mean",
        ]
        if extra_args:
            cmd.extend(extra_args)
        try:
            result = subprocess.run(cmd, capture_output=True, text=True, timeout=180)
            if result.returncode == 0 and result.stdout.strip():
                return True, json.loads(result.stdout)
            return False, result.stderr if result.stderr else result.stdout
        except subprocess.TimeoutExpired:
            return False, "TIMEOUT (180s)"
        except Exception as e:
            return False, str(e)
    
    def test_chart_basics(self, case_id, data):
        """验证基础排盘：上升、太阳、月亮星座"""
        ok, chart = self._run_engine(data["birth"], "chart")
        self.results["total"] += 1
        
        if not ok:
            self.results["failed"] += 1
            err = f"{data['name']}: chart calculation FAILED — {chart}"
            self.results["errors"].append(err)
            self.results["details"].append({"case": data["name"], "test": "chart", "result": "FAIL", "error": str(chart)})
            return
        
        planets = chart.get("planets", {})
        asc = chart.get("ascendant", {}).get("sign", "?").split()[0] if isinstance(chart.get("ascendant"), dict) else "?"
        
        checks = []
        if asc and data.get("known_lagna"):
            match = asc.lower() == data["known_lagna"].lower()
            checks.append(("Ascendant", match, f"{asc} vs {data['known_lagna']}"))
        
        sun = planets.get("Sun", {}).get("sign", "?")
        if sun and data.get("known_sun_sign"):
            match = sun.lower() == data.get("known_sun_sign", "").lower()
            checks.append(("Sun sign", match, f"{sun} vs {data['known_sun_sign']}"))
        
        moon = planets.get("Moon", {}).get("sign", "?")
        if moon and data.get("known_moon_sign"):
            match = moon.lower() == data.get("known_moon_sign", "").lower()
            checks.append(("Moon sign", match, f"{moon} vs {data['known_moon_sign']}"))
        
        all_pass = all(c[1] for c in checks)
        if all_pass:
            self.results["passed"] += 1
        else:
            self.results["failed"] += 1
            fails = [c[2] for c in checks if not c[1]]
            self.results["errors"].append(f"{data['name']}: chart mismatch — {', '.join(fails)}")
        
        self.results["details"].append({
            "case": data["name"], "test": "chart",
            "result": "PASS" if all_pass else "FAIL",
            "checks": [{"type": c[0], "pass": c[1], "detail": c[2]} for c in checks]
        })
    
    def test_dasha_timeline(self, case_id, data):
        """验证 Dasha 时间线与已知事件的交叉"""
        events = data.get("events", [])
        if not events:
            return
        
        ok, dasha = self._run_engine(data["birth"], "dasha", ["--years", "120"])
        self.results["total"] += len(events)
        
        if not ok:
            for ev in events:
                self.results["failed"] += 1
                self.results["errors"].append(f"{data['name']}: dasha calc FAILED for '{ev['desc']}'")
            return
        
        dasa_periods = dasha.get("timeline", []) if isinstance(dasha, dict) else []
        if not dasa_periods:
            dasa_periods = dasha if isinstance(dasha, list) else []
        
        for ev in events:
            ev_date = datetime.strptime(ev["date"].replace("-00", "-01"), "%Y-%m-%d")
            # Find active Mahadasha at event date
            active_md = None
            for d in dasa_periods:
                start = datetime.strptime(d.get("start", "1900-01-01")[:10], "%Y-%m-%d")
                end = datetime.strptime(d.get("end", "2100-01-01")[:10], "%Y-%m-%d")
                if start <= ev_date <= end:
                    active_md = d.get("lord", "?")
                    break
            
            if ev["type"] == "career_start" or ev["type"] == "career_peak" or ev["type"] == "career_return":
                favorable = active_md in ["Sun", "Moon", "Mars", "Jupiter", "Venus", "Mercury"]
            elif ev["type"] == "career_loss":
                favorable = active_md in ["Saturn", "Rahu", "Ketu"]
            elif ev["type"] == "death":
                favorable = active_md in ["Saturn", "Rahu", "Ketu", "Mars"]
            elif ev["type"] == "marriage":
                favorable = active_md in ["Jupiter", "Venus", "Mercury"]
            elif ev["type"] == "honor":
                favorable = active_md in ["Sun", "Jupiter", "Venus"]
            else:
                favorable = True  # neutral test
            
            test_name = f"{ev['desc']} [{ev['date']}] Dasha={active_md}"
            if favorable:
                self.results["passed"] += 1
            else:
                self.results["failed"] += 1
                self.results["errors"].append(
                    f"{data['name']}: '{ev['desc']}' ({ev['date']}) — active Mahadasha={active_md} — unexpected for {ev['type']}"
                )
            
            self.results["details"].append({
                "case": data["name"], "test": "dasha_event",
                "event": ev["desc"], "date": ev["date"],
                "active_dasha": active_md,
                "result": "PASS" if favorable else "FAIL",
            })
    
    def run_all(self, case_filter=None):
        """运行所有回归测试"""
        print("=" * 70)
        print("Jyotish 事件应期回归测试")
        print("=" * 70)
        
        cases = CELEBRITY_EVENTS
        if case_filter:
            cases = {k: v for k, v in cases.items() if k in case_filter}
        
        for case_id, data in cases.items():
            print(f"\n{'─' * 60}")
            print(f"  Case: {data['name']}")
            print(f"{'─' * 60}")
            
            # Test 1: Chart basics
            print(f"  [1/2] Chart validation...")
            self.test_chart_basics(case_id, data)
            
            # Test 2: Dasha event timeline
            if data.get("events"):
                print(f"  [2/2] Dasha event timeline ({len(data['events'])} events)...")
                self.test_dasha_timeline(case_id, data)
        
        return self.results
    
    def report(self, fmt="text"):
        """生成测试报告"""
        r = self.results
        if fmt == "json":
            return json.dumps(r, ensure_ascii=False, indent=2)
        
        lines = []
        lines.append("\n" + "=" * 70)
        lines.append("  回归测试报告")
        lines.append("=" * 70)
        lines.append(f"  总计: {r['total']} | 通过: {r['passed']} | 失败: {r['failed']}")
        if r["total"] > 0:
            rate = r["passed"] / r["total"] * 100
            lines.append(f"  通过率: {rate:.1f}%")
        
        if r["errors"]:
            lines.append(f"\n  失败项 ({len(r['errors'])}):")
            for err in r["errors"]:
                lines.append(f"    ✗ {err}")
        
        lines.append("=" * 70)
        return "\n".join(lines)


def main():
    parser = argparse.ArgumentParser(description="Jyotish 事件应期回归测试")
    parser.add_argument("--case", help="指定案例 ID，逗号分隔")
    parser.add_argument("--report", default="text", choices=["text", "json"])
    parser.add_argument("--skip-dasha", action="store_true", help="跳过 Dasha 推运测试")
    args = parser.parse_args()
    
    case_filter = None
    if args.case:
        case_filter = [c.strip() for c in args.case.split(",")]
    
    runner = SmokeTestRunner()
    runner.run_all(case_filter)
    print(runner.report(args.report))
    
    # 退出码：任何失败 → 非零
    sys.exit(1 if runner.results["failed"] > 0 else 0)


if __name__ == "__main__":
    main()