summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_compiler.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_compiler.py')
-rw-r--r--Lib/test/test_compiler.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/Lib/test/test_compiler.py b/Lib/test/test_compiler.py
index 7d5ec7c..d491b78 100644
--- a/Lib/test/test_compiler.py
+++ b/Lib/test/test_compiler.py
@@ -3,6 +3,7 @@ from compiler.ast import flatten
import os, sys, time, unittest
import test.test_support
from random import random
+from StringIO import StringIO
# How much time in seconds can pass before we print a 'Still working' message.
_PRINT_WORKING_MSG_INTERVAL = 5 * 60
@@ -156,6 +157,16 @@ class CompilerTest(unittest.TestCase):
self.assertEquals(dct.get('result'), 1)
+ def testPrintFunction(self):
+ c = compiler.compile('from __future__ import print_function\n'
+ 'print("a", "b", sep="**", end="++", '
+ 'file=output)',
+ '<string>',
+ 'exec' )
+ dct = {'output': StringIO()}
+ exec c in dct
+ self.assertEquals(dct['output'].getvalue(), 'a**b++')
+
def _testErrEnc(self, src, text, offset):
try:
compile(src, "", "exec")