Another test for __future__ print_function.

This commit is contained in:
Eric Smith 2008-03-19 03:13:34 +00:00
parent 2724ab99c8
commit 14cae96fa6
1 changed files with 11 additions and 0 deletions

View File

@ -3,6 +3,7 @@ from compiler.ast import flatten
import os, sys, time, unittest import os, sys, time, unittest
import test.test_support import test.test_support
from random import random from random import random
from StringIO import StringIO
# How much time in seconds can pass before we print a 'Still working' message. # How much time in seconds can pass before we print a 'Still working' message.
_PRINT_WORKING_MSG_INTERVAL = 5 * 60 _PRINT_WORKING_MSG_INTERVAL = 5 * 60
@ -156,6 +157,16 @@ class CompilerTest(unittest.TestCase):
self.assertEquals(dct.get('result'), 1) 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): def _testErrEnc(self, src, text, offset):
try: try:
compile(src, "", "exec") compile(src, "", "exec")