| """ |
| Test cases for codeop.py |
| Nick Mathewson |
| """ |
| import unittest |
| from test.test_support import run_unittest, is_jython |
| |
| from codeop import compile_command, PyCF_DONT_IMPLY_DEDENT |
| |
| if is_jython: |
| import sys |
| import cStringIO |
| |
| def unify_callables(d): |
| for n,v in d.items(): |
| if callable(v): |
| d[n] = callable |
| return d |
| |
| class CodeopTests(unittest.TestCase): |
| |
| def assertValid(self, str, symbol='single'): |
| '''succeed iff str is a valid piece of code''' |
| if is_jython: |
| code = compile_command(str, "<input>", symbol) |
| self.assertTrue(code) |
| if symbol == "single": |
| d,r = {},{} |
| saved_stdout = sys.stdout |
| sys.stdout = cStringIO.StringIO() |
| try: |
| exec code in d |
| exec compile(str,"<input>","single") in r |
| finally: |
| sys.stdout = saved_stdout |
| elif symbol == 'eval': |
| ctx = {'a': 2} |
| d = { 'value': eval(code,ctx) } |
| r = { 'value': eval(str,ctx) } |
| self.assertEqual(unify_callables(r),unify_callables(d)) |
| else: |
| expected = compile(str, "<input>", symbol, PyCF_DONT_IMPLY_DEDENT) |
| self.assertEqual(compile_command(str, "<input>", symbol), expected) |
| |
| def assertIncomplete(self, str, symbol='single'): |
| '''succeed iff str is the start of a valid piece of code''' |
| self.assertEqual(compile_command(str, symbol=symbol), None) |
| |
| def assertInvalid(self, str, symbol='single', is_syntax=1): |
| '''succeed iff str is the start of an invalid piece of code''' |
| try: |
| compile_command(str,symbol=symbol) |
| self.fail("No exception raised for invalid code") |
| except SyntaxError: |
| self.assertTrue(is_syntax) |
| except OverflowError: |
| self.assertTrue(not is_syntax) |
| |
| def test_valid(self): |
| av = self.assertValid |
| |
| # special case |
| if not is_jython: |
| self.assertEqual(compile_command(""), |
| compile("pass", "<input>", 'single', |
| PyCF_DONT_IMPLY_DEDENT)) |
| self.assertEqual(compile_command("\n"), |
| compile("pass", "<input>", 'single', |
| PyCF_DONT_IMPLY_DEDENT)) |
| else: |
| av("") |
| av("\n") |
| |
| av("a = 1") |
| av("\na = 1") |
| av("a = 1\n") |
| av("a = 1\n\n") |
| av("\n\na = 1\n\n") |
| |
| av("def x():\n pass\n") |
| av("if 1:\n pass\n") |
| |
| av("\n\nif 1: pass\n") |
| av("\n\nif 1: pass\n\n") |
| |
| av("def x():\n\n pass\n") |
| av("def x():\n pass\n \n") |
| av("def x():\n pass\n \n") |
| |
| av("pass\n") |
| av("3**3\n") |
| |
| av("if 9==3:\n pass\nelse:\n pass\n") |
| av("if 1:\n pass\n if 1:\n pass\n else:\n pass\n") |
| |
| av("#a\n#b\na = 3\n") |
| av("#a\n\n \na=3\n") |
| av("a=3\n\n") |
| av("a = 9+ \\\n3") |
| |
| av("3**3","eval") |
| av("(lambda z: \n z**3)","eval") |
| |
| av("9+ \\\n3","eval") |
| av("9+ \\\n3\n","eval") |
| |
| av("\n\na**3","eval") |
| av("\n \na**3","eval") |
| av("#a\n#b\na**3","eval") |
| |
| av("\n\na = 1\n\n") |
| av("\n\nif 1: a=1\n\n") |
| |
| av("if 1:\n pass\n if 1:\n pass\n else:\n pass\n") |
| av("#a\n\n \na=3\n\n") |
| |
| av("\n\na**3","eval") |
| av("\n \na**3","eval") |
| av("#a\n#b\na**3","eval") |
| |
| av("def f():\n try: pass\n finally: [x for x in (1,2)]\n") |
| av("def f():\n pass\n#foo\n") |
| av("@a.b.c\ndef f():\n pass\n") |
| |
| def test_incomplete(self): |
| ai = self.assertIncomplete |
| |
| ai("(a **") |
| ai("(a,b,") |
| ai("(a,b,(") |
| ai("(a,b,(") |
| ai("a = (") |
| ai("a = {") |
| ai("b + {") |
| |
| ai("if 9==3:\n pass\nelse:") |
| ai("if 9==3:\n pass\nelse:\n") |
| ai("if 9==3:\n pass\nelse:\n pass") |
| ai("if 1:") |
| ai("if 1:\n") |
| ai("if 1:\n pass\n if 1:\n pass\n else:") |
| ai("if 1:\n pass\n if 1:\n pass\n else:\n") |
| ai("if 1:\n pass\n if 1:\n pass\n else:\n pass") |
| |
| ai("def x():") |
| ai("def x():\n") |
| ai("def x():\n\n") |
| |
| ai("def x():\n pass") |
| ai("def x():\n pass\n ") |
| ai("def x():\n pass\n ") |
| ai("\n\ndef x():\n pass") |
| |
| ai("a = 9+ \\") |
| ai("a = 'a\\") |
| ai("a = '''xy") |
| |
| ai("","eval") |
| ai("\n","eval") |
| ai("(","eval") |
| ai("(\n\n\n","eval") |
| ai("(9+","eval") |
| ai("9+ \\","eval") |
| ai("lambda z: \\","eval") |
| |
| ai("if True:\n if True:\n if True: \n") |
| |
| ai("@a(") |
| ai("@a(b") |
| ai("@a(b,") |
| ai("@a(b,c") |
| ai("@a(b,c,") |
| |
| ai("from a import (") |
| ai("from a import (b") |
| ai("from a import (b,") |
| ai("from a import (b,c") |
| ai("from a import (b,c,") |
| |
| ai("["); |
| ai("[a"); |
| ai("[a,"); |
| ai("[a,b"); |
| ai("[a,b,"); |
| |
| ai("{"); |
| ai("{a"); |
| ai("{a:"); |
| ai("{a:b"); |
| ai("{a:b,"); |
| ai("{a:b,c"); |
| ai("{a:b,c:"); |
| ai("{a:b,c:d"); |
| ai("{a:b,c:d,"); |
| |
| ai("a(") |
| ai("a(b") |
| ai("a(b,") |
| ai("a(b,c") |
| ai("a(b,c,") |
| |
| ai("a[") |
| ai("a[b") |
| ai("a[b,") |
| ai("a[b:") |
| ai("a[b:c") |
| ai("a[b:c:") |
| ai("a[b:c:d") |
| |
| ai("def a(") |
| ai("def a(b") |
| ai("def a(b,") |
| ai("def a(b,c") |
| ai("def a(b,c,") |
| |
| ai("(") |
| ai("(a") |
| ai("(a,") |
| ai("(a,b") |
| ai("(a,b,") |
| |
| ai("if a:\n pass\nelif b:") |
| ai("if a:\n pass\nelif b:\n pass\nelse:") |
| |
| ai("while a:") |
| ai("while a:\n pass\nelse:") |
| |
| ai("for a in b:") |
| ai("for a in b:\n pass\nelse:") |
| |
| ai("try:") |
| ai("try:\n pass\nexcept:") |
| ai("try:\n pass\nfinally:") |
| ai("try:\n pass\nexcept:\n pass\nfinally:") |
| |
| ai("with a:") |
| ai("with a as b:") |
| |
| ai("class a:") |
| ai("class a(") |
| ai("class a(b") |
| ai("class a(b,") |
| ai("class a():") |
| |
| ai("[x for") |
| ai("[x for x in") |
| ai("[x for x in (") |
| |
| ai("(x for") |
| ai("(x for x in") |
| ai("(x for x in (") |
| |
| def test_invalid(self): |
| ai = self.assertInvalid |
| ai("a b") |
| |
| ai("a @") |
| ai("a b @") |
| ai("a ** @") |
| |
| ai("a = ") |
| ai("a = 9 +") |
| |
| ai("def x():\n\npass\n") |
| |
| ai("\n\n if 1: pass\n\npass") |
| |
| ai("a = 9+ \\\n") |
| ai("a = 'a\\ ") |
| ai("a = 'a\\\n") |
| |
| ai("a = 1","eval") |
| ai("a = (","eval") |
| ai("]","eval") |
| ai("())","eval") |
| ai("[}","eval") |
| ai("9+","eval") |
| ai("lambda z:","eval") |
| ai("a b","eval") |
| |
| ai("return 2.3") |
| ai("if (a == 1 and b = 2): pass") |
| |
| ai("del 1") |
| ai("del ()") |
| ai("del (1,)") |
| ai("del [1]") |
| ai("del '1'") |
| |
| ai("[i for i in range(10)] = (1, 2, 3)") |
| |
| def test_filename(self): |
| self.assertEqual(compile_command("a = 1\n", "abc").co_filename, |
| compile("a = 1\n", "abc", 'single').co_filename) |
| self.assertNotEqual(compile_command("a = 1\n", "abc").co_filename, |
| compile("a = 1\n", "def", 'single').co_filename) |
| |
| |
| def test_main(): |
| run_unittest(CodeopTests) |
| |
| |
| if __name__ == "__main__": |
| test_main() |