fix test after 'with' statement change

6595d8d1 · Stefan Behnel · e89ad8c8 · 6595d8d1
Commit 6595d8d1 authored Apr 23, 2011 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 5 deletions

Cython/Compiler/Tests/TestTreeFragment.py Cython/Compiler/Tests/TestTreeFragment.py +4 -5

No files found.
--- a/Cython/Compiler/Tests/TestTreeFragment.py
+++ b/Cython/Compiler/Tests/TestTreeFragment.py
@@ -54,11 +54,10 @@ class TestTreeFragments(CythonTest):
            x = TMP
        """)
        T = F.substitute(temps=[u"TMP"])
-        s = T.stats
-        self.assert_(s[0].expr.name == "__tmpvar_1")
-#        self.assert_(isinstance(s[0].expr, TempRefNode))
-#        self.assert_(isinstance(s[1].rhs, TempRefNode))
-#        self.assert_(s[0].expr.handle is s[1].rhs.handle)
+        s = T.body.stats
+        self.assert_(isinstance(s[0].expr, TempRefNode))
+        self.assert_(isinstance(s[1].rhs, TempRefNode))
+        self.assert_(s[0].expr.handle is s[1].rhs.handle)

 if __name__ == "__main__":
    import unittest