feat: improve typing for decorated genfuncs

Author: BobTheBuidler

mypyc/test-data/irbuild-basic.test

@@ -3799,3 +3799,21 @@ L0:
     r2.__mypyc_env__ = r0; r3 = is_error
     wrapper = r2
     return wrapper
+
+[case testStaticMethodGenerator]
+from typing import Iterator
+class C:
+    @staticmethod
+    def staticgen() -> Iterator[int]:
+        yield 1
+    @classmethod
+    def classgen(cls) -> Iterator[int]:
+        yield 1
+[out]
+def C.staticgen():
+    r0 = CPyGen_Create(...)
+    return r0
+def C.classgen(cls):
+    cls :: object
+    r0 = CPyGen_Create(...)
+    return r0

mypyc/test-data/irbuild-generics.test

@@ -773,3 +773,18 @@ def f(x):
     x :: int
 L0:
     return x
+
+[case testTypePreservingDecoratorGenerator]
+from typing import Callable, TypeVar, Iterator
+
+T = TypeVar("T", bound=Callable)
+def identity_decorator(f: T) -> T:
+    return f
+
+@identity_decorator
+def gen() -> Iterator[int]:
+    yield 1
+[out]
+def gen():
+    r0 = CPyGen_Create(...)
+    return r0

mypyc/test-data/run-generators.test

@@ -907,3 +907,41 @@ def test_same_names() -> None:
         # matches the variable name in the input code, since internally it's generated
         # with a prefix.
         list(undefined())
+
+[case testStaticMethodGeneratorRuntime]
+class C:
+    @staticmethod
+    def staticgen():
+        yield 42
+
+from testutil import run_generator
+assert run_generator(C.staticgen()) == ((42,), None)
+[out]
+42
+
+[case testClassMethodGeneratorRuntime]
+class C:
+    @classmethod
+    def classgen(cls):
+        yield 99
+
+from testutil import run_generator
+assert run_generator(C.classgen()) == ((99,), None)
+[out]
+99
+
+[case testTypePreservingDecoratorGeneratorRuntime]
+from typing import Callable, TypeVar
+
+T = TypeVar("T", bound=Callable)
+def identity_decorator(f: T) -> T:
+    return f
+
+@identity_decorator
+def gen():
+    yield 7
+
+from testutil import run_generator
+assert run_generator(gen()) == ((7,), None)
+[out]
+7

mypyc/test-data/run-misc.test

@@ -58,6 +58,31 @@ f(5)
 [out]
 5
 
+[case test_macro_splice_decorator_runtime]
+def log_decorator(fn):
+    def wrapper(*args, **kwargs):
+        print("Calling function")
+        return fn(*args, **kwargs)
+    return wrapper
+
+@log_decorator
+def add(x: int, y: int) -> int:
+    return x + y
+
+@log_decorator
+def mul(x: int, y: int) -> int:
+    return x * y
+
+def test_decorator_runtime() -> None:
+    print(add(2, 3))
+    print(mul(4, 5))
+
+[out]
+Calling function
+5
+Calling function
+20
+
 [case testOptional]
 from typing import Optional