summaryrefslogtreecommitdiff
path: root/lib/python2.7/test/test_print.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python2.7/test/test_print.py')
-rw-r--r--lib/python2.7/test/test_print.py141
1 files changed, 0 insertions, 141 deletions
diff --git a/lib/python2.7/test/test_print.py b/lib/python2.7/test/test_print.py
deleted file mode 100644
index b726347..0000000
--- a/lib/python2.7/test/test_print.py
+++ /dev/null
@@ -1,141 +0,0 @@
-"""Test correct operation of the print function.
-"""
-
-# In 2.6, this gives us the behavior we want. In 3.0, it has
-# no function, but it still must parse correctly.
-from __future__ import print_function
-
-import unittest
-from test import test_support
-
-from StringIO import StringIO
-
-NotDefined = object()
-
-# A dispatch table all 8 combinations of providing
-# sep, end, and file
-# I use this machinery so that I'm not just passing default
-# values to print, I'm either passing or not passing in the
-# arguments
-dispatch = {
- (False, False, False):
- lambda args, sep, end, file: print(*args),
- (False, False, True):
- lambda args, sep, end, file: print(file=file, *args),
- (False, True, False):
- lambda args, sep, end, file: print(end=end, *args),
- (False, True, True):
- lambda args, sep, end, file: print(end=end, file=file, *args),
- (True, False, False):
- lambda args, sep, end, file: print(sep=sep, *args),
- (True, False, True):
- lambda args, sep, end, file: print(sep=sep, file=file, *args),
- (True, True, False):
- lambda args, sep, end, file: print(sep=sep, end=end, *args),
- (True, True, True):
- lambda args, sep, end, file: print(sep=sep, end=end, file=file, *args),
- }
-
-# Class used to test __str__ and print
-class ClassWith__str__:
- def __init__(self, x):
- self.x = x
- def __str__(self):
- return self.x
-
-class TestPrint(unittest.TestCase):
- def check(self, expected, args,
- sep=NotDefined, end=NotDefined, file=NotDefined):
- # Capture sys.stdout in a StringIO. Call print with args,
- # and with sep, end, and file, if they're defined. Result
- # must match expected.
-
- # Look up the actual function to call, based on if sep, end, and file
- # are defined
- fn = dispatch[(sep is not NotDefined,
- end is not NotDefined,
- file is not NotDefined)]
-
- with test_support.captured_stdout() as t:
- fn(args, sep, end, file)
-
- self.assertEqual(t.getvalue(), expected)
-
- def test_print(self):
- def x(expected, args, sep=NotDefined, end=NotDefined):
- # Run the test 2 ways: not using file, and using
- # file directed to a StringIO
-
- self.check(expected, args, sep=sep, end=end)
-
- # When writing to a file, stdout is expected to be empty
- o = StringIO()
- self.check('', args, sep=sep, end=end, file=o)
-
- # And o will contain the expected output
- self.assertEqual(o.getvalue(), expected)
-
- x('\n', ())
- x('a\n', ('a',))
- x('None\n', (None,))
- x('1 2\n', (1, 2))
- x('1 2\n', (1, ' ', 2))
- x('1*2\n', (1, 2), sep='*')
- x('1 s', (1, 's'), end='')
- x('a\nb\n', ('a', 'b'), sep='\n')
- x('1.01', (1.0, 1), sep='', end='')
- x('1*a*1.3+', (1, 'a', 1.3), sep='*', end='+')
- x('a\n\nb\n', ('a\n', 'b'), sep='\n')
- x('\0+ +\0\n', ('\0', ' ', '\0'), sep='+')
-
- x('a\n b\n', ('a\n', 'b'))
- x('a\n b\n', ('a\n', 'b'), sep=None)
- x('a\n b\n', ('a\n', 'b'), end=None)
- x('a\n b\n', ('a\n', 'b'), sep=None, end=None)
-
- x('*\n', (ClassWith__str__('*'),))
- x('abc 1\n', (ClassWith__str__('abc'), 1))
-
- # 2.x unicode tests
- x(u'1 2\n', ('1', u'2'))
- x(u'u\1234\n', (u'u\1234',))
- x(u' abc 1\n', (' ', ClassWith__str__(u'abc'), 1))
-
- # errors
- self.assertRaises(TypeError, print, '', sep=3)
- self.assertRaises(TypeError, print, '', end=3)
- self.assertRaises(AttributeError, print, '', file='')
-
- def test_mixed_args(self):
- # If an unicode arg is passed, sep and end should be unicode, too.
- class Recorder(object):
-
- def __init__(self, must_be_unicode):
- self.buf = []
- self.force_unicode = must_be_unicode
-
- def write(self, what):
- if self.force_unicode and not isinstance(what, unicode):
- raise AssertionError("{0!r} is not unicode".format(what))
- self.buf.append(what)
-
- buf = Recorder(True)
- print(u'hi', file=buf)
- self.assertEqual(u''.join(buf.buf), 'hi\n')
- del buf.buf[:]
- print(u'hi', u'nothing', file=buf)
- self.assertEqual(u''.join(buf.buf), 'hi nothing\n')
- buf = Recorder(False)
- print('hi', 'bye', end=u'\n', file=buf)
- self.assertIsInstance(buf.buf[1], unicode)
- self.assertIsInstance(buf.buf[3], unicode)
- del buf.buf[:]
- print(sep=u'x', file=buf)
- self.assertIsInstance(buf.buf[-1], unicode)
-
-
-def test_main():
- test_support.run_unittest(TestPrint)
-
-if __name__ == "__main__":
- test_main()