[pypy-svn] r66253 - pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test

[benjamin at codespeak.net]

Author: benjamin
Date: Wed Jul 15 21:43:41 2009
New Revision: 66253

Modified:
   pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_astbuilder.py
   pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_symtable.py
Log:
run these tests with the 'with' statement enabled

Modified: pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_astbuilder.py
==============================================================================
--- pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_astbuilder.py	(original)
+++ pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_astbuilder.py	Wed Jul 15 21:43:41 2009
@@ -7,7 +7,7 @@
 from pypy.interpreter.pyparser import pyparse
 from pypy.interpreter.pyparser.error import SyntaxError
 from pypy.interpreter.astcompiler.astbuilder import ast_from_node
-from pypy.interpreter.astcompiler import ast2 as ast
+from pypy.interpreter.astcompiler import ast2 as ast, consts
 
 
 class TestAstBuilder:
@@ -16,7 +16,8 @@
         cls.parser = pyparse.PythonParser(cls.space)
 
     def get_ast(self, source, p_mode="exec"):
-        info = pyparse.CompileInfo("<test>", p_mode)
+        info = pyparse.CompileInfo("<test>", p_mode,
+                                   consts.CO_FUTURE_WITH_STATEMENT)
         tree = self.parser.parse_source(source, info)
         ast_node = ast_from_node(self.space, tree, info)
         return ast_node

Modified: pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_symtable.py
==============================================================================
--- pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_symtable.py	(original)
+++ pypy/branch/parser-compiler/pypy/interpreter/astcompiler/test/test_symtable.py	Wed Jul 15 21:43:41 2009
@@ -1,6 +1,7 @@
 import string
 import py
-from pypy.interpreter.astcompiler import ast2 as ast, astbuilder, symtable
+from pypy.interpreter.astcompiler import (ast2 as ast, astbuilder, symtable,
+                                          consts)
 from pypy.interpreter.pyparser import pyparse
 from pypy.interpreter.pyparser.error import SyntaxError
 
@@ -11,7 +12,8 @@
         cls.parser = pyparse.PythonParser(cls.space)
 
     def mod_scope(self, source, mode="exec"):
-        info = pyparse.CompileInfo("<test>", mode)
+        info = pyparse.CompileInfo("<test>", mode,
+                                   consts.CO_FUTURE_WITH_STATEMENT)
         tree = self.parser.parse_source(source, info)
         module = astbuilder.ast_from_node(self.space, tree, info)
         builder = symtable.SymtableBuilder(self.space, module, info)