Test Cost Levers

1"""Content lock for references/decision-refs/cost-levers.md.
2
3Cost-optimization levers are the ONLY legal source for discount/lever citations in
4agent-advisor estimates and reports. These tests pin the five canonical levers and
5verify that estimate.md references cost-levers.md and mentions the drivers[] field.
6"""
7import pathlib
8import re
9
10COST_LEVERS_MD = (pathlib.Path(__file__).parent.parent
11                  / "references" / "decision-refs" / "cost-levers.md")
12ESTIMATE_MD = (pathlib.Path(__file__).parent.parent
13               / "references" / "phases" / "estimate" / "estimate.md")
14
15
16def _cost_levers_text():
17    return COST_LEVERS_MD.read_text()
18
19
20def _estimate_text():
21    return ESTIMATE_MD.read_text()
22
23
24def test_cost_levers_file_exists():
25    assert COST_LEVERS_MD.exists(), COST_LEVERS_MD
26
27
28def test_only_legal_source_phrase():
29    """The doc must state it is the 'only legal source' for levers."""
30    text = _cost_levers_text()
31    assert "only legal source" in text.lower()
32
33
34def test_five_levers_present():
35    """All five canonical levers must appear in the table."""
36    text = _cost_levers_text()
37    levers = [
38        "Model tier routing",
39        "Prompt caching",
40        "Batch inference",
41        "Scale-to-zero runtimes",
42        "Quota scheduling",
43    ]
44    for lever in levers:
45        assert lever in text, f"Lever '{lever}' not found in cost-levers.md"
46
47
48def test_estimate_references_cost_levers():
49    """estimate.md must reference cost-levers.md in its _knowledge frontmatter."""
50    text = _estimate_text()
51    assert "cost-levers.md" in text
52
53
54def test_estimate_mentions_drivers():
55    """estimate.md must mention the drivers[] field in its steps."""
56    text = _estimate_text()
57    assert "drivers[]" in text or "drivers[" in text