Slightly updated async generator tests for readability

Author: Sylvain MARIE

tests/_test_py36.py

@@ -1,21 +1,17 @@
-from makefun import wraps, partial
-
-
 def make_async_generator():
+    """Returns a new async generator function to use in tests."""
+
     async def f(v):
         yield v
 
     return f
 
 
-def make_async_generator_wrapper(f):
-    @wraps(f)
+def make_async_generator_wrapper(async_gen_f):
+    """Returns a new async generator function wrapping `f`, to use in tests."""
+
     async def wrapper(*args, **kwargs):
-        async for v in f(*args, **kwargs):
+        async for v in async_gen_f(*args, **kwargs):
             yield v
 
     return wrapper
-
-
-def make_async_generator_partial(f, *args, **kwargs):
-    return partial(f, *args, **kwargs)

tests/test_issues.py

@@ -233,7 +233,7 @@ def test_issue_77_async_generator_wraps():
     from ._test_py36 import make_async_generator, make_async_generator_wrapper
 
     f = make_async_generator()
-    wrapper = make_async_generator_wrapper(f)
+    wrapper = wraps(f)(make_async_generator_wrapper(f))
 
     assert inspect.isasyncgenfunction(f)
     assert inspect.isasyncgenfunction(wrapper)
@@ -244,10 +244,10 @@ def test_issue_77_async_generator_wraps():
 @pytest.mark.skipif(sys.version_info < (3, 6), reason="requires python 3.6 or higher (async generator)")
 def test_issue_77_async_generator_partial():
     import asyncio
-    from ._test_py36 import make_async_generator, make_async_generator_partial
+    from ._test_py36 import make_async_generator
 
     f = make_async_generator()
-    f_partial = make_async_generator_partial(f, v=1)
+    f_partial = partial(f, v=1)
 
     assert inspect.isasyncgenfunction(f)
     assert inspect.isasyncgenfunction(f_partial)