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
|
"""Test method in BaseTable class."""
import pytest
from terminaltables.base_table import BaseTable
from terminaltables.width_and_alignment import max_dimensions
SINGLE_LINE = (
('Name', 'Color', 'Type'),
('Avocado', 'green', 'nut'),
('Tomato', 'red', 'fruit'),
('Lettuce', 'green', 'vegetable'),
)
@pytest.mark.parametrize('inner_column_border', [True, False])
@pytest.mark.parametrize('style', ['top', 'bottom'])
def test_top_bottom(inner_column_border, style):
"""Test top and bottom borders.
:param bool inner_column_border: Passed to table class.
:param str style: Passed to method.
"""
table = BaseTable(SINGLE_LINE, 'Example')
table.inner_column_border = inner_column_border
outer_widths = max_dimensions(table.table_data, table.padding_left, table.padding_right)[2]
# Determine expected.
if style == 'top' and inner_column_border:
expected = '+Example--+-------+-----------+'
elif style == 'top':
expected = '+Example--------------------+'
elif style == 'bottom' and inner_column_border:
expected = '+---------+-------+-----------+'
else:
expected = '+---------------------------+'
# Test.
actual = ''.join(table.horizontal_border(style, outer_widths))
assert actual == expected
@pytest.mark.parametrize('inner_column_border', [True, False])
@pytest.mark.parametrize('outer_border', [True, False])
@pytest.mark.parametrize('style', ['heading', 'footing'])
def test_heading_footing(inner_column_border, outer_border, style):
"""Test heading and footing borders.
:param bool inner_column_border: Passed to table class.
:param bool outer_border: Passed to table class.
:param str style: Passed to method.
"""
table = BaseTable(SINGLE_LINE)
table.inner_column_border = inner_column_border
table.outer_border = outer_border
outer_widths = max_dimensions(table.table_data, table.padding_left, table.padding_right)[2]
# Determine expected.
if style == 'heading' and outer_border:
expected = '+---------+-------+-----------+' if inner_column_border else '+---------------------------+'
elif style == 'heading':
expected = '---------+-------+-----------' if inner_column_border else '---------------------------'
elif style == 'footing' and outer_border:
expected = '+---------+-------+-----------+' if inner_column_border else '+---------------------------+'
else:
expected = '---------+-------+-----------' if inner_column_border else '---------------------------'
# Test.
actual = ''.join(table.horizontal_border(style, outer_widths))
assert actual == expected
@pytest.mark.parametrize('inner_column_border', [True, False])
@pytest.mark.parametrize('outer_border', [True, False])
def test_row(inner_column_border, outer_border):
"""Test inner borders.
:param bool inner_column_border: Passed to table class.
:param bool outer_border: Passed to table class.
"""
table = BaseTable(SINGLE_LINE)
table.inner_column_border = inner_column_border
table.outer_border = outer_border
outer_widths = max_dimensions(table.table_data, table.padding_left, table.padding_right)[2]
# Determine expected.
if inner_column_border and outer_border:
expected = '+---------+-------+-----------+'
elif inner_column_border:
expected = '---------+-------+-----------'
elif outer_border:
expected = '+---------------------------+'
else:
expected = '---------------------------'
# Test.
actual = ''.join(table.horizontal_border('row', outer_widths))
assert actual == expected
|
