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
|
"""Test method in BaseTable class."""
import pytest
from terminaltables3.base_table import BaseTable
from terminaltables3.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
|
