1 files changed, 8 insertions, 1 deletions
diff --git a/tests/roots/test-ext-autodoc/target/wrappedfunction.py b/tests/roots/test-ext-autodoc/target/wrappedfunction.py
index ea872f086..0bd2d2069 100644
--- a/tests/roots/test-ext-autodoc/target/wrappedfunction.py
+++ b/tests/roots/test-ext-autodoc/target/wrappedfunction.py
@@ -1,8 +1,15 @@
-# for py32 or above
+from contextlib import contextmanager
 from functools import lru_cache
+from typing import Generator
 
 
 @lru_cache(maxsize=None)
 def slow_function(message, timeout):
     """This function is slow."""
     print(message)
+
+
+@contextmanager
+def feeling_good(x: int, y: int) -> Generator:
+    """You'll feel better in this context!"""
+    yield