annotate Lib/test/test_symtable.py @ 58925:eecf1bef052d 2.6

Merged revisions 78207 via svnmerge from svn+ssh://pythondev@svn.python.org/python/trunk ........ r78207 | ezio.melotti | 2010-02-17 01:26:09 +0200 (Wed, 17 Feb 2010) | 1 line #7930: fix stripid ........
author Ezio Melotti <ezio.melotti@gmail.com>
date Tue, 16 Feb 2010 23:29:44 +0000
parents 6910792fcd4b
children 762d89faceaa
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
1 """
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
2 Test the API of the symtable module.
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
3 """
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
4 import symtable
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
5 import unittest
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
6 import warnings
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
7
36380
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
8 from test import test_support
16422
8b318b8d78c9 Add minimal interface to symtable: _symtable module.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
9
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
10
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
11 TEST_CODE = """
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
12 import sys
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
13
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
14 glob = 42
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
15
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
16 class Mine:
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
17 instance_var = 24
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
18 def a_method(p1, p2):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
19 pass
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
20
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
21 def spam(a, b, *var, **kw):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
22 global bar
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
23 bar = 47
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
24 x = 23
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
25 glob
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
26 def internal():
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
27 return x
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
28 return internal
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
29
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
30 def foo():
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
31 exec 'm'
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
32 from sys import *
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
33
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
34 def namespace_test(): pass
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
35 def namespace_test(): pass
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
36 """
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
37
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
38
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
39 def find_block(block, name):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
40 for ch in block.get_children():
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
41 if ch.get_name() == name:
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
42 return ch
16422
8b318b8d78c9 Add minimal interface to symtable: _symtable module.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
43
8b318b8d78c9 Add minimal interface to symtable: _symtable module.
Jeremy Hylton <jeremy@alum.mit.edu>
parents:
diff changeset
44
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
45 class SymtableTest(unittest.TestCase):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
46
49099
ff96a313cd1c warnings.catch_warnings() now returns a list or None instead of the custom
Brett Cannon <bcannon@gmail.com>
parents: 48315
diff changeset
47 with warnings.catch_warnings():
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
48 # Ignore warnings about "from blank import *"
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
49 warnings.simplefilter("ignore", SyntaxWarning)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
50 top = symtable.symtable(TEST_CODE, "?", "exec")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
51 # These correspond to scopes in TEST_CODE
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
52 Mine = find_block(top, "Mine")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
53 a_method = find_block(Mine, "a_method")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
54 spam = find_block(top, "spam")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
55 internal = find_block(spam, "internal")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
56 foo = find_block(top, "foo")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
57
48315
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
58 def test_noops(self):
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
59 # Check methods that don't work. They should warn and return False.
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
60 def check(w, msg):
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
61 self.assertEqual(str(w.message), msg)
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
62 sym = self.top.lookup("glob")
49146
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
63 with test_support.check_warnings() as w:
48315
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
64 warnings.simplefilter("always", DeprecationWarning)
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
65 self.assertFalse(sym.is_vararg())
49146
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
66 check(w, "is_vararg() is obsolete and will be removed")
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
67 w.reset()
48315
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
68 self.assertFalse(sym.is_keywordarg())
49146
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
69 check(w, "is_keywordarg() is obsolete and will be removed")
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
70 w.reset()
48315
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
71 self.assertFalse(sym.is_in_tuple())
49146
f28b999a69f8 Issue #3781: Final cleanup of warnings.catch_warnings and its usage in the test suite. Closes issue w.r.t. 2.6 (R: Brett Cannon)
Nick Coghlan <ncoghlan@gmail.com>
parents: 49099
diff changeset
72 check(w, "is_in_tuple() is obsolete and will be removed")
48315
abde4db307a5 deprecate some useless, noop methods in symtable
Benjamin Peterson <benjamin@python.org>
parents: 48269
diff changeset
73
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
74 def test_type(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
75 self.assertEqual(self.top.get_type(), "module")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
76 self.assertEqual(self.Mine.get_type(), "class")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
77 self.assertEqual(self.a_method.get_type(), "function")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
78 self.assertEqual(self.spam.get_type(), "function")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
79 self.assertEqual(self.internal.get_type(), "function")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
80
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
81 def test_optimized(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
82 self.assertFalse(self.top.is_optimized())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
83 self.assertFalse(self.top.has_exec())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
84 self.assertFalse(self.top.has_import_star())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
85
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
86 self.assertTrue(self.spam.is_optimized())
35738
e018fce14f10 Disable some tests in anticipation of merging ast-branch to the head
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 29870
diff changeset
87
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
88 self.assertFalse(self.foo.is_optimized())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
89 self.assertTrue(self.foo.has_exec())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
90 self.assertTrue(self.foo.has_import_star())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
91
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
92 def test_nested(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
93 self.assertFalse(self.top.is_nested())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
94 self.assertFalse(self.Mine.is_nested())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
95 self.assertFalse(self.spam.is_nested())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
96 self.assertTrue(self.internal.is_nested())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
97
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
98 def test_children(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
99 self.assertTrue(self.top.has_children())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
100 self.assertTrue(self.Mine.has_children())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
101 self.assertFalse(self.foo.has_children())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
102
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
103 def test_lineno(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
104 self.assertEqual(self.top.get_lineno(), 0)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
105 self.assertEqual(self.spam.get_lineno(), 11)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
106
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
107 def test_function_info(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
108 func = self.spam
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
109 self.assertEqual(func.get_parameters(), ("a", "b", "kw", "var"))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
110 self.assertEqual(func.get_locals(),
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
111 ("a", "b", "bar", "internal", "kw", "var", "x"))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
112 self.assertEqual(func.get_globals(), ("bar", "glob"))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
113 self.assertEqual(self.internal.get_frees(), ("x",))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
114
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
115 def test_globals(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
116 self.assertTrue(self.spam.lookup("glob").is_global())
52520
6910792fcd4b Merged revisions 70801,70809 via svnmerge from
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 49146
diff changeset
117 self.assertFalse(self.spam.lookup("glob").is_declared_global())
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
118 self.assertTrue(self.spam.lookup("bar").is_global())
52520
6910792fcd4b Merged revisions 70801,70809 via svnmerge from
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 49146
diff changeset
119 self.assertTrue(self.spam.lookup("bar").is_declared_global())
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
120 self.assertFalse(self.internal.lookup("x").is_global())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
121 self.assertFalse(self.Mine.lookup("instance_var").is_global())
29870
030227de8051 SF patch 763201: handling of SyntaxErrors in symbol table build
Jeremy Hylton <jeremy@alum.mit.edu>
parents: 24364
diff changeset
122
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
123 def test_local(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
124 self.assertTrue(self.spam.lookup("x").is_local())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
125 self.assertFalse(self.internal.lookup("x").is_local())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
126
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
127 def test_referenced(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
128 self.assertTrue(self.internal.lookup("x").is_referenced())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
129 self.assertTrue(self.spam.lookup("internal").is_referenced())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
130 self.assertFalse(self.spam.lookup("x").is_referenced())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
131
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
132 def test_parameters(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
133 for sym in ("a", "var", "kw"):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
134 self.assertTrue(self.spam.lookup(sym).is_parameter())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
135 self.assertFalse(self.spam.lookup("x").is_parameter())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
136
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
137 def test_symbol_lookup(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
138 self.assertEqual(len(self.top.get_identifiers()),
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
139 len(self.top.get_symbols()))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
140
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
141 self.assertRaises(KeyError, self.top.lookup, "not_here")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
142
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
143 def test_namespaces(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
144 self.assertTrue(self.top.lookup("Mine").is_namespace())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
145 self.assertTrue(self.Mine.lookup("a_method").is_namespace())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
146 self.assertTrue(self.top.lookup("spam").is_namespace())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
147 self.assertTrue(self.spam.lookup("internal").is_namespace())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
148 self.assertTrue(self.top.lookup("namespace_test").is_namespace())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
149 self.assertFalse(self.spam.lookup("x").is_namespace())
36380
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
150
48269
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
151 self.assert_(self.top.lookup("spam").get_namespace() is self.spam)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
152 ns_test = self.top.lookup("namespace_test")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
153 self.assertEqual(len(ns_test.get_namespaces()), 2)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
154 self.assertRaises(ValueError, ns_test.get_namespace)
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
155
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
156 def test_assigned(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
157 self.assertTrue(self.spam.lookup("x").is_assigned())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
158 self.assertTrue(self.spam.lookup("bar").is_assigned())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
159 self.assertTrue(self.top.lookup("spam").is_assigned())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
160 self.assertTrue(self.Mine.lookup("a_method").is_assigned())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
161 self.assertFalse(self.internal.lookup("x").is_assigned())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
162
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
163 def test_imported(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
164 self.assertTrue(self.top.lookup("sys").is_imported())
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
165
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
166 def test_name(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
167 self.assertEqual(self.top.get_name(), "top")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
168 self.assertEqual(self.spam.get_name(), "spam")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
169 self.assertEqual(self.spam.lookup("x").get_name(), "x")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
170 self.assertEqual(self.Mine.get_name(), "Mine")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
171
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
172 def test_class_info(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
173 self.assertEqual(self.Mine.get_methods(), ('a_method',))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
174
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
175 def test_filename_correct(self):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
176 ### Bug tickler: SyntaxError file name correct whether error raised
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
177 ### while parsing or building symbol table.
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
178 def checkfilename(brokencode):
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
179 try:
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
180 symtable.symtable(brokencode, "spam", "exec")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
181 except SyntaxError as e:
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
182 self.assertEqual(e.filename, "spam")
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
183 else:
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
184 self.fail("no SyntaxError for %r" % (brokencode,))
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
185 checkfilename("def f(x): foo)(") # parse-time
c36ccf2a71e1 get the symtable module back in working order
Benjamin Peterson <benjamin@python.org>
parents: 36380
diff changeset
186 checkfilename("def f(x): global x") # symtable-build-time
36380
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
187
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
188 def test_eval(self):
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
189 symbols = symtable.symtable("42", "?", "eval")
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
190
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
191 def test_single(self):
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
192 symbols = symtable.symtable("42", "?", "single")
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
193
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
194 def test_exec(self):
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
195 symbols = symtable.symtable("def f(x): return x", "?", "exec")
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
196
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
197
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
198 def test_main():
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
199 test_support.run_unittest(SymtableTest)
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
200
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
201 if __name__ == '__main__':
8140f000e9d4 Use unittest and make sure a few other cases don't crash
Neal Norwitz <nnorwitz@gmail.com>
parents: 35944
diff changeset
202 test_main()