Repository navigation
Expand file tree
/
Copy pathtest_textwrapper.py
More file actions
222 lines (176 loc) · 7.08 KB
/
Copy pathtest_textwrapper.py
File metadata and controls
222 lines (176 loc) · 7.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
"""Discretely test functionality of our custom TextWrapper"""
import datetime
from tabulate import _CustomTextWrap as CTW, tabulate, _strip_ansi
from textwrap import TextWrapper as OTW
from common import skip, assert_equal
def test_wrap_multiword_non_wide():
"""TextWrapper: non-wide character regression tests"""
data = "this is a test string for regression splitting"
for width in range(1, len(data)):
orig = OTW(width=width)
cust = CTW(width=width)
assert orig.wrap(data) == cust.wrap(
data
), "Failure on non-wide char multiword regression check for width " + str(width)
def test_wrap_multiword_non_wide_with_hypens():
"""TextWrapper: non-wide character regression tests that contain hyphens"""
data = "how should-we-split-this non-sense string that-has-lots-of-hypens"
for width in range(1, len(data)):
orig = OTW(width=width)
cust = CTW(width=width)
assert orig.wrap(data) == cust.wrap(
data
), "Failure on non-wide char hyphen regression check for width " + str(width)
def test_wrap_longword_non_wide():
"""TextWrapper: Some non-wide character regression tests"""
data = "ThisIsASingleReallyLongWordThatWeNeedToSplit"
for width in range(1, len(data)):
orig = OTW(width=width)
cust = CTW(width=width)
assert orig.wrap(data) == cust.wrap(
data
), "Failure on non-wide char longword regression check for width " + str(width)
def test_wrap_wide_char_multiword():
"""TextWrapper: wrapping support for wide characters with multiple words"""
try:
import wcwidth # noqa
except ImportError:
skip("test_wrap_wide_char is skipped")
data = "약간 감싸면 더 잘 보일 수있는 다소 긴 설명입니다"
expected = ["약간 감싸면 더", "잘 보일 수있는", "다소 긴", "설명입니다"]
wrapper = CTW(width=15)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_wide_char_longword():
"""TextWrapper: wrapping wide char word that needs to be broken up"""
try:
import wcwidth # noqa
except ImportError:
skip("test_wrap_wide_char_longword is skipped")
data = "약간감싸면더잘보일수있"
expected = ["약간", "감싸", "면더", "잘보", "일수", "있"]
# Explicit odd number to ensure the 2 width is taken into account
wrapper = CTW(width=5)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_mixed_string():
"""TextWrapper: wrapping string with mix of wide and non-wide chars"""
try:
import wcwidth # noqa
except ImportError:
skip("test_wrap_wide_char is skipped")
data = (
"This content of this string (この文字列のこの内容) contains "
"multiple character types (複数の文字タイプが含まれています)"
)
expected = [
"This content of this",
"string (この文字列の",
"この内容) contains",
"multiple character",
"types (複数の文字タイ",
"プが含まれています)",
]
wrapper = CTW(width=21)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrapper_len_ignores_color_chars():
data = "\033[31m\033[104mtenletters\033[0m"
result = CTW._len(data)
assert_equal(10, result)
def test_wrap_full_line_color():
"""TextWrapper: Wrap a line when the full thing is enclosed in color tags"""
# This has both a text color and a background color
data = (
"\033[31m\033[104mThis is a test string for testing TextWrap with colors\033[0m"
)
expected = [
"\033[31m\033[104mThis is a test\033[0m",
"\033[31m\033[104mstring for testing\033[0m",
"\033[31m\033[104mTextWrap with colors\033[0m",
]
wrapper = CTW(width=20)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_color_in_single_line():
"""TextWrapper: Wrap a line - preserve internal color tags, and don't
propagate them to other lines when they don't need to be"""
# This has both a text color and a background color
data = "This is a test string for testing \033[31mTextWrap\033[0m with colors"
expected = [
"This is a test string for",
"testing \033[31mTextWrap\033[0m with",
"colors",
]
wrapper = CTW(width=25)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_color_line_splillover():
"""TextWrapper: Wrap a line - preserve internal color tags and wrap them to
other lines when required, requires adding the colors tags to other lines as appropriate
"""
# This has both a text color and a background color
data = "This is a \033[31mtest string for testing TextWrap\033[0m with colors"
expected = [
"This is a \033[31mtest string for\033[0m",
"\033[31mtesting TextWrap\033[0m with",
"colors",
]
wrapper = CTW(width=25)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_color_line_longword():
"""TextWrapper: Wrap a line - preserve internal color tags and wrap them to
other lines when required, requires adding the colors tags to other lines as appropriate
and avoiding splitting escape codes."""
data = "This_is_a_\033[31mtest_string_for_testing_TextWrap\033[0m_with_colors"
expected = [
"This_is_a_\033[31mte\033[0m",
"\033[31mst_string_fo\033[0m",
"\033[31mr_testing_Te\033[0m",
"\033[31mxtWrap\033[0m_with_",
"colors",
]
wrapper = CTW(width=12)
result = wrapper.wrap(data)
assert_equal(expected, result)
def test_wrap_color_line_multiple_escapes():
data = "012345(\x1b[32ma\x1b[0mbc\x1b[32mdefghij\x1b[0m)"
expected = [
"012345(\x1b[32ma\x1b[0mbc\x1b[32m\x1b[0m",
"\x1b[32mdefghij\x1b[0m)",
]
wrapper = CTW(width=10)
result = wrapper.wrap(data)
assert_equal(expected, result)
clean_data = _strip_ansi(data)
for width in range(2, len(clean_data)):
wrapper = CTW(width=width)
result = wrapper.wrap(data)
# Comparing after stripping ANSI should be enough to catch broken escape codes
assert_equal(clean_data, _strip_ansi("".join(result)))
def test_wrap_datetime():
"""TextWrapper: Show that datetimes can be wrapped without crashing"""
data = [
["First Entry", datetime.datetime(2020, 1, 1, 5, 6, 7)],
["Second Entry", datetime.datetime(2021, 2, 2, 0, 0, 0)],
]
headers = ["Title", "When"]
result = tabulate(data, headers=headers, tablefmt="grid", maxcolwidths=[7, 5])
expected = [
"+---------+--------+",
"| Title | When |",
"+=========+========+",
"| First | 2020- |",
"| Entry | 01-01 |",
"| | 05:06 |",
"| | :07 |",
"+---------+--------+",
"| Second | 2021- |",
"| Entry | 02-02 |",
"| | 00:00 |",
"| | :00 |",
"+---------+--------+",
]
expected = "\n".join(expected)
assert_equal(expected, result)