summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_ast.py
diff options
context:
space:
mode:
authorGeorg Brandl <georg@python.org>2008-03-30 07:09:22 (GMT)
committerGeorg Brandl <georg@python.org>2008-03-30 07:09:22 (GMT)
commitebc8dedd1996db68c889c2cdf2701fd7eadc501f (patch)
treecb6b11ec24f080daad11632bbbda7d45641c993c /Lib/test/test_ast.py
parentc52ed59473cfbff9d304fcd6eda0e99bb8098e37 (diff)
downloadcpython-ebc8dedd1996db68c889c2cdf2701fd7eadc501f.zip
cpython-ebc8dedd1996db68c889c2cdf2701fd7eadc501f.tar.gz
cpython-ebc8dedd1996db68c889c2cdf2701fd7eadc501f.tar.bz2
Convert test_ast to unittest and add a test for r62049.
Diffstat (limited to 'Lib/test/test_ast.py')
-rw-r--r--Lib/test/test_ast.py95
1 files changed, 61 insertions, 34 deletions
diff --git a/Lib/test/test_ast.py b/Lib/test/test_ast.py
index 9dd27ea..4184bc6 100644
--- a/Lib/test/test_ast.py
+++ b/Lib/test/test_ast.py
@@ -1,4 +1,5 @@
-import sys, itertools
+import sys, itertools, unittest
+from test import test_support
import _ast
def to_tuple(t):
@@ -15,6 +16,7 @@ def to_tuple(t):
result.append(to_tuple(getattr(t, f)))
return tuple(result)
+
# These tests are compiled through "exec"
# There should be atleast one test per statement
exec_tests = [
@@ -118,41 +120,66 @@ eval_tests = [
# TODO: expr_context, slice, boolop, operator, unaryop, cmpop, comprehension
# excepthandler, arguments, keywords, alias
-if __name__=='__main__' and sys.argv[1:] == ['-g']:
- for statements, kind in ((exec_tests, "exec"), (single_tests, "single"),
- (eval_tests, "eval")):
- print kind+"_results = ["
- for s in statements:
- print repr(to_tuple(compile(s, "?", kind, 0x400)))+","
- print "]"
- print "run_tests()"
- raise SystemExit
+class AST_Tests(unittest.TestCase):
+
+ def _assert_order(self, ast_node, parent_pos):
+ if not isinstance(ast_node, _ast.AST) or ast_node._fields is None:
+ return
+ if isinstance(ast_node, (_ast.expr, _ast.stmt, _ast.excepthandler)):
+ node_pos = (ast_node.lineno, ast_node.col_offset)
+ self.assert_(node_pos >= parent_pos)
+ parent_pos = (ast_node.lineno, ast_node.col_offset)
+ for name in ast_node._fields:
+ value = getattr(ast_node, name)
+ if isinstance(value, list):
+ for child in value:
+ self._assert_order(child, parent_pos)
+ elif value is not None:
+ self._assert_order(value, parent_pos)
+
+ def test_snippets(self):
+ for input, output, kind in ((exec_tests, exec_results, "exec"),
+ (single_tests, single_results, "single"),
+ (eval_tests, eval_results, "eval")):
+ for i, o in itertools.izip(input, output):
+ ast_tree = compile(i, "?", kind, _ast.PyCF_ONLY_AST)
+ self.assertEquals(to_tuple(ast_tree), o)
+ self._assert_order(ast_tree, (0, 0))
+
+ def test_nodeclasses(self):
+ x = _ast.BinOp(1, 2, 3, lineno=0)
+ self.assertEquals(x.left, 1)
+ self.assertEquals(x.op, 2)
+ self.assertEquals(x.right, 3)
+ self.assertEquals(x.lineno, 0)
+
+ # node raises exception when not given enough arguments
+ self.assertRaises(TypeError, _ast.BinOp, 1, 2)
+
+ # can set attributes through kwargs too
+ x = _ast.BinOp(left=1, op=2, right=3, lineno=0)
+ self.assertEquals(x.left, 1)
+ self.assertEquals(x.op, 2)
+ self.assertEquals(x.right, 3)
+ self.assertEquals(x.lineno, 0)
+
-def test_order(ast_node, parent_pos):
+def test_main():
+ test_support.run_unittest(AST_Tests)
- if not isinstance(ast_node, _ast.AST) or ast_node._fields is None:
+def main():
+ if __name__ != '__main__':
return
- if isinstance(ast_node, (_ast.expr, _ast.stmt, _ast.excepthandler)):
- node_pos = (ast_node.lineno, ast_node.col_offset)
- assert node_pos >= parent_pos, (node_pos, parent_pos)
- parent_pos = (ast_node.lineno, ast_node.col_offset)
- for name in ast_node._fields:
- value = getattr(ast_node, name)
- if isinstance(value, list):
- for child in value:
- test_order(child, parent_pos)
- elif value is not None:
- test_order(value, parent_pos)
-
-def run_tests():
- for input, output, kind in ((exec_tests, exec_results, "exec"),
- (single_tests, single_results, "single"),
- (eval_tests, eval_results, "eval")):
- for i, o in itertools.izip(input, output):
- ast_tree = compile(i, "?", kind, 0x400)
- print to_tuple(ast_tree), o
- assert to_tuple(ast_tree) == o
- test_order(ast_tree, (0, 0))
+ if sys.argv[1:] == ['-g']:
+ for statements, kind in ((exec_tests, "exec"), (single_tests, "single"),
+ (eval_tests, "eval")):
+ print kind+"_results = ["
+ for s in statements:
+ print repr(to_tuple(compile(s, "?", kind, 0x400)))+","
+ print "]"
+ print "main()"
+ raise SystemExit
+ test_main()
#### EVERYTHING BELOW IS GENERATED #####
exec_results = [
@@ -202,4 +229,4 @@ eval_results = [
('Expression', ('Tuple', (1, 0), [('Num', (1, 0), 1), ('Num', (1, 2), 2), ('Num', (1, 4), 3)], ('Load',))),
('Expression', ('Call', (1, 0), ('Attribute', (1, 0), ('Attribute', (1, 0), ('Attribute', (1, 0), ('Name', (1, 0), 'a', ('Load',)), 'b', ('Load',)), 'c', ('Load',)), 'd', ('Load',)), [('Subscript', (1, 8), ('Attribute', (1, 8), ('Name', (1, 8), 'a', ('Load',)), 'b', ('Load',)), ('Slice', ('Num', (1, 12), 1), ('Num', (1, 14), 2), None), ('Load',))], [], None, None)),
]
-run_tests()
+main()