Fix parsing crashes when tests include with statements

fixes nestorsalceda#104
pinerylabs · Jan 3, 2020 · f620fac · f620fac
1 parent 0a2c83e
commit f620fac
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/mamba/nodetransformers.py b/mamba/nodetransformers.py
@@ -64,11 +64,15 @@ def _get_name(self, node):
 
         if isinstance(context_expr, ast.Call):
             if hasattr(context_expr.func, 'value'):
-                return context_expr.func.value.id
-            return context_expr.func.id
+                if hasattr(context_expr.func.value, 'id'):
+                    return context_expr.func.value.id
+
+            if hasattr(context_expr.func, 'id'):
+                return context_expr.func.id
 
         if isinstance(context_expr, ast.Attribute):
-            return context_expr.value.id
+            if hasattr(context_expr.value, 'id'):
+                return context_expr.value.id
 
     def _context_expr_for(self, node):
         return node.context_expr