tests/test_docs_contract.py

import re
import unittest
from pathlib import Path


ROOT = Path(__file__).resolve().parents[1]


class DocsContractTest(unittest.TestCase):
    def test_agents_required_reading_paths_exist(self):
        agents = (ROOT / "AGENTS.md").read_text(encoding="utf-8")
        paths = re.findall(r"`([^`]+\.md)`", agents)
        required_paths = [path for path in paths if path == "AGENTS.md" or path.startswith("initial_docs/")]
        self.assertGreaterEqual(len(required_paths), 7)
        for path in required_paths:
            self.assertTrue((ROOT / path).exists(), path)

    def test_top_level_agents_and_initial_docs_layout(self):
        self.assertTrue((ROOT / "AGENTS.md").is_file())
        self.assertTrue((ROOT / "initial_docs" / "docs" / "ARCHITECTURE.md").is_file())
        self.assertTrue((ROOT / "initial_docs" / "docs" / "STAGES.md").is_file())


if __name__ == "__main__":
    unittest.main()