105 lines
2.2 KiB
Python
105 lines
2.2 KiB
Python
|
"""Test function in module."""
|
||
|
|
|
||
|
|
from terminaltables.build import build_row
|
||
|
|
|
||
|
|
|
||
|
|
def test_one_line():
|
||
|
|
"""Test with one line cells."""
|
||
|
|
row = [
|
||
|
|
['Left Cell'], ['Center Cell'], ['Right Cell'],
|
||
|
|
]
|
||
|
|
actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', 'Left Cell', '|', 'Center Cell', '|', 'Right Cell', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_two_line():
|
||
|
|
"""Test with two line cells."""
|
||
|
|
row = [
|
||
|
|
[
|
||
|
|
'Left ',
|
||
|
|
'Cell1',
|
||
|
|
],
|
||
|
|
|
||
|
|
[
|
||
|
|
'Center',
|
||
|
|
'Cell2 ',
|
||
|
|
],
|
||
|
|
|
||
|
|
[
|
||
|
|
'Right',
|
||
|
|
'Cell3',
|
||
|
|
],
|
||
|
|
]
|
||
|
|
actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', 'Left ', '|', 'Center', '|', 'Right', '<'),
|
||
|
|
('>', 'Cell1', '|', 'Cell2 ', '|', 'Cell3', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_three_line():
|
||
|
|
"""Test with three line cells."""
|
||
|
|
row = [
|
||
|
|
[
|
||
|
|
'Left ',
|
||
|
|
'Cell1',
|
||
|
|
' ',
|
||
|
|
],
|
||
|
|
|
||
|
|
[
|
||
|
|
'Center',
|
||
|
|
'Cell2 ',
|
||
|
|
' ',
|
||
|
|
],
|
||
|
|
|
||
|
|
[
|
||
|
|
'Right',
|
||
|
|
'Cell3',
|
||
|
|
' ',
|
||
|
|
],
|
||
|
|
]
|
||
|
|
actual = [tuple(i) for i in build_row(row, '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', 'Left ', '|', 'Center', '|', 'Right', '<'),
|
||
|
|
('>', 'Cell1', '|', 'Cell2 ', '|', 'Cell3', '<'),
|
||
|
|
('>', ' ', '|', ' ', '|', ' ', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_single():
|
||
|
|
"""Test with single cell."""
|
||
|
|
actual = [tuple(i) for i in build_row([['Cell']], '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', 'Cell', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_empty():
|
||
|
|
"""Test with empty cell."""
|
||
|
|
actual = [tuple(i) for i in build_row([['']], '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', '', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
|
||
|
|
def test_no_cells():
|
||
|
|
"""Test with no cells."""
|
||
|
|
actual = [tuple(i) for i in build_row([[]], '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|
||
|
|
|
||
|
|
actual = [tuple(i) for i in build_row([], '>', '|', '<')]
|
||
|
|
expected = [
|
||
|
|
('>', '<'),
|
||
|
|
]
|
||
|
|
assert actual == expected
|