add async/await functionality to Python (#57)

Author: scottrogowski

code2flow/python.py

@@ -177,7 +177,7 @@ def separate_namespaces(tree):
         nodes = []
         body = []
         for el in tree.body:
-            if type(el) == ast.FunctionDef:
+            if type(el) in (ast.FunctionDef, ast.AsyncFunctionDef):
                 nodes.append(el)
             elif type(el) == ast.ClassDef:
                 groups.append(el)

tests/test_code/py/async_basic/async_basic.py

@@ -0,0 +1,13 @@
+import asyncio
+
+class A:
+    def __init__(self):
+        pass
+    async def test(self):
+        print("hello world")
+
+async def main():
+    a = A()
+    await a.test()
+
+asyncio.run(main())

tests/testdata.py

@@ -241,6 +241,18 @@
             "expected_edges": [],
             "expected_nodes": ["file_b::(global)", "file_b::b", "file_b::c"]
         },
+        {
+            "test_name": "async_basic",
+            "directory": "async_basic",
+            "kwargs": {},
+            "expected_edges": [["async_basic::main", "async_basic::A.__init__"],
+                               ["async_basic::(global)", "async_basic::main"],
+                               ["async_basic::main", "async_basic::A.test"]],
+            "expected_nodes": ["async_basic::(global)",
+                               "async_basic::A.__init__",
+                               "async_basic::A.test",
+                               "async_basic::main"]
+        },
         {
             "test_name": "exclude_modules",
             "comment": "Correct name resolution when third-party modules are involved",