Synchronize AST import-helper guard inventory with discovered imports

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

Author: cursoragent

tests/test_ast_import_helper_usage.py

@@ -17,9 +17,6 @@ def test_ast_import_guard_modules_reuse_shared_import_helper():
     violating_modules: list[str] = []
     for module_path in AST_IMPORT_GUARD_MODULES:
         module_text = Path(module_path).read_text(encoding="utf-8")
-        if "ast_import_utils import imports_collect_function_sources" not in module_text:
-            violating_modules.append(module_path)
-            continue
         if not imports_imports_collect_function_sources(module_text):
             violating_modules.append(module_path)
             continue
@@ -30,3 +27,21 @@ def test_ast_import_guard_modules_reuse_shared_import_helper():
             violating_modules.append(module_path)
 
     assert violating_modules == []
+
+
+def test_ast_import_guard_inventory_stays_in_sync_with_helper_imports():
+    excluded_modules = {
+        "tests/test_ast_import_helper_usage.py",
+        "tests/test_ast_import_utils.py",
+    }
+    discovered_modules: list[str] = []
+    for module_path in sorted(Path("tests").glob("test_*.py")):
+        normalized_path = module_path.as_posix()
+        if normalized_path in excluded_modules:
+            continue
+        module_text = module_path.read_text(encoding="utf-8")
+        if not imports_imports_collect_function_sources(module_text):
+            continue
+        discovered_modules.append(normalized_path)
+
+    assert sorted(AST_IMPORT_GUARD_MODULES) == discovered_modules