+++ /dev/null
-"""\r
-Test the API of the symtable module.\r
-"""\r
-import symtable\r
-import unittest\r
-\r
-from test import test_support\r
-\r
-\r
-TEST_CODE = """\r
-import sys\r
-\r
-glob = 42\r
-\r
-class Mine:\r
- instance_var = 24\r
- def a_method(p1, p2):\r
- pass\r
-\r
-def spam(a, b, *var, **kw):\r
- global bar\r
- bar = 47\r
- x = 23\r
- glob\r
- def internal():\r
- return x\r
- return internal\r
-\r
-def foo():\r
- exec 'm'\r
- from sys import *\r
-\r
-def namespace_test(): pass\r
-def namespace_test(): pass\r
-"""\r
-\r
-\r
-def find_block(block, name):\r
- for ch in block.get_children():\r
- if ch.get_name() == name:\r
- return ch\r
-\r
-\r
-class SymtableTest(unittest.TestCase):\r
-\r
- with test_support.check_warnings(\r
- ("import \* only allowed at module level", SyntaxWarning)):\r
- top = symtable.symtable(TEST_CODE, "?", "exec")\r
- # These correspond to scopes in TEST_CODE\r
- Mine = find_block(top, "Mine")\r
- a_method = find_block(Mine, "a_method")\r
- spam = find_block(top, "spam")\r
- internal = find_block(spam, "internal")\r
- foo = find_block(top, "foo")\r
-\r
- def test_type(self):\r
- self.assertEqual(self.top.get_type(), "module")\r
- self.assertEqual(self.Mine.get_type(), "class")\r
- self.assertEqual(self.a_method.get_type(), "function")\r
- self.assertEqual(self.spam.get_type(), "function")\r
- self.assertEqual(self.internal.get_type(), "function")\r
-\r
- def test_optimized(self):\r
- self.assertFalse(self.top.is_optimized())\r
- self.assertFalse(self.top.has_exec())\r
- self.assertFalse(self.top.has_import_star())\r
-\r
- self.assertTrue(self.spam.is_optimized())\r
-\r
- self.assertFalse(self.foo.is_optimized())\r
- self.assertTrue(self.foo.has_exec())\r
- self.assertTrue(self.foo.has_import_star())\r
-\r
- def test_nested(self):\r
- self.assertFalse(self.top.is_nested())\r
- self.assertFalse(self.Mine.is_nested())\r
- self.assertFalse(self.spam.is_nested())\r
- self.assertTrue(self.internal.is_nested())\r
-\r
- def test_children(self):\r
- self.assertTrue(self.top.has_children())\r
- self.assertTrue(self.Mine.has_children())\r
- self.assertFalse(self.foo.has_children())\r
-\r
- def test_lineno(self):\r
- self.assertEqual(self.top.get_lineno(), 0)\r
- self.assertEqual(self.spam.get_lineno(), 11)\r
-\r
- def test_function_info(self):\r
- func = self.spam\r
- self.assertEqual(func.get_parameters(), ("a", "b", "kw", "var"))\r
- self.assertEqual(func.get_locals(),\r
- ("a", "b", "internal", "kw", "var", "x"))\r
- self.assertEqual(func.get_globals(), ("bar", "glob"))\r
- self.assertEqual(self.internal.get_frees(), ("x",))\r
-\r
- def test_globals(self):\r
- self.assertTrue(self.spam.lookup("glob").is_global())\r
- self.assertFalse(self.spam.lookup("glob").is_declared_global())\r
- self.assertTrue(self.spam.lookup("bar").is_global())\r
- self.assertTrue(self.spam.lookup("bar").is_declared_global())\r
- self.assertFalse(self.internal.lookup("x").is_global())\r
- self.assertFalse(self.Mine.lookup("instance_var").is_global())\r
-\r
- def test_local(self):\r
- self.assertTrue(self.spam.lookup("x").is_local())\r
- self.assertFalse(self.internal.lookup("x").is_local())\r
-\r
- def test_referenced(self):\r
- self.assertTrue(self.internal.lookup("x").is_referenced())\r
- self.assertTrue(self.spam.lookup("internal").is_referenced())\r
- self.assertFalse(self.spam.lookup("x").is_referenced())\r
-\r
- def test_parameters(self):\r
- for sym in ("a", "var", "kw"):\r
- self.assertTrue(self.spam.lookup(sym).is_parameter())\r
- self.assertFalse(self.spam.lookup("x").is_parameter())\r
-\r
- def test_symbol_lookup(self):\r
- self.assertEqual(len(self.top.get_identifiers()),\r
- len(self.top.get_symbols()))\r
-\r
- self.assertRaises(KeyError, self.top.lookup, "not_here")\r
-\r
- def test_namespaces(self):\r
- self.assertTrue(self.top.lookup("Mine").is_namespace())\r
- self.assertTrue(self.Mine.lookup("a_method").is_namespace())\r
- self.assertTrue(self.top.lookup("spam").is_namespace())\r
- self.assertTrue(self.spam.lookup("internal").is_namespace())\r
- self.assertTrue(self.top.lookup("namespace_test").is_namespace())\r
- self.assertFalse(self.spam.lookup("x").is_namespace())\r
-\r
- self.assertTrue(self.top.lookup("spam").get_namespace() is self.spam)\r
- ns_test = self.top.lookup("namespace_test")\r
- self.assertEqual(len(ns_test.get_namespaces()), 2)\r
- self.assertRaises(ValueError, ns_test.get_namespace)\r
-\r
- def test_assigned(self):\r
- self.assertTrue(self.spam.lookup("x").is_assigned())\r
- self.assertTrue(self.spam.lookup("bar").is_assigned())\r
- self.assertTrue(self.top.lookup("spam").is_assigned())\r
- self.assertTrue(self.Mine.lookup("a_method").is_assigned())\r
- self.assertFalse(self.internal.lookup("x").is_assigned())\r
-\r
- def test_imported(self):\r
- self.assertTrue(self.top.lookup("sys").is_imported())\r
-\r
- def test_name(self):\r
- self.assertEqual(self.top.get_name(), "top")\r
- self.assertEqual(self.spam.get_name(), "spam")\r
- self.assertEqual(self.spam.lookup("x").get_name(), "x")\r
- self.assertEqual(self.Mine.get_name(), "Mine")\r
-\r
- def test_class_info(self):\r
- self.assertEqual(self.Mine.get_methods(), ('a_method',))\r
-\r
- def test_filename_correct(self):\r
- ### Bug tickler: SyntaxError file name correct whether error raised\r
- ### while parsing or building symbol table.\r
- def checkfilename(brokencode):\r
- try:\r
- symtable.symtable(brokencode, "spam", "exec")\r
- except SyntaxError as e:\r
- self.assertEqual(e.filename, "spam")\r
- else:\r
- self.fail("no SyntaxError for %r" % (brokencode,))\r
- checkfilename("def f(x): foo)(") # parse-time\r
- checkfilename("def f(x): global x") # symtable-build-time\r
-\r
- def test_eval(self):\r
- symbols = symtable.symtable("42", "?", "eval")\r
-\r
- def test_single(self):\r
- symbols = symtable.symtable("42", "?", "single")\r
-\r
- def test_exec(self):\r
- symbols = symtable.symtable("def f(x): return x", "?", "exec")\r
-\r
-\r
-def test_main():\r
- test_support.run_unittest(SymtableTest)\r
-\r
-if __name__ == '__main__':\r
- test_main()\r