Merge 02ed5ce into 0a6aff7

rubik · Dec 30, 2020 · 6ac2978 · 6ac2978
2 parents 0a6aff7 + 02ed5ce
commit 6ac2978
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/radon/tests/test_halstead.py b/radon/tests/test_halstead.py
@@ -74,6 +74,18 @@ def f():
      ''',
         (2, 4, 2, 4),
     ),
+    (
+        '''
+     a = 2
+     b = 3
+     a *= b
+
+     async def f():
+         b = 2
+         b += 4
+     ''',
+        (2, 4, 2, 4),
+    ),
     (
         '''
      a = b < 4

diff --git a/radon/visitors.py b/radon/visitors.py
@@ -445,3 +445,9 @@ def visit_FunctionDef(self, node):
 
         # Save the visited function visitor for later reference.
         self.function_visitors.append(func_visitor)
+
+    def visit_AsyncFunctionDef(self, node):
+        '''Async functions are similar to standard functions, so treat them as
+        such.
+        '''
+        self.visit_FunctionDef(node)