beefed up version: jython support, covers now fixed differences between CPython/Jython.

Lib/test/test_codeop.py

@@ -3,17 +3,43 @@
    Nick Mathewson
 """
 import unittest
-from test.test_support import run_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'''
-        expected = compile(str, "<input>", symbol, PyCF_DONT_IMPLY_DEDENT)
-        self.assertEquals( compile_command(str, "<input>", symbol), expected)
-
+        if is_jython:
+            code = compile_command(str, "<input>", symbol)
+            self.assert_(code)
+            if symbol == "single":
+                d,r = {},{}
+                sys.stdout = cStringIO.StringIO()
+                try:
+                    exec code in d
+                    exec compile(str,"<input>","single") in r
+                finally:
+                    sys.stdout = sys.__stdout__
+            elif symbol == 'eval':
+                ctx = {'a': 2}
+                d = { 'value': eval(code,ctx) }
+                r = { 'value': eval(str,ctx) }
+            self.assertEquals(unify_callables(r),unify_callables(d))
+        else:
+            expected = compile(str, "<input>", symbol, PyCF_DONT_IMPLY_DEDENT)
+            self.assertEquals( compile_command(str, "<input>", symbol), expected)
 
     def assertIncomplete(self, str, symbol='single'):
         '''succeed iff str is the start of a valid piece of code'''
@@ -31,36 +57,91 @@ class CodeopTests(unittest.TestCase):
 
     def test_valid(self):
         av = self.assertValid
+
+        # special case
+        if not is_jython:
+            self.assertEquals(compile_command(""),
+                            compile("pass", "<input>", 'single',
+                                    PyCF_DONT_IMPLY_DEDENT))
+            self.assertEquals(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")
-
-        # special case
-        self.assertEquals(compile_command(""),
-                          compile("pass", "<input>", 'single',
-                                  PyCF_DONT_IMPLY_DEDENT))
+        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")
 
     def test_incomplete(self):
         ai = self.assertIncomplete
+
         ai("(a **")
-        ai("def x():\n")
         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("a = (")
+        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")
@@ -70,9 +151,22 @@ class CodeopTests(unittest.TestCase):
     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")