Files
Jyotisha/tests/test_life_event_graph_v1.py
T
2026-07-01 19:15:35 +08:00

296 lines
11 KiB
Python

#!/usr/bin/env python3
"""Regression tests for Life Event Graph v1."""
from __future__ import annotations
from mcp_server import _build_life_event_graph, _collect_strict_evidence
def test_life_event_graph_folds_strict_evidence_and_vedastro_top_event() -> None:
strict = {
"question_type": "relationship",
"event_judgement": {
"event_family": "relationship",
"score": 85,
"verdict": "high_probability_window",
"dominant_label": "legal_marriage",
"secondary_context": [
"darakaraka_active",
"jaimini_support",
"ul_support",
"external_activation_support",
"synastry_support",
"synastry_compatibility_support",
"synastry_protective_kuta_support",
],
"primary_drivers": [
"marriage_convergence",
"vimshottari_current",
"narayana_current",
"darakaraka",
"upapada_lagna",
],
},
"present_evidence": {
"vimshottari_current": {"mahadasha": "Venus", "antardasha": "Moon"},
"narayana_current": {"sign": "Libra", "lord": "Venus"},
"marriage_convergence": {"convergence_level": "L4", "probability": "70-85%"},
"external_activation": {
"level": "moderate",
"source": "vedastro_service_adapter_candidate",
"signals": ["vedastro_range_scan"],
"events": [
{
"event_id": "GocharJupiterIn7th",
"signal_key": "gochar_jupiter_7th_marriage",
"signal_label": "Jupiter in 7th marriage window",
"signal_family": "marriage_trigger",
"score": 72,
"start": "2026-05-01",
"end": "2026-06-01",
"tags": ["marriage", "transit"],
}
],
},
},
"confidence_cap": "medium-high",
"missing_evidence": [],
"blocked": False,
}
graph = _build_life_event_graph("relationship", strict)
assert graph["version"] == "life_event_graph_v1"
assert graph["route"] == "relationship"
assert graph["dominant_label"] == "legal_marriage"
assert graph["confidence_cap"] == "medium-high"
assert graph["blocked"] is False
assert graph["event_nodes"][0] == {
"kind": "judgement",
"label": "legal_marriage",
"verdict": "high_probability_window",
"score": 85,
"source": "strict_workflow",
}
assert graph["event_nodes"][1] == {
"kind": "dasha_window",
"label": "Venus/Moon",
"source": "vimshottari_current",
}
assert graph["event_nodes"][2] == {
"kind": "dasha_window",
"label": "Libra/Venus",
"source": "narayana_current",
}
assert graph["event_nodes"][3] == {
"kind": "convergence",
"label": "marriage_convergence",
"level": "L4",
"probability": "70-85%",
"source": "dasa_convergence",
}
assert graph["event_nodes"][4] == {
"kind": "external_window",
"label": "Jupiter in 7th marriage window",
"event_id": "GocharJupiterIn7th",
"signal_key": "gochar_jupiter_7th_marriage",
"signal_family": "marriage_trigger",
"score": 72,
"start": "2026-05-01",
"end": "2026-06-01",
"tags": ["marriage", "transit"],
"source": "vedastro_service_adapter_candidate",
}
assert {
"kind": "context",
"label": "synastry_compatibility_support",
"source": "event_judgement.secondary_context",
} in graph["event_nodes"]
assert {
"kind": "context",
"label": "synastry_protective_kuta_support",
"source": "event_judgement.secondary_context",
} in graph["event_nodes"]
def test_life_event_graph_is_returned_from_strict_relationship_evidence() -> None:
result = {
"modules": {
"varga_full": {"D9_Navamsa": {"summary": "ok"}},
"special_lagnas": {"Upapada_Lagna": {"sign": "Libra", "lord": "Venus"}},
"jaimini": {
"darakaraka": {"planet": "Venus", "house": 7},
"marriage_support": {"dk_7h_link": True},
},
"vivah_saham": {"sign": "Taurus", "house": 7},
"dasha": {"current_dasha": {"mahadasha": "Venus", "antardasha": "Moon"}},
"narayana_dasha": {"current_dasha": {"sign": "Libra", "lord": "Venus"}},
"dasa_convergence": {
"domain_activations": {
"marriage_partnership": {"convergence_level": "L4", "probability": "70-85%"}
}
},
"external_activation": {
"evidence_ledger": [
{
"source": "vedastro_service_adapter_candidate",
"operation": "range_scan",
"domain": "marriage",
"event_id": "GocharJupiterIn7th",
"signal_key": "gochar_jupiter_7th_marriage",
"signal_label": "Jupiter in 7th marriage window",
"signal_family": "marriage_trigger",
"score": 72,
"start": "2026-05-01",
"end": "2026-06-01",
"tags": ["marriage", "transit"],
}
]
},
}
}
strict = _collect_strict_evidence("relationship", result)
assert strict["life_event_graph"]["version"] == "life_event_graph_v1"
assert strict["life_event_graph"]["route"] == "relationship"
assert strict["life_event_graph"]["dominant_label"] == "legal_marriage"
assert any(node["kind"] == "external_window" for node in strict["life_event_graph"]["event_nodes"])
def test_strict_workflow_accepts_adapter_range_scan_result_without_manual_repackaging() -> None:
result = {
"modules": {
"varga_full": {"D9_Navamsa": {"summary": "ok"}},
"special_lagnas": {"Upapada_Lagna": {"sign": "Libra", "lord": "Venus"}},
"jaimini": {
"darakaraka": {"planet": "Venus", "house": 7},
"marriage_support": {"dk_7h_link": True},
},
"vivah_saham": {"sign": "Taurus", "house": 7},
"dasha": {"current_dasha": {"mahadasha": "Venus", "antardasha": "Moon"}},
"narayana_dasha": {"current_dasha": {"sign": "Libra", "lord": "Venus"}},
"dasa_convergence": {
"domain_activations": {
"marriage_partnership": {"convergence_level": "L4", "probability": "70-85%"}
}
},
"vedastro_range_scan_result": {
"backend": "vedastro_service_adapter_candidate",
"available": True,
"status": "ok",
"operation": "range_scan",
"domain": "marriage",
"evidence_ledger": [
{
"source": "vedastro_service_adapter_candidate",
"operation": "range_scan",
"domain": "marriage",
"event_id": "GocharJupiterIn7th",
"signal_key": "gochar_jupiter_7th_marriage",
"signal_label": "Jupiter in 7th marriage window",
"signal_family": "marriage_trigger",
"score": 72,
"start": "2026-05-01",
"end": "2026-06-01",
"tags": ["marriage", "transit"],
}
],
"source_metadata": {
"request_hash": "a" * 64,
"response_hash": "b" * 64,
"artifact_path": "scratch/local/vedastro_adapter/range.json",
},
},
}
}
strict = _collect_strict_evidence("relationship", result)
external = strict["present_evidence"]["external_activation"]
assert external["level"] == "moderate"
assert external["source"] == "vedastro_service_adapter_candidate"
assert external["provenance"]["request_hash"] == "a" * 64
assert any(node["kind"] == "external_window" for node in strict["life_event_graph"]["event_nodes"])
def test_life_event_graph_surfaces_main_entry_overview_boundary_node() -> None:
strict = {
"event_judgement": {
"event_family": "relationship",
"score": 74,
"verdict": "moderate_probability_window",
"dominant_label": "legal_marriage",
"secondary_context": ["external_activation_support"],
},
"present_evidence": {
"external_activation": {
"level": "moderate",
"source": "vedastro_service_adapter_candidate",
"signals": ["vedastro_range_scan"],
"events": [
{
"event_id": "GocharJupiterIn7th",
"signal_key": "gochar_jupiter_7th_marriage",
"signal_label": "Jupiter in 7th marriage window",
"signal_family": "marriage_trigger",
"score": 72,
"start": "2026-06-29",
"end": "2026-06-29",
"tags": ["marriage", "transit"],
}
],
"provenance": {
"ingestion_profile": "main_entry_overview",
"search_scope": "single_day_overview",
"reference_date": "2026-06-29",
},
}
},
"confidence_cap": "medium",
"missing_evidence": [],
"blocked": False,
}
graph = _build_life_event_graph("relationship", strict)
assert {
"kind": "external_overview",
"label": "VedAstro main-entry overview",
"ingestion_profile": "main_entry_overview",
"search_scope": "single_day_overview",
"reference_date": "2026-06-29",
"source": "vedastro_service_adapter_candidate",
} in graph["event_nodes"]
def test_life_event_graph_surfaces_ranked_official_day_window_nodes() -> None:
strict = {
"event_judgement": {"event_family": "career", "verdict": "moderate_probability_window", "score": 74},
"present_evidence": {
"external_activation": {
"level": "moderate",
"source": "vedastro_service_adapter_candidate",
"daily_windows": [
{
"date": "2026-07-18",
"domain": "career",
"score": 5,
"confidence": "medium_high",
"event_count": 2,
"signal_families": ["career_trigger"],
"event_ids": ["GocharJupiterAspect10th", "CareerExpansionWindow"],
"top_signal_label": "Career expansion window",
}
],
}
},
"confidence_cap": "medium",
"missing_evidence": [],
"blocked": False,
}
graph = _build_life_event_graph("career", strict)
assert any(node["kind"] == "official_day_window" and node["date"] == "2026-07-18" for node in graph["event_nodes"])