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, PyCF_DONT_IMPLY_DEDENT
   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, PyCF_DONT_IMPLY_DEDENT)
  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                                   PyCF_DONT_IMPLY_DEDENT))
  47
  48         av("3**3","eval")
  49         av("(lambda z: \n z**3)","eval")
  50         av("#a\n#b\na**3","eval")
  51
  52     def test_incomplete(self):
  53         ai = self.assertIncomplete
  54         ai("(a **")
  55         ai("def x():\n")
  56         ai("(a,b,")
  57         ai("(a,b,(")
  58         ai("(a,b,(")
  59         ai("if 9==3:\n   pass\nelse:\n")
  60         ai("if 9==3:\n   pass\nelse:\n   pass")
  61         ai("a = (")
  62         ai("a = 9+ \\")
  63
  64         ai("(","eval")
  65         ai("(\n\n\n","eval")
  66         ai("(9+","eval")
  67         ai("9+ \\","eval")
  68         ai("lambda z: \\","eval")
  69
  70     def test_invalid(self):
  71         ai = self.assertInvalid
  72         ai("a b")
  73         ai("a = ")
  74         ai("a = 9 +")
  75
  76         ai("a = 1","eval")
  77         ai("a = (","eval")
  78         ai("]","eval")
  79         ai("())","eval")
  80         ai("[}","eval")
  81         ai("9+","eval")
  82         ai("lambda z:","eval")
  83         ai("a b","eval")
  84
  85     def test_filename(self):
  86         self.assertEquals(compile_command("a = 1\n", "abc").co_filename,
  87                           compile("a = 1\n", "abc", 'single').co_filename)
  88         self.assertNotEquals(compile_command("a = 1\n", "abc").co_filename,
  89                              compile("a = 1\n", "def", 'single').co_filename)
  90
  91
  92 def test_main():
  93     run_unittest(CodeopTests)
  94
  95
  96 if __name__ == "__main__":
  97     test_main()