Add AST import helper contract test and boundary allowance

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

Author: cursoragent

CONTRIBUTING.md

@@ -93,6 +93,7 @@ This runs lint, format checks, compile checks, tests, and package build.
   - `tests/test_ast_function_source_utils.py` (shared AST function-source helper contract validation),
   - `tests/test_ast_import_helper_import_boundary.py` (shared 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_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/test_architecture_marker_usage.py

@@ -22,6 +22,7 @@
     "tests/test_ast_function_source_utils.py",
     "tests/test_ast_import_helper_import_boundary.py",
     "tests/test_ast_import_helper_usage.py",
+    "tests/test_ast_import_utils.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_helper_import_boundary.py

@@ -7,6 +7,8 @@
 
 pytestmark = pytest.mark.architecture
 
+EXPECTED_EXTRA_IMPORTER_MODULES = ("tests/test_ast_import_utils.py",)
+
 
 def _imports_ast_import_helper(module_text: str) -> bool:
     module_ast = ast.parse(module_text)
@@ -28,4 +30,5 @@ def test_ast_import_helper_imports_are_centralized():
             continue
         discovered_modules.append(module_path.as_posix())
 
-    assert discovered_modules == sorted(AST_IMPORT_GUARD_MODULES)
+    expected_modules = sorted([*AST_IMPORT_GUARD_MODULES, *EXPECTED_EXTRA_IMPORTER_MODULES])
+    assert discovered_modules == expected_modules

tests/test_ast_import_utils.py

@@ -0,0 +1,23 @@
+import pytest
+
+from tests.ast_import_utils import imports_collect_function_sources
+
+pytestmark = pytest.mark.architecture
+
+
+def test_imports_collect_function_sources_detects_expected_import():
+    module_text = (
+        "from tests.ast_function_source_utils import collect_function_sources\n"
+        "collect_function_sources('tests/test_job_request_wrapper_internal_reuse.py')\n"
+    )
+
+    assert imports_collect_function_sources(module_text) is True
+
+
+def test_imports_collect_function_sources_ignores_non_matching_imports():
+    module_text = (
+        "from tests.ast_function_source_utils import something_else\n"
+        "from tests.other_helper import collect_function_sources\n"
+    )
+
+    assert imports_collect_function_sources(module_text) is False