asyn.py ExitGate sleep() accepts short values.

peterhinch · peterhinch · commit ebde27eaf5de · 2017-08-09T12:11:28.000+01:00
diff --git a/asyn.py b/asyn.py
@@ -216,12 +216,17 @@ async def __aexit__(self, *args):
     # Sleep while checking for premature ending. Return True on normal ending,
     # False if premature.
     async def sleep(self, t):
-        n, rem = divmod(t * 1000, self._granularity)
-        for _ in range(n):
+        t *= 1000  # ms
+        granularity = self._granularity
+        if t <= granularity:
+            await asyncio.sleep_ms(t)
+        else:
+            n, rem = divmod(t, granularity)
+            for _ in range(n):
+                if self._ending:
+                    return False
+                await asyncio.sleep_ms(granularity)
             if self._ending:
                 return False
-            await asyncio.sleep_ms(self._granularity)
-        if self._ending:
-            return False
-        await asyncio.sleep_ms(rem)
+            await asyncio.sleep_ms(rem)
         return not self._ending
