Add AST import-utils module import boundary guard

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

CONTRIBUTING.md

@@ -96,6 +96,7 @@ This runs lint, format checks, compile checks, tests, and package build.
   - `tests/test_ast_import_helper_secondary_import_boundary.py` (secondary AST import-helper import boundary enforcement across test modules),
   - `tests/test_ast_import_helper_usage.py` (shared AST import-helper usage enforcement across AST import-boundary guard suites),
   - `tests/test_ast_import_utils.py` (shared AST import-helper contract validation),
+  - `tests/test_ast_import_utils_module_import_boundary.py` (shared AST import-helper module import boundary enforcement across test modules),
   - `tests/test_binary_file_open_helper_usage.py` (shared binary file open helper usage enforcement),
   - `tests/test_browser_use_payload_helper_usage.py` (browser-use payload helper usage enforcement),
   - `tests/test_ci_workflow_quality_gates.py` (CI guard-stage + make-target enforcement),

tests/ast_import_utils.py

@@ -1,6 +1,18 @@
 import ast
 
 
+def imports_from_module(module_text: str, module: str) -> bool:
+    module_ast = ast.parse(module_text)
+    for node in module_ast.body:
+        if isinstance(node, ast.ImportFrom) and node.module == module:
+            return True
+        if not isinstance(node, ast.Import):
+            continue
+        if any(alias.name == module for alias in node.names):
+            return True
+    return False
+
+
 def imports_symbol_from_module(module_text: str, module: str, symbol: str) -> bool:
     module_ast = ast.parse(module_text)
     for node in module_ast.body:

tests/test_architecture_marker_usage.py

@@ -25,6 +25,7 @@
     "tests/test_ast_import_helper_secondary_import_boundary.py",
     "tests/test_ast_import_helper_usage.py",
     "tests/test_ast_import_utils.py",
+    "tests/test_ast_import_utils_module_import_boundary.py",
     "tests/test_guardrail_ast_utils.py",
     "tests/test_helper_transport_usage_boundary.py",
     "tests/test_manager_model_dump_usage.py",

tests/test_ast_import_utils.py

@@ -2,9 +2,10 @@
 
 from tests.ast_import_utils import (
     calls_symbol,
-    imports_symbol_from_module,
     imports_collect_function_sources,
+    imports_from_module,
     imports_imports_collect_function_sources,
+    imports_symbol_from_module,
 )
 
 pytestmark = pytest.mark.architecture
@@ -28,6 +29,33 @@ def test_imports_collect_function_sources_ignores_non_matching_imports():
     assert imports_collect_function_sources(module_text) is False
 
 
+def test_imports_from_module_detects_expected_from_import():
+    module_text = (
+        "from tests.ast_import_utils import imports_collect_function_sources\n"
+        "imports_collect_function_sources('dummy')\n"
+    )
+
+    assert imports_from_module(module_text, module="tests.ast_import_utils") is True
+
+
+def test_imports_from_module_detects_expected_direct_import():
+    module_text = (
+        "import tests.ast_import_utils as import_utils\n"
+        "import_utils.imports_collect_function_sources('dummy')\n"
+    )
+
+    assert imports_from_module(module_text, module="tests.ast_import_utils") is True
+
+
+def test_imports_from_module_ignores_unrelated_module_imports():
+    module_text = (
+        "from tests.ast_function_source_utils import collect_function_sources\n"
+        "collect_function_sources('dummy')\n"
+    )
+
+    assert imports_from_module(module_text, module="tests.ast_import_utils") is False
+
+
 def test_imports_symbol_from_module_detects_expected_symbol():
     module_text = (
         "from tests.ast_import_utils import imports_collect_function_sources\n"

tests/test_ast_import_utils_module_import_boundary.py

@@ -0,0 +1,30 @@
+from pathlib import Path
+
+import pytest
+
+from tests.ast_import_utils import imports_from_module
+
+pytestmark = pytest.mark.architecture
+
+
+EXPECTED_AST_IMPORT_UTILS_IMPORTERS = (
+    "tests/test_ast_call_symbol_helper_import_boundary.py",
+    "tests/test_ast_function_source_helper_usage.py",
+    "tests/test_ast_function_source_import_boundary.py",
+    "tests/test_ast_import_helper_import_boundary.py",
+    "tests/test_ast_import_helper_secondary_import_boundary.py",
+    "tests/test_ast_import_helper_usage.py",
+    "tests/test_ast_import_utils.py",
+    "tests/test_ast_import_utils_module_import_boundary.py",
+)
+
+
+def test_ast_import_utils_imports_are_centralized():
+    discovered_modules: list[str] = []
+    for module_path in sorted(Path("tests").glob("test_*.py")):
+        module_text = module_path.read_text(encoding="utf-8")
+        if not imports_from_module(module_text, module="tests.ast_import_utils"):
+            continue
+        discovered_modules.append(module_path.as_posix())
+
+    assert discovered_modules == list(EXPECTED_AST_IMPORT_UTILS_IMPORTERS)