forked from pool/python-texttable
4acbb44c14
- Update to 1.6.0 * Added texttable.pyi * Added missing set_header_align method stub * Allow chaining of method calls * Replace distutils with setuptools * Create a method for redefining the max_width * Switch to MIT License * Use wcwidth if available for better rendering (basic emoji support) * Update README.md OBS-URL: https://build.opensuse.org/request/show/676515 OBS-URL: https://build.opensuse.org/package/show/devel:languages:python/python-texttable?expand=0&rev=9
291 lines
8.5 KiB
Python
291 lines
8.5 KiB
Python
#coding: utf-8
|
||
|
||
import re
|
||
import sys
|
||
from textwrap import dedent
|
||
from texttable import Texttable
|
||
|
||
if sys.version >= '3':
|
||
u_dedent = dedent
|
||
else:
|
||
def u_dedent(b):
|
||
return unicode(dedent(b), 'utf-8')
|
||
|
||
def clean(text):
|
||
return re.sub(r'( +)$', '', text, flags=re.MULTILINE) + '\n'
|
||
|
||
def test_texttable():
|
||
table = Texttable()
|
||
table.set_cols_align(["l", "r", "c"])
|
||
table.set_cols_valign(["t", "m", "b"])
|
||
table.add_rows([
|
||
["Name", "Age", "Nickname"],
|
||
["Mr\nXavier\nHuon", 32, "Xav'"],
|
||
["Mr\nBaptiste\nClement", 1, "Baby"],
|
||
["Mme\nLouise\nBourgeau", 28, "Lou\n \nLoue"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
+----------+-----+----------+
|
||
| Name | Age | Nickname |
|
||
+==========+=====+==========+
|
||
| Mr | | |
|
||
| Xavier | 32 | |
|
||
| Huon | | Xav' |
|
||
+----------+-----+----------+
|
||
| Mr | | |
|
||
| Baptiste | 1 | |
|
||
| Clement | | Baby |
|
||
+----------+-----+----------+
|
||
| Mme | | Lou |
|
||
| Louise | 28 | |
|
||
| Bourgeau | | Loue |
|
||
+----------+-----+----------+
|
||
''')
|
||
|
||
def test_texttable_header():
|
||
table = Texttable()
|
||
table.set_deco(Texttable.HEADER)
|
||
table.set_cols_dtype([
|
||
't', # text
|
||
'f', # float (decimal)
|
||
'e', # float (exponent)
|
||
'i', # integer
|
||
'a', # automatic
|
||
])
|
||
table.set_cols_align(["l", "r", "r", "r", "l"])
|
||
table.add_rows([
|
||
["text", "float", "exp", "int", "auto"],
|
||
["abcd", "67", 654, 89, 128.001],
|
||
["efghijk", 67.5434, .654, 89.6, 12800000000000000000000.00023],
|
||
["lmn", 5e-78, 5e-78, 89.4, .000000000000128],
|
||
["opqrstu", .023, 5e+78, 92., 12800000000000000000000],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
text float exp int auto
|
||
==============================================
|
||
abcd 67.000 6.540e+02 89 128.001
|
||
efghijk 67.543 6.540e-01 90 1.280e+22
|
||
lmn 0.000 5.000e-78 89 0.000
|
||
opqrstu 0.023 5.000e+78 92 1.280e+22
|
||
''')
|
||
|
||
def test_set_cols_width():
|
||
table = Texttable()
|
||
table.set_deco(Texttable.HEADER)
|
||
table.set_cols_width([10, 10])
|
||
table.add_rows([
|
||
["key", "value"],
|
||
[1, "a"],
|
||
[2, "b"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
key value
|
||
=======================
|
||
1 a
|
||
2 b
|
||
''')
|
||
|
||
def test_exceeding_max_width():
|
||
table = Texttable(max_width=35)
|
||
table.set_deco(Texttable.HEADER)
|
||
table.add_rows([
|
||
["key", "value"],
|
||
[1, "a"],
|
||
[2, "b"],
|
||
[3, "very long, very long, very long"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
key value
|
||
===================================
|
||
1 a
|
||
2 b
|
||
3 very long, very long, very
|
||
long
|
||
''')
|
||
|
||
def test_exceeding_max_width2():
|
||
table = Texttable(max_width=14)
|
||
table.add_rows([
|
||
["a", "b"],
|
||
[1, "+"],
|
||
[22, "++++++++"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
+----+-------+
|
||
| a | b |
|
||
+====+=======+
|
||
| 1 | + |
|
||
+----+-------+
|
||
| 22 | +++++ |
|
||
| | +++ |
|
||
+----+-------+
|
||
''')
|
||
|
||
def test_exceeding_max_width3():
|
||
table = Texttable()
|
||
table.set_max_width(35)
|
||
table.set_deco(Texttable.HEADER)
|
||
table.add_rows([
|
||
["key", "value"],
|
||
[1, "a"],
|
||
[2, "b"],
|
||
[3, "very long, very long, very long"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
key value
|
||
===================================
|
||
1 a
|
||
2 b
|
||
3 very long, very long, very
|
||
long
|
||
''')
|
||
|
||
def test_exceeding_max_width4():
|
||
table = Texttable()
|
||
table.set_max_width(14)
|
||
table.add_rows([
|
||
["a", "b"],
|
||
[1, "+"],
|
||
[22, "++++++++"],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
+----+-------+
|
||
| a | b |
|
||
+====+=======+
|
||
| 1 | + |
|
||
+----+-------+
|
||
| 22 | +++++ |
|
||
| | +++ |
|
||
+----+-------+
|
||
''')
|
||
|
||
def test_obj2unicode():
|
||
table = Texttable()
|
||
table.set_deco(Texttable.HEADER)
|
||
table.add_rows([
|
||
["key", "value"],
|
||
[1, "a"],
|
||
[2, 1],
|
||
[3, None],
|
||
])
|
||
assert clean(table.draw()) == dedent('''\
|
||
key value
|
||
===========
|
||
1 a
|
||
2 1
|
||
3 None
|
||
''')
|
||
|
||
def test_combining_char():
|
||
table = Texttable()
|
||
table.set_cols_align(["l", "r", "r"])
|
||
table.add_rows([
|
||
["str", "code-point\nlength", "display\nwidth"],
|
||
["ā", 2, 1],
|
||
["a", 1, 1],
|
||
])
|
||
assert clean(table.draw()) == u_dedent('''\
|
||
+-----+------------+---------+
|
||
| str | code-point | display |
|
||
| | length | width |
|
||
+=====+============+=========+
|
||
| ā | 2 | 1 |
|
||
+-----+------------+---------+
|
||
| a | 1 | 1 |
|
||
+-----+------------+---------+
|
||
''')
|
||
|
||
def test_combining_char2():
|
||
table = Texttable()
|
||
table.add_rows([
|
||
["a", "b", "c"],
|
||
["诶诶诶", "bbb", "西西西"],
|
||
], False)
|
||
assert clean(table.draw()) == u_dedent('''\
|
||
+--------+-----+--------+
|
||
| a | b | c |
|
||
+--------+-----+--------+
|
||
| 诶诶诶 | bbb | 西西西 |
|
||
+--------+-----+--------+
|
||
''')
|
||
|
||
|
||
def test_user_dtype():
|
||
table = Texttable()
|
||
|
||
table.set_cols_align(["l", "r", "r"])
|
||
table.set_cols_dtype([
|
||
'a', # automatic
|
||
lambda s:str(s)+"s", # user-def
|
||
lambda s:('%s'%s) if s>=0 else '[%s]'%(-s), # user-def
|
||
])
|
||
table.add_rows([
|
||
["str", "code-point\nlength", "display\nwidth"],
|
||
["a", 2, 1],
|
||
["a", 1,-3],
|
||
])
|
||
assert clean(table.draw()) == u_dedent('''\
|
||
+-----+------------+---------+
|
||
| str | code-point | display |
|
||
| | length | width |
|
||
+=====+============+=========+
|
||
| a | 2s | 1 |
|
||
+-----+------------+---------+
|
||
| a | 1s | [3] |
|
||
+-----+------------+---------+
|
||
''')
|
||
|
||
def test_cjkwarp():
|
||
try:
|
||
import cjkwrap
|
||
table = Texttable()
|
||
|
||
table.set_cols_align(["r", "l"])
|
||
table.add_rows([
|
||
["Name", 'Discuz! 6.x/7.x 全局变量防御绕过导致命令执行'],
|
||
["Description", '由于php5.3.x版本里php.ini的设置里request_order默认值为GP，导致Discuz! 6.x/7.x 全局变量防御绕过漏洞'],
|
||
], header = False)
|
||
assert clean(table.draw()) == u_dedent('''\
|
||
+-------------+----------------------------------------------------------------+
|
||
| Name | Discuz! 6.x/7.x 全局变量防御绕过导致命令执行 |
|
||
+-------------+----------------------------------------------------------------+
|
||
| Description | 由于php5.3.x版本里php.ini的设置里request_order默认值为GP，导致 |
|
||
| | Discuz! 6.x/7.x 全局变量防御绕过漏洞 |
|
||
+-------------+----------------------------------------------------------------+
|
||
''')
|
||
except ImportError:
|
||
True
|
||
|
||
def test_chaining():
|
||
table = Texttable()
|
||
table.reset()
|
||
table.set_max_width(50)
|
||
table.set_chars(list('-|+='))
|
||
table.set_deco(Texttable.BORDER)
|
||
table.set_header_align(list('lll'))
|
||
table.set_cols_align(list('lll'))
|
||
table.set_cols_valign(list('mmm'))
|
||
table.set_cols_dtype(list('ttt'))
|
||
table.set_cols_width([3, 3, 3])
|
||
table.set_precision(3)
|
||
table.header(list('abc'))
|
||
table.add_row(list('def'))
|
||
table.add_rows([list('ghi')], False)
|
||
s1 = table.draw()
|
||
s2 = (Texttable()
|
||
.reset()
|
||
.set_max_width(50)
|
||
.set_chars(list('-|+='))
|
||
.set_deco(Texttable.BORDER)
|
||
.set_header_align(list('lll'))
|
||
.set_cols_align(list('lll'))
|
||
.set_cols_valign(list('mmm'))
|
||
.set_cols_dtype(list('ttt'))
|
||
.set_cols_width([3, 3, 3])
|
||
.set_precision(3)
|
||
.header(list('abc'))
|
||
.add_row(list('def'))
|
||
.add_rows([list('ghi')], False)
|
||
.draw())
|
||
assert s1 == s2
|