1import test.test_support
2compiler = test.test_support.import_module('compiler', deprecated=True)
3from compiler.ast import flatten
4import os, sys, time, unittest
5from random import random
6from StringIO import StringIO
7
8# How much time in seconds can pass before we print a 'Still working' message.
9_PRINT_WORKING_MSG_INTERVAL = 5 * 60
10
11class TrivialContext(object):
12    def __enter__(self):
13        return self
14    def __exit__(self, *exc_info):
15        pass
16
17class CompilerTest(unittest.TestCase):
18
19    def testCompileLibrary(self):
20        # A simple but large test.  Compile all the code in the
21        # standard library and its test suite.  This doesn't verify
22        # that any of the code is correct, merely the compiler is able
23        # to generate some kind of code for it.
24
25        next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL
26        # warning: if 'os' or 'test_support' are moved in some other dir,
27        # they should be changed here.
28        libdir = os.path.dirname(os.__file__)
29        testdir = os.path.dirname(test.test_support.__file__)
30
31        for dir in [libdir, testdir]:
32            for basename in os.listdir(dir):
33                # Print still working message since this test can be really slow
34                if next_time <= time.time():
35                    next_time = time.time() + _PRINT_WORKING_MSG_INTERVAL
36                    print >>sys.__stdout__, \
37                       '  testCompileLibrary still working, be patient...'
38                    sys.__stdout__.flush()
39
40                if not basename.endswith(".py"):
41                    continue
42                if not TEST_ALL and random() < 0.98:
43                    continue
44                path = os.path.join(dir, basename)
45                if test.test_support.verbose:
46                    print "compiling", path
47                f = open(path, "U")
48                buf = f.read()
49                f.close()
50                if "badsyntax" in basename or "bad_coding" in basename:
51                    self.assertRaises(SyntaxError, compiler.compile,
52                                      buf, basename, "exec")
53                else:
54                    try:
55                        compiler.compile(buf, basename, "exec")
56                    except Exception, e:
57                        args = list(e.args)
58                        args.append("in file %s]" % basename)
59                        #args[0] += "[in file %s]" % basename
60                        e.args = tuple(args)
61                        raise
62
63    def testNewClassSyntax(self):
64        compiler.compile("class foo():pass\n\n","<string>","exec")
65
66    def testYieldExpr(self):
67        compiler.compile("def g(): yield\n\n", "<string>", "exec")
68
69    def testKeywordAfterStarargs(self):
70        def f(*args, **kwargs):
71            self.assertEqual((args, kwargs), ((2,3), {'x': 1, 'y': 4}))
72        c = compiler.compile('f(x=1, *(2, 3), y=4)', '<string>', 'exec')
73        exec c in {'f': f}
74
75        self.assertRaises(SyntaxError, compiler.parse, "foo(a=1, b)")
76        self.assertRaises(SyntaxError, compiler.parse, "foo(1, *args, 3)")
77
78    def testTryExceptFinally(self):
79        # Test that except and finally clauses in one try stmt are recognized
80        c = compiler.compile("try:\n 1//0\nexcept:\n e = 1\nfinally:\n f = 1",
81                             "<string>", "exec")
82        dct = {}
83        exec c in dct
84        self.assertEqual(dct.get('e'), 1)
85        self.assertEqual(dct.get('f'), 1)
86
87    def testDefaultArgs(self):
88        self.assertRaises(SyntaxError, compiler.parse, "def foo(a=1, b): pass")
89
90    def testDocstrings(self):
91        c = compiler.compile('"doc"', '<string>', 'exec')
92        self.assertIn('__doc__', c.co_names)
93        c = compiler.compile('def f():\n "doc"', '<string>', 'exec')
94        g = {}
95        exec c in g
96        self.assertEqual(g['f'].__doc__, "doc")
97
98    def testLineNo(self):
99        # Test that all nodes except Module have a correct lineno attribute.
100        filename = __file__
101        if filename.endswith((".pyc", ".pyo")):
102            filename = filename[:-1]
103        tree = compiler.parseFile(filename)
104        self.check_lineno(tree)
105
106    def check_lineno(self, node):
107        try:
108            self._check_lineno(node)
109        except AssertionError:
110            print node.__class__, node.lineno
111            raise
112
113    def _check_lineno(self, node):
114        if not node.__class__ in NOLINENO:
115            self.assertIsInstance(node.lineno, int,
116                "lineno=%s on %s" % (node.lineno, node.__class__))
117            self.assertTrue(node.lineno > 0,
118                "lineno=%s on %s" % (node.lineno, node.__class__))
119        for child in node.getChildNodes():
120            self.check_lineno(child)
121
122    def testFlatten(self):
123        self.assertEqual(flatten([1, [2]]), [1, 2])
124        self.assertEqual(flatten((1, (2,))), [1, 2])
125
126    def testNestedScope(self):
127        c = compiler.compile('def g():\n'
128                             '    a = 1\n'
129                             '    def f(): return a + 2\n'
130                             '    return f()\n'
131                             'result = g()',
132                             '<string>',
133                             'exec')
134        dct = {}
135        exec c in dct
136        self.assertEqual(dct.get('result'), 3)
137
138    def testGenExp(self):
139        c = compiler.compile('list((i,j) for i in range(3) if i < 3'
140                             '           for j in range(4) if j > 2)',
141                             '<string>',
142                             'eval')
143        self.assertEqual(eval(c), [(0, 3), (1, 3), (2, 3)])
144
145    def testSetLiteral(self):
146        c = compiler.compile('{1, 2, 3}', '<string>', 'eval')
147        self.assertEqual(eval(c), {1,2,3})
148        c = compiler.compile('{1, 2, 3,}', '<string>', 'eval')
149        self.assertEqual(eval(c), {1,2,3})
150
151    def testDictLiteral(self):
152        c = compiler.compile('{1:2, 2:3, 3:4}', '<string>', 'eval')
153        self.assertEqual(eval(c), {1:2, 2:3, 3:4})
154        c = compiler.compile('{1:2, 2:3, 3:4,}', '<string>', 'eval')
155        self.assertEqual(eval(c), {1:2, 2:3, 3:4})
156
157    def testSetComp(self):
158        c = compiler.compile('{x for x in range(1, 4)}', '<string>', 'eval')
159        self.assertEqual(eval(c), {1, 2, 3})
160        c = compiler.compile('{x * y for x in range(3) if x != 0'
161                             '       for y in range(4) if y != 0}',
162                             '<string>',
163                             'eval')
164        self.assertEqual(eval(c), {1, 2, 3, 4, 6})
165
166    def testDictComp(self):
167        c = compiler.compile('{x:x+1 for x in range(1, 4)}', '<string>', 'eval')
168        self.assertEqual(eval(c), {1:2, 2:3, 3:4})
169        c = compiler.compile('{(x, y) : y for x in range(2) if x != 0'
170                             '            for y in range(3) if y != 0}',
171                             '<string>',
172                             'eval')
173        self.assertEqual(eval(c), {(1, 2): 2, (1, 1): 1})
174
175    def testWith(self):
176        # SF bug 1638243
177        c = compiler.compile('from __future__ import with_statement\n'
178                             'def f():\n'
179                             '    with TrivialContext():\n'
180                             '        return 1\n'
181                             'result = f()',
182                             '<string>',
183                             'exec' )
184        dct = {'TrivialContext': TrivialContext}
185        exec c in dct
186        self.assertEqual(dct.get('result'), 1)
187
188    def testWithAss(self):
189        c = compiler.compile('from __future__ import with_statement\n'
190                             'def f():\n'
191                             '    with TrivialContext() as tc:\n'
192                             '        return 1\n'
193                             'result = f()',
194                             '<string>',
195                             'exec' )
196        dct = {'TrivialContext': TrivialContext}
197        exec c in dct
198        self.assertEqual(dct.get('result'), 1)
199
200    def testWithMult(self):
201        events = []
202        class Ctx:
203            def __init__(self, n):
204                self.n = n
205            def __enter__(self):
206                events.append(self.n)
207            def __exit__(self, *args):
208                pass
209        c = compiler.compile('from __future__ import with_statement\n'
210                             'def f():\n'
211                             '    with Ctx(1) as tc, Ctx(2) as tc2:\n'
212                             '        return 1\n'
213                             'result = f()',
214                             '<string>',
215                             'exec' )
216        dct = {'Ctx': Ctx}
217        exec c in dct
218        self.assertEqual(dct.get('result'), 1)
219        self.assertEqual(events, [1, 2])
220
221    def testGlobal(self):
222        code = compiler.compile('global x\nx=1', '<string>', 'exec')
223        d1 = {'__builtins__': {}}
224        d2 = {}
225        exec code in d1, d2
226        # x should be in the globals dict
227        self.assertEqual(d1.get('x'), 1)
228
229    def testPrintFunction(self):
230        c = compiler.compile('from __future__ import print_function\n'
231                             'print("a", "b", sep="**", end="++", '
232                                    'file=output)',
233                             '<string>',
234                             'exec' )
235        dct = {'output': StringIO()}
236        exec c in dct
237        self.assertEqual(dct['output'].getvalue(), 'a**b++')
238
239    def _testErrEnc(self, src, text, offset):
240        try:
241            compile(src, "", "exec")
242        except SyntaxError, e:
243            self.assertEqual(e.offset, offset)
244            self.assertEqual(e.text, text)
245
246    def testSourceCodeEncodingsError(self):
247        # Test SyntaxError with encoding definition
248        sjis = "print '\x83\x70\x83\x43\x83\x5c\x83\x93', '\n"
249        ascii = "print '12345678', '\n"
250        encdef = "#! -*- coding: ShiftJIS -*-\n"
251
252        # ascii source without encdef
253        self._testErrEnc(ascii, ascii, 19)
254
255        # ascii source with encdef
256        self._testErrEnc(encdef+ascii, ascii, 19)
257
258        # non-ascii source with encdef
259        self._testErrEnc(encdef+sjis, sjis, 19)
260
261        # ShiftJIS source without encdef
262        self._testErrEnc(sjis, sjis, 19)
263
264
265NOLINENO = (compiler.ast.Module, compiler.ast.Stmt, compiler.ast.Discard)
266
267###############################################################################
268# code below is just used to trigger some possible errors, for the benefit of
269# testLineNo
270###############################################################################
271
272class Toto:
273    """docstring"""
274    pass
275
276a, b = 2, 3
277[c, d] = 5, 6
278l = [(x, y) for x, y in zip(range(5), range(5,10))]
279l[0]
280l[3:4]
281d = {'a': 2}
282d = {}
283d = {x: y for x, y in zip(range(5), range(5,10))}
284s = {x for x in range(10)}
285s = {1}
286t = ()
287t = (1, 2)
288l = []
289l = [1, 2]
290if l:
291    pass
292else:
293    a, b = b, a
294
295try:
296    print yo
297except:
298    yo = 3
299else:
300    yo += 3
301
302try:
303    a += b
304finally:
305    b = 0
306
307from math import *
308
309###############################################################################
310
311def test_main():
312    global TEST_ALL
313    TEST_ALL = test.test_support.is_resource_enabled("cpu")
314    test.test_support.run_unittest(CompilerTest)
315
316if __name__ == "__main__":
317    test_main()
318