ensure that arguments and exceptions are included

kavigupta · Feb 13, 2024 · 3454d21 · 3454d21
1 parent b700f5e
commit 3454d21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/utils.py b/tests/utils.py
@@ -83,6 +83,8 @@ def all_nodes_gen_for_variables(variables):
     yield from variables.functions
     yield from variables.classes
     yield from variables.import_statements
+    yield from variables.arguments
+    yield from variables.exceptions
 
 def all_nodes_gen_for_scope(scope):
     if hasattr(scope, 'children'):