"""
印度占星端点测试 — 计算层 + 端点层
跑法: <spike-venv>/bin/python test_vedic_chart.py
固定输入: 1990-01-01 12:00, New Delhi (28.6139, 77.2090), Asia/Kolkata
基准值来自 Task 0 spike 验证 + 与 kerykeion 交叉一致。
"""
import sys, os, io

VEDIC = os.path.join(os.path.dirname(os.path.abspath(__file__)), "vedic")
sys.path.insert(0, VEDIC)

FIXED = dict(year=1990, month=1, day=1, hour=12, minute=0,
             lat=28.6139, lon=77.2090, tz_str="Asia/Kolkata")


def _calc():
    _real = sys.stdout
    sys.stdout = io.StringIO()  # 静音 PyJHora noisy import/print
    try:
        from engine import calculate_full_chart
        return calculate_full_chart(**FIXED)
    finally:
        sys.stdout = _real


def test_lagna_correct():
    c = _calc()
    assert c["lagna"]["sign"] == "Pisces", c["lagna"]["sign"]
    assert c["lagna"]["nakshatra"]["name"] == "Uttara Bhadrapada", c["lagna"]["nakshatra"]
    print("  ✓ test_lagna_correct")


def test_all_sections_present():
    c = _calc()
    for k in ["ayanamsa", "lagna", "planets", "dashas", "shadbala", "d9", "karakas", "aspects"]:
        assert k in c, f"missing section: {k}"
    for g in ["Sun", "Moon", "Mars", "Mercury", "Jupiter", "Venus", "Saturn", "Rahu", "Ketu"]:
        assert g in c["planets"], f"missing graha: {g}"
    print("  ✓ test_all_sections_present")


def test_dasha_has_periods():
    c = _calc()
    assert c["dashas"], "dashas empty"
    print(f"  ✓ test_dasha_has_periods ({len(c['dashas'])} periods)")


def test_json_serializable():
    """端点核心: 整盘能用端点的 default lambda 安全序列化成 JSON (numpy/tuple 不炸)"""
    import json
    c = _calc()
    s = json.dumps(c, default=lambda o: float(o) if hasattr(o, "__float__") else str(o))
    back = json.loads(s)
    assert back["lagna"]["sign"] == "Pisces"
    assert "shadbala" in back and "dashas" in back
    print(f"  ✓ test_json_serializable ({len(s)} bytes)")


if __name__ == "__main__":
    print("=== Vedic calc-layer tests ===")
    test_lagna_correct()
    test_all_sections_present()
    test_dasha_has_periods()
    test_json_serializable()
    print("✅ ALL CALC-LAYER TESTS PASS")