From e0c128beec6d6cf88f4e0299d5fb0774434b6b2a Mon Sep 17 00:00:00 2001
From: Benjamin Peterson <benjamin@python.org>
Date: Thu, 19 May 2011 21:49:37 -0500
Subject: [PATCH] unconditionally override lineno and col_offset on generated
 ast

---
 _pytest/assertrewrite.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/_pytest/assertrewrite.py b/_pytest/assertrewrite.py
index 6b978bff8..ed9e0ca55 100644
--- a/_pytest/assertrewrite.py
+++ b/_pytest/assertrewrite.py
@@ -67,6 +67,19 @@ binop_map = {
 }
 
 
+def set_location(node, lineno, col_offset):
+    """Set node location information recursively."""
+    def _fix(node, lineno, col_offset):
+        if "lineno" in node._attributes:
+            node.lineno = lineno
+        if "col_offset" in node._attributes:
+            node.col_offset = col_offset
+        for child in ast.iter_child_nodes(node):
+            _fix(child, lineno, col_offset)
+    _fix(node, lineno, col_offset)
+    return node
+
+
 class AssertionRewriter(ast.NodeVisitor):
 
     def run(self, mod):
@@ -196,9 +209,7 @@ class AssertionRewriter(ast.NodeVisitor):
             self.statements.append(delete)
         # Fix line numbers.
         for stmt in self.statements:
-            stmt.lineno = assert_.lineno
-            stmt.col_offset = assert_.col_offset
-            ast.fix_missing_locations(stmt)
+            set_location(stmt, assert_.lineno, assert_.col_offset)
         return self.statements
 
     def visit_Name(self, name):