Implemented new checker await-outside-async (#4514)

1 files changed, 30 insertions, 0 deletions

diff --git a/tests/functional/a/await_outside_async.py b/tests/functional/a/await_outside_async.py
new file mode 100644
index 000000000..2bc126761
--- /dev/null
+++ b/tests/functional/a/await_outside_async.py
@@ -0,0 +1,30 @@
+# pylint: disable=missing-docstring,unused-variable
+import asyncio
+
+async def nested():
+    return 42
+
+async def main():
+    nested()
+    print(await nested())  # This is okay
+
+def not_async():
+    print(await nested())  # [await-outside-async]
+
+
+async def func(i):
+    return i**2
+
+async def okay_function():
+    var = [await func(i) for i in range(5)]  # This should be okay
+
+
+# Test nested functions
+async def func2():
+    def inner_func():
+        await asyncio.sleep(1)  # [await-outside-async]
+
+
+def outer_func():
+    async def inner_func():
+        await asyncio.sleep(1)