Test functools wrapped as annotate function

Author: dr-carlos

Lib/test/test_annotationlib.py

@@ -8,6 +8,7 @@
 import itertools
 import pickle
 from string.templatelib import Template, Interpolation
+import random
 import typing
 import sys
 import unittest
@@ -1648,7 +1649,6 @@ def format(format, second, /, *, third, __Format=Format,
         self.assertEqual(annotations, {"x": Format.VALUE * 5 * 6})
 
     def test_callable_cache_annotate_forwardref_value_fallback(self):
-        import random
         # If Format.STRING and Format.VALUE_WITH_FAKE_GLOBALS are not
         # supported fall back to Format.VALUE and convert to strings
         @functools.cache
@@ -1672,6 +1672,28 @@ def format(format, /, __Format=Format,
         new_anns = annotationlib.call_annotate_function(format, Format.FORWARDREF)
         self.assertEqual(annotations, new_anns)
 
+    def test_callable_wrapped_annotate_forwardref_value_fallback(self):
+        # If Format.STRING and Format.VALUE_WITH_FAKE_GLOBALS are not
+        # supported fall back to Format.VALUE and convert to strings
+        def multiple_format(fn):
+            inputs = {"x": int}
+            @functools.wraps(fn)
+            def format(format, /, __Format=Format,
+            __NotImplementedError=NotImplementedError):
+                if format == __Format.VALUE:
+                    return {**inputs, **fn()}
+                else:
+                    raise __NotImplementedError(format)
+
+            return format
+
+        annotations = annotationlib.call_annotate_function(
+            multiple_format(lambda: {"y": str}),
+            Format.FORWARDREF,
+        )
+
+        self.assertEqual(annotations, {"x": int, "y": str})
+
     def test_callable_object_annotate_string_fakeglobals(self):
         # If Format.STRING is not supported but Format.VALUE_WITH_FAKE_GLOBALS is
         # prefer that over Format.VALUE