Fixed NameError in case of unknown symbols in type hints. Fixes #37

Author: Sylvain MARIE

makefun/main.py

@@ -310,6 +310,8 @@ def _signature_symbol_needs_protection(symbol, evaldict):
         try:
             deflt = eval(repr(symbol), evaldict)
             needs_protection = deflt != symbol
+        except NameError:
+            needs_protection = True
         except SyntaxError:
             needs_protection = True
     else:

makefun/tests/_test_py35.py

@@ -22,3 +22,13 @@ def ref(a: A) -> A:
         pass
 
     return ref
+
+
+def make_ref_function2():
+    """ """
+    from typing import Any
+
+    def ref(a: Any):
+        pass
+
+    return ref

makefun/tests/test_advanced.py

@@ -141,3 +141,17 @@ def foo(a):
         return a
 
     assert foo(10) == 10
+
+
+@pytest.mark.skipif(sys.version_info < (3, 5), reason="requires python 3.5 or higher (non-comment type hints)")
+def test_type_hint_error2():
+    """ Test for https://github.com/smarie/python-makefun/issues/32 """
+
+    from makefun.tests._test_py35 import make_ref_function2
+    ref_f = make_ref_function2()
+
+    @wraps(ref_f)
+    def foo(a):
+        return a
+
+    assert foo(10) == 10