feat: add globals if available

src/flowrep/parsers/object_scope.py

@@ -94,7 +94,8 @@ def get_scope(func: Callable[..., Any] | type[Any]) -> ScopeProxy:
             "inspect.getmodule() returned None and no resolvable __module__ "
             "attribute was found."
         )
-    return ScopeProxy(module.__dict__ | vars(builtins))
+    scope = module.__dict__ | getattr(func, "__globals__", {})
+    return ScopeProxy(scope | vars(builtins))
 
 
 def resolve_attribute_to_object(attribute: str, scope: ScopeProxy | object) -> object:

tests/unit/parsers/test_object_scope.py

@@ -136,6 +136,24 @@ def test_returns_module_globals(self):
         self.assertIs(scope.Outer, Outer)
         self.assertIs(scope.object_scope, object_scope)
 
+    def test_prefers_function_globals_when_module_differs(self):
+        """Resolve names from a function's globals when its module is unrelated."""
+        module = types.ModuleType("_test_unrelated_mod")
+        sys.modules[module.__name__] = module
+        marker = object()
+        try:
+            func = types.FunctionType(
+                (lambda: None).__code__,
+                {"marker": marker},
+                "_test_func",
+            )
+            func.__module__ = module.__name__
+            with patch.object(object_scope.inspect, "getmodule", return_value=module):
+                scope = object_scope.get_scope(func)
+            self.assertIs(scope.marker, marker)
+        finally:
+            del sys.modules[module.__name__]
+
     def test_includes_builtins(self):
         scope = object_scope.get_scope(add)
         self.assertIs(scope.len, len)