Fix syntax_async.py for 3.11

youknowone · youknowone · commit 75f3f3c8c485 · 2023-02-20T02:06:21.000+09:00
diff --git a/extra_tests/snippets/syntax_async.py b/extra_tests/snippets/syntax_async.py
@@ -1,6 +1,8 @@
+import sys
 import asyncio
 import unittest
 
+
 class ContextManager:
     async def __aenter__(self):
         print("Entrada")
@@ -41,15 +43,18 @@ async def a(s, m):
     async for i in AIterWrap(range(0, 2)):
         print(i)
         ls.append(m)
-        await asyncio.sleep(1)
+        await asyncio.sleep(0.1)
+
+
+async def main():
+    tasks = [
+        asyncio.create_task(c)
+        for c in [a(0, "hello1"), a(0.1, "hello2"), a(0.2, "hello3"), a(0.3, "hello4")]
+    ]
+    await asyncio.wait(tasks)
 
 
-loop = asyncio.get_event_loop()
-loop.run_until_complete(
-    asyncio.wait(
-        [a(0, "hello1"), a(0.75, "hello2"), a(1.5, "hello3"), a(2.25, "hello4")]
-    )
-)
+asyncio.run(main(), debug=True)
 
 
 assert ls == [
@@ -72,41 +77,43 @@ async def a(s, m):
 ]
 
 
-class TestAsyncWith(unittest.TestCase):
-    def testAenterAttributeError1(self):
-        class LacksAenter(object):
-            async def __aexit__(self, *exc):
-                pass
+if sys.version_info < (3, 11, 0):
 
-        async def foo():
-            async with LacksAenter():
-                pass
-        
-        with self.assertRaisesRegex(AttributeError, '__aenter__'):
-            foo().send(None)
+    class TestAsyncWith(unittest.TestCase):
+        def testAenterAttributeError1(self):
+            class LacksAenter(object):
+                async def __aexit__(self, *exc):
+                    pass
 
-    def testAenterAttributeError2(self):
-        class LacksAenterAndAexit(object):
-            pass
+            async def foo():
+                async with LacksAenter():
+                    pass
 
-        async def foo():
-            async with LacksAenterAndAexit():
+            with self.assertRaisesRegex(AttributeError, "__aenter__"):
+                foo().send(None)
+
+        def testAenterAttributeError2(self):
+            class LacksAenterAndAexit(object):
                 pass
 
-        with self.assertRaisesRegex(AttributeError, '__aenter__'):
-            foo().send(None)
+            async def foo():
+                async with LacksAenterAndAexit():
+                    pass
 
-    def testAexitAttributeError(self):
-        class LacksAexit(object):
-            async def __aenter__(self):
-                pass
+            with self.assertRaisesRegex(AttributeError, "__aenter__"):
+                foo().send(None)
 
-        async def foo():
-            async with LacksAexit():
-                pass
-            
-        with self.assertRaisesRegex(AttributeError, '__aexit__'):
-            foo().send(None)
+        def testAexitAttributeError(self):
+            class LacksAexit(object):
+                async def __aenter__(self):
+                    pass
+
+            async def foo():
+                async with LacksAexit():
+                    pass
+
+            with self.assertRaisesRegex(AttributeError, "__aexit__"):
+                foo().send(None)
 
 
 if __name__ == "__main__":
