YAPF me harder

Author: smurfix

tests/interop/test_calls.py

@@ -16,31 +16,37 @@ async def async_gen_to_list(generator):
         result.append(item)
     return result
 
+
 class TrioContext:
     def __init__(self, parent):
         self.parent = parent
+
     async def __aenter__(self):
         assert self.parent.did_it == 0
         self.parent.did_it = 1
         assert sniffio.current_async_library() == "trio"
         await trio.sleep(0.01)
         self.parent.did_it = 2
         return self
+
     async def __aexit__(self, *tb):
         assert self.parent.did_it == 3
         self.parent.did_it = 4
 
+
 class AioContext:
     def __init__(self, parent, loop):
         self.parent = parent
         self.loop = loop
+
     async def __aenter__(self):
         assert self.parent.did_it == 0
         self.parent.did_it = 1
         assert sniffio.current_async_library() == "asyncio"
         await asyncio.sleep(0.01, loop=self.loop)
         self.parent.did_it = 2
         return self
+
     async def __aexit__(self, *tb):
         assert self.parent.did_it == 3
         self.parent.did_it = 4
@@ -99,6 +105,7 @@ async def _call_trio_ctx():
                 assert self.did_it == 2
                 self.did_it = 3
             assert self.did_it == 4
+
         await loop.run_asyncio(_call_trio_ctx)
 
     @pytest.mark.trio
@@ -333,12 +340,11 @@ async def cancel_soon(nursery):
     @pytest.mark.trio
     async def test_trio_asyncio_iterator(self, loop):
         async def slow_nums():
-            for n in range(1,6):
+            for n in range(1, 6):
                 asyncio.sleep(0.01, loop=loop)
                 yield n
 
-        sum=0
+        sum = 0
         async for n in loop.run_asyncio(slow_nums()):
             sum += n
         assert sum == 15
-

tests/python/test_windows_events.py

@@ -12,7 +12,7 @@
 import asyncio
 try:
     import _overlapped
-except ImportError: # py<3.7
+except ImportError:  # py<3.7
     from asyncio import _overlapped
 from .. import utils as test_utils
 from asyncio import windows_events

tests/python/test_windows_utils.py

@@ -13,7 +13,7 @@
 
 try:
     import _overlapped
-except ImportError: # py<3.7
+except ImportError:  # py<3.7
     from asyncio import _overlapped
 from asyncio import windows_utils
 try:

trio_asyncio/adapter.py

@@ -71,7 +71,7 @@ def _allow_asyncio(fn, *args):
     if isinstance(coro, asyncio.Future):
         return (yield from trio_asyncio.run_future(coro))
 
-    p,a = coro.send,None
+    p, a = coro.send, None
     while True:
         try:
             yielded = p(a)
@@ -83,9 +83,10 @@ def _allow_asyncio(fn, *args):
             else:
                 next_send = yield yielded
         except BaseException as exc:
-            p,a = coro.throw,exc
+            p, a = coro.throw, exc
         else:
-            p,a = coro.send,next_send
+            p, a = coro.send, next_send
+
 
 async def allow_asyncio(fn, *args):
     """
@@ -100,11 +101,10 @@ async def allow_asyncio(fn, *args):
     This function must be called from :mod:`trio` context.
     """
     shim = trio_asyncio.base._shim_running
-    if shim.get(): # nested call: skip
+    if shim.get():  # nested call: skip
         return await fn(*args)
     token = shim.set(True)
     try:
         return await _allow_asyncio(fn, *args)
     finally:
         shim.reset(token)
-

trio_asyncio/async_.py

@@ -128,4 +128,3 @@ def _main_loop_exit(self):
                 nursery.cancel_scope.cancel()
                 current_loop.reset(old_loop)
                 current_policy.reset(old_policy)
-

trio_asyncio/base.py

@@ -271,23 +271,31 @@ def run_asyncio(self, proc, *args):
         You can also use this for calling an asyncio-style
         async iterator or async context manager.
         """
+
         class t2aWrapper:
             def __await__(slf):
                 f = proc(*args)
                 return self.run_coroutine(f).__await__()
+
             def __aenter__(slf):
                 proc_enter = getattr(proc, "__aenter__", None)
                 if proc_enter is None or args:
-                    raise RuntimeError("Call 'run_asyncio(ctxfactory(*args))', not 'run_asyncio(ctxfactory, *args)'")
+                    raise RuntimeError(
+                        "Call 'run_asyncio(ctxfactory(*args))', not 'run_asyncio(ctxfactory, *args)'"
+                    )
                 f = proc_enter()
                 return self.run_coroutine(f)
+
             def __aexit__(slf, *tb):
                 f = proc.__aexit__(*tb)
                 return self.run_coroutine(f)
+
             def __aiter__(slf):
                 proc_iter = getattr(proc, "__anext__", None)
                 if proc_iter is None or args:
-                    raise RuntimeError("Call 'run_asyncio(gen(*args))', not 'run_asyncio(gen, *args)'")
+                    raise RuntimeError(
+                        "Call 'run_asyncio(gen(*args))', not 'run_asyncio(gen, *args)'"
+                    )
                 return run_generator(self, proc)
 
         return t2aWrapper()