Add test_unicode() to ensure that 1) textwrap doesn't crash on unicode

input, and 2) unicode input means unicode output.  This closes
SF bug #622831.

diff --git a/Lib/test/test_textwrap.py b/Lib/test/test_textwrap.py
index d164d75ae659ef08e103267f8053657416650999..b1cee3053914b7d9621b236b44bc0516ed2506cd 100644 (file)
--- a/Lib/test/test_textwrap.py
+++ b/Lib/test/test_textwrap.py
@@ -239,6 +239,18 @@ What a mess!
         self.check_wrap(text, 30,
                         [" This is a sentence with", "leading whitespace."])
 
+    def test_unicode(self):
+        # *Very* simple test of wrapping Unicode strings.  I'm sure
+        # there's more to it than this, but let's at least make
+        # sure textwrap doesn't crash on Unicode input!
+        text = u"Hello there, how are you today?"
+        self.check_wrap(text, 50, [u"Hello there, how are you today?"])
+        self.check_wrap(text, 20, [u"Hello there, how are", "you today?"])
+        olines = self.wrapper.wrap(text)
+        assert isinstance(olines, list) and isinstance(olines[0], unicode)
+        otext = self.wrapper.fill(text)
+        assert isinstance(otext, unicode)
+
     def test_split(self):
         # Ensure that the standard _split() method works as advertised
         # in the comments