Test Collapse Invariant

1# scripts/test_collapse_invariant.py
2"""Stage-A release gate: a single-unit run is byte-equivalent to the legacy flow
3on every Python surface (scoring call, diagram render). Prose surfaces carry the
4same invariant via phase _asserts (single-unit: no grouping question, no delta
5questions, no unit cards)."""
6import build_diagram
7import scoring
8
9ANSWERS = {
10    "session_duration": "over_8hr", "traffic_pattern": "bursty",
11    "platform_fit": "none", "session_state": "stateful",
12    "ops_preference": "low", "isolation": "not_needed",
13    "memory_needs": "cross_session", "multi_agent": "no",
14    "framework": "langgraph", "existing_cluster": "none",
15    "multi_cloud": "no", "idle_resume": "unknown",
16    "compute_tier": "light", "launch_concurrency": "unknown",
17}
18
19
20def test_scoring_identical_via_unit_loop():
21    # The clarify loop merges system+unit answers; for one unit that merge IS the
22    # legacy flat answers dict — the scoring call must be literally identical.
23    legacy = scoring.score({"answers": ANSWERS})
24    system = {k: ANSWERS[k] for k in
25              ("ops_preference", "existing_cluster", "multi_cloud", "platform_fit")}
26    assert set(system.keys()) == {"ops_preference", "existing_cluster", "multi_cloud", "platform_fit"}
27    unit = {k: v for k, v in ANSWERS.items() if k not in system}
28    unitized = scoring.score({"answers": {**system, **unit}})
29    assert unitized == legacy
30
31
32def test_diagram_identical_for_single_unit_design():
33    result = {"verdict": "agentcore", "agentcore_services": ["memory"],
34              "model_recommendation": {"model": "claude_sonnet_5"}}
35    design = {"units": [{"id": "only", "workload_class": "agent_session",
36                         "verdict": "agentcore",
37                         "model_recommendation": {"model": "claude_sonnet_5"}}],
38              "platform": {"mode": "split", "runtime": None,
39                           "interconnect": "in_process", "shared_services": []}}
40    assert build_diagram.build_diagram(result, {}, design=design) == \
41        build_diagram.build_diagram(result, {})