fix test after 'with' statement change

62682701 · Stefan Behnel · c147fe6d · 62682701
Commit 62682701 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