""" | |
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 thrown 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() |