Files
Jyotisha/scripts/research/capture_consult_evidence_card_golden.py
T
Jesse_ChenandClaude Opus 5.5 75a1844cdf fix(consult): project the running Narayana period and pratyantar dates to the model (BUG-1054)
timingKeys now admits current_dasha.md/ad/pd (sign, lord, years,
start_age, end_age), remaining_years and pratyantar_dasha_timeline. Depth
and item caps unchanged. Golden regression over three public AA engine
captures asserts values, not key presence.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_017eEAG8HD3mm8gsKXgk8uU8
2026-09-27 02:54:11 +08:00

99 lines
3.7 KiB
Python

"""Capture real engine consultation responses for the evidence-card tests.
Three public AA charts (Steve Jobs, Barack Obama, Elizabeth Taylor), the
research reference date, raman ayanamsa, mean nodes, family route. External
VedAstro is not called (same stand-in as the research runner). The response is
trimmed by key only: every kept value is the engine's own value, unchanged.
PYTHONHASHSEED=0 python scripts/research/capture_consult_evidence_card_golden.py
"""
from __future__ import annotations
import json
import os
import sys
from pathlib import Path
from typing import Any
ROOT = Path(__file__).resolve().parents[2]
sys.path[:0] = [str(ROOT), str(ROOT / "scripts"), str(ROOT / "scripts" / "research")]
from consult_evidence_card_lib import ( # noqa: E402
AYANAMSA,
NODE_MODE,
QUESTIONS,
REFERENCE_DATE,
load_public_charts,
)
import consult_evidence_card_run as runner # noqa: E402
OUT = ROOT / "frontend" / "tests" / "fixtures" / "consult-evidence-card-golden.json"
TOP_KEEP = (
"success", "question", "routing", "consumer_context", "thematic_report",
"birth_time_sensitivity", "reference_transparency", "candidate_range",
"range_boundary_contexts",
)
RECTIFICATION_KEEP = ("effective_accuracy", "lagna_boundary", "summary", "enabled_vargas")
CHART_KEEP = ("success", "birth", "ascendant", "planets", "houses", "shadbala", "dasha", "yogas")
MODULE_KEEP = (
"varga_spectrum", "shadbala", "arudha_padas", "jaimini", "narayana_dasha",
"ashtakavarga", "dasha_sub_periods", "kp_cusps", "gulika",
"functional_benefic_malefic", "kakshya", "yogas", "chara_dasha", "transits",
)
NARAYANA_KEEP = ("lagna_sign", "current_dasha", "current_year", "current_age", "mahadasha_sequence")
def _pick(value: dict[str, Any], keys: tuple[str, ...]) -> dict[str, Any]:
return {key: value[key] for key in keys if key in value}
def trim(workflow: dict[str, Any]) -> dict[str, Any]:
out = _pick(workflow, TOP_KEEP)
out["rectification"] = _pick(workflow.get("rectification") or {}, RECTIFICATION_KEEP)
chart = workflow.get("chart") or {}
kept_chart = _pick(chart, CHART_KEEP)
modules = _pick(chart.get("modules") or {}, MODULE_KEEP)
if isinstance(modules.get("narayana_dasha"), dict):
modules["narayana_dasha"] = _pick(modules["narayana_dasha"], NARAYANA_KEEP)
kept_chart["modules"] = modules
out["chart"] = kept_chart
return out
def main() -> int:
if os.environ.get("PYTHONHASHSEED") != "0":
raise SystemExit("Set PYTHONHASHSEED=0 before starting this process (ERR-111).")
runner._block_external_vedastro()
question = next(item for item in QUESTIONS if item["id"] == "family")
charts = []
for chart in load_public_charts(ROOT):
workflow = runner._run_workflow(runner._workflow_body(chart, question))
charts.append({
"id": chart["id"],
"label": chart["label"],
"source": chart["source"],
"case_id": chart["case_id"],
"rodden_rating": chart["rodden_rating"],
"workflow": trim(workflow),
})
payload = {
"source": "scripts/research/capture_consult_evidence_card_golden.py",
"note": "Real engine consultation_workflow responses for three public AA charts, trimmed by key only.",
"reference_date": REFERENCE_DATE,
"ayanamsa": AYANAMSA,
"node_mode": NODE_MODE,
"route": question["domain"],
"question": question["question"],
"external_vedastro": "not_called",
"charts": charts,
}
OUT.write_text(json.dumps(payload, ensure_ascii=False, separators=(",", ":")) + "\n", encoding="utf-8")
print(f"wrote {OUT} ({OUT.stat().st_size} bytes)")
return 0
if __name__ == "__main__":
raise SystemExit(main())