Cover AST import-helper alias and import-style edge cases

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

Author: cursoragent

tests/test_ast_import_utils.py

@@ -21,3 +21,21 @@ def test_imports_collect_function_sources_ignores_non_matching_imports():
     )
 
     assert imports_collect_function_sources(module_text) is False
+
+
+def test_imports_collect_function_sources_supports_aliased_import():
+    module_text = (
+        "from tests.ast_function_source_utils import collect_function_sources as cfs\n"
+        "cfs('tests/test_model_request_wrapper_internal_reuse.py')\n"
+    )
+
+    assert imports_collect_function_sources(module_text) is True
+
+
+def test_imports_collect_function_sources_ignores_non_from_imports():
+    module_text = (
+        "import tests.ast_function_source_utils as source_utils\n"
+        "source_utils.collect_function_sources('tests/test_web_request_wrapper_internal_reuse.py')\n"
+    )
+
+    assert imports_collect_function_sources(module_text) is False