1
# Copyright (C) 2011 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
"""Tests of the breezy.utextwrap."""
26
# Japanese "Good morning".
27
# Each character have double width. So total 8 width on console.
28
_str_D = u'\u304a\u306f\u3088\u3046'
32
# Combine single width characters and double width characters.
33
_str_SD = _str_S + _str_D
34
_str_DS = _str_D + _str_S
37
class TestUTextWrap(tests.TestCase):
39
def check_width(self, text, expected_width):
40
w = utextwrap.UTextWrapper()
44
"Width of %r should be %d" % (text, expected_width))
47
self.check_width(_str_D, 8)
48
self.check_width(_str_SD, 13)
50
def check_cut(self, text, width, pos):
51
w = utextwrap.UTextWrapper()
52
self.assertEqual((text[:pos], text[pos:]), w._cut(text, width))
56
self.check_cut(s, 0, 0)
57
self.check_cut(s, 1, 1)
58
self.check_cut(s, 5, 5)
59
self.check_cut(s, 6, 5)
60
self.check_cut(s, 7, 6)
61
self.check_cut(s, 12, 8)
62
self.check_cut(s, 13, 9)
63
self.check_cut(s, 14, 9)
64
self.check_cut(u'A' * 5, 3, 3)
67
w = utextwrap.UTextWrapper()
68
self.assertEqual(list(_str_D), w._split(_str_D))
69
self.assertEqual([_str_S] + list(_str_D), w._split(_str_SD))
70
self.assertEqual(list(_str_D) + [_str_S], w._split(_str_DS))
73
self.assertEqual(list(_str_D), utextwrap.wrap(_str_D, 1))
74
self.assertEqual(list(_str_D), utextwrap.wrap(_str_D, 2))
75
self.assertEqual(list(_str_D), utextwrap.wrap(_str_D, 3))
76
self.assertEqual(list(_str_D),
77
utextwrap.wrap(_str_D, 3, break_long_words=False))
80
class TestUTextFill(tests.TestCase):
82
def test_fill_simple(self):
83
# Test only can call fill() because it's just '\n'.join(wrap(text)).
84
self.assertEqual("%s\n%s" % (_str_D[:2], _str_D[2:]),
85
utextwrap.fill(_str_D, 4))
87
def test_fill_with_breaks(self):
88
# Demonstrate complicated case.
89
text = u"spam ham egg spamhamegg" + _str_D + u" spam" + _str_D * 2
90
self.assertEqual(u'\n'.join(["spam ham",
95
_str_D[2:] + _str_D[:2],
97
utextwrap.fill(text, 8))
99
def test_fill_without_breaks(self):
100
text = u"spam ham egg spamhamegg" + _str_D + u" spam" + _str_D * 2
101
self.assertEqual(u'\n'.join(["spam ham",
104
# border between single width and double
108
_str_D[2:] + _str_D[:2],
110
utextwrap.fill(text, 8, break_long_words=False))
112
def test_fill_indent_with_breaks(self):
113
w = utextwrap.UTextWrapper(8, initial_indent=' ' * 4,
114
subsequent_indent=' ' * 4)
115
self.assertEqual(u'\n'.join([" hell",
122
def test_fill_indent_without_breaks(self):
123
w = utextwrap.UTextWrapper(8, initial_indent=' ' * 4,
124
subsequent_indent=' ' * 4)
125
w.break_long_words = False
126
self.assertEqual(u'\n'.join([" hello",
132
def test_fill_indent_without_breaks_with_fixed_width(self):
133
w = utextwrap.UTextWrapper(8, initial_indent=' ' * 4,
134
subsequent_indent=' ' * 4)
135
w.break_long_words = False
137
self.assertEqual(u'\n'.join([" hello",
146
class TestUTextWrapAmbiWidth(tests.TestCase):
147
_cyrill_char = u"\u0410" # east_asian_width() == 'A'
149
def test_ambiwidth1(self):
150
w = utextwrap.UTextWrapper(4, ambiguous_width=1)
151
s = self._cyrill_char * 8
152
self.assertEqual([self._cyrill_char * 4] * 2, w.wrap(s))
154
def test_ambiwidth2(self):
155
w = utextwrap.UTextWrapper(4, ambiguous_width=2)
156
s = self._cyrill_char * 8
157
self.assertEqual([self._cyrill_char * 2] * 4, w.wrap(s))
160
# Regression test with Python's test_textwrap
161
# Note that some distribution including Ubuntu doesn't install
162
# Python's test suite.
164
from test import test_textwrap
166
def override_textwrap_symbols(testcase):
167
# Override the symbols imported by test_textwrap so it uses our own
169
testcase.overrideAttr(test_textwrap, 'TextWrapper',
170
utextwrap.UTextWrapper)
171
testcase.overrideAttr(test_textwrap, 'wrap', utextwrap.wrap)
172
testcase.overrideAttr(test_textwrap, 'fill', utextwrap.fill)
174
def setup_both(testcase, base_class, reused_class):
175
super(base_class, testcase).setUp()
176
override_textwrap_symbols(testcase)
177
reused_class.setUp(testcase)
179
class TestWrap(tests.TestCase, test_textwrap.WrapTestCase):
182
setup_both(self, TestWrap, test_textwrap.WrapTestCase)
184
class TestLongWord(tests.TestCase, test_textwrap.LongWordTestCase):
187
setup_both(self, TestLongWord, test_textwrap.LongWordTestCase)
189
class TestIndent(tests.TestCase, test_textwrap.IndentTestCases):
192
setup_both(self, TestIndent, test_textwrap.IndentTestCases)
196
class TestWrap(tests.TestCase):
199
raise tests.TestSkipped("test.test_textwrap is not available.")
201
class TestLongWord(tests.TestCase):
203
def test_longword(self):
204
raise tests.TestSkipped("test.test_textwrap is not available.")
206
class TestIndent(tests.TestCase):
208
def test_indent(self):
209
raise tests.TestSkipped("test.test_textwrap is not available.")