From: Eric Smith <eric@trueblade.com>
Date: Wed, 19 Mar 2008 03:13:34 +0000 (+0000)
Subject: Another test for __future__ print_function.
X-Git-Tag: v2.6a2~241
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=14cae96fa686cbddfd1df8dde8af22d6c6cda956;p=python

Another test for __future__ print_function.
---

diff --git a/Lib/test/test_compiler.py b/Lib/test/test_compiler.py
index 7d5ec7c5b8..d491b78096 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")