Lib/test/test_codeop.py

   1 """
   2    Test cases for codeop.py
   3    Nick Mathewson
   4 """
   5 import unittest
   6 from test.test_support import run_unittest
   7
   8 from codeop import compile_command
   9
  10 class CodeopTests(unittest.TestCase):
  11
  12     def assertValid(self, str, symbol='single'):
  13         '''succeed iff str is a valid piece of code'''
  14         expected = compile(str, "<input>", symbol)
  15         self.assertEquals( compile_command(str, "<input>", symbol), expected)
  16
  17
  18     def assertIncomplete(self, str, symbol='single'):
  19         '''succeed iff str is the start of a valid piece of code'''
  20         self.assertEquals( compile_command(str, symbol=symbol), None)
  21
  22     def assertInvalid(self, str, symbol='single', is_syntax=1):
  23         '''succeed iff str is the start of an invalid piece of code'''
  24         try:
  25             compile_command(str,symbol=symbol)
  26             self.fail("No exception thrown for invalid code")
  27         except SyntaxError:
  28             self.assert_(is_syntax)
  29         except OverflowError:
  30             self.assert_(not is_syntax)
  31
  32     def test_valid(self):
  33         av = self.assertValid
  34         av("a = 1\n")
  35         av("def x():\n  pass\n")
  36         av("pass\n")
  37         av("3**3\n")
  38         av("if 9==3:\n   pass\nelse:\n   pass\n")
  39         av("#a\n#b\na = 3\n")
  40         av("#a\n\n   \na=3\n")
  41         av("a=3\n\n")
  42
  43         # special case
  44         self.assertEquals(compile_command(""),
  45                           compile("pass", "<input>", 'single'))
  46
  47         av("3**3","eval")
  48         av("(lambda z: \n z**3)","eval")
  49         av("#a\n#b\na**3","eval")
  50
  51     def test_incomplete(self):
  52         ai = self.assertIncomplete
  53         ai("(a **")
  54         ai("def x():\n")
  55         ai("(a,b,")
  56         ai("(a,b,(")
  57         ai("(a,b,(")
  58         ai("if 9==3:\n   pass\nelse:\n")
  59         ai("if 9==3:\n   pass\nelse:\n   pass")
  60         ai("a = (")
  61         ai("a = 9+ \\")
  62
  63         ai("(","eval")
  64         ai("(\n\n\n","eval")
  65         ai("(9+","eval")
  66         ai("9+ \\","eval")
  67         ai("lambda z: \\","eval")
  68
  69     def test_invalid(self):
  70         ai = self.assertInvalid
  71         ai("a b")
  72         ai("a = ")
  73         ai("a = 9 +")
  74
  75         ai("a = 1","eval")
  76         ai("a = (","eval")
  77         ai("]","eval")
  78         ai("())","eval")
  79         ai("[}","eval")
  80         ai("9+","eval")
  81         ai("lambda z:","eval")
  82         ai("a b","eval")
  83
  84     def test_filename(self):
  85         self.assertEquals(compile_command("a = 1\n", "abc").co_filename,
  86                           compile("a = 1\n", "abc", 'single').co_filename)
  87         self.assertNotEquals(compile_command("a = 1\n", "abc").co_filename,
  88                              compile("a = 1\n", "def", 'single').co_filename)
  89
  90
  91 def test_main():
  92     run_unittest(CodeopTests)
  93
  94
  95 if __name__ == "__main__":
  96     test_main()