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
|
from __future__ import annotations
from prettytable import PrettyTable, TableStyle
class TestRowEndSection:
EXPECTED_RESULT = """
┌──────────┬──────────┬──────────┐
│ Field 1 │ Field 2 │ Field 3 │
├──────────┼──────────┼──────────┤
│ value 4 │ value 5 │ value 6 │
│ value 7 │ value 8 │ value 9 │
├──────────┼──────────┼──────────┤
│ value 10 │ value 11 │ value 12 │
└──────────┴──────────┴──────────┘
""".strip()
TEST_ROWS = [
["value 4", "value 5", "value 6"],
["value 7", "value 8", "value 9"],
["value 10", "value 11", "value 12"],
]
def test_row_end_section_via_argument(self) -> None:
table = PrettyTable()
table.set_style(TableStyle.SINGLE_BORDER)
table.add_row(self.TEST_ROWS[0])
table.add_row(self.TEST_ROWS[1], divider=True)
table.add_row(self.TEST_ROWS[2])
assert table.get_string().strip() == self.EXPECTED_RESULT
def test_row_end_section_via_method(self) -> None:
table = PrettyTable()
table.set_style(TableStyle.SINGLE_BORDER)
table.add_row(self.TEST_ROWS[0])
table.add_row(self.TEST_ROWS[1])
table.add_divider()
table.add_row(self.TEST_ROWS[2])
assert table.get_string().strip() == self.EXPECTED_RESULT
def test_add_rows_divider(self) -> None:
"""A table created with two add_rows calls, one with divider=True has a
divider"""
table = PrettyTable()
table.set_style(TableStyle.SINGLE_BORDER)
table.add_rows(self.TEST_ROWS[0:2], divider=True)
table.add_rows(self.TEST_ROWS[2:])
assert table.get_string().strip() == self.EXPECTED_RESULT
class TestClearing:
def test_clear_rows(self, helper_table: PrettyTable) -> None:
helper_table.add_row([0, "a", "b", "c"], divider=True)
helper_table.clear_rows()
assert helper_table.rows == []
assert helper_table.dividers == []
assert helper_table.field_names == ["", "Field 1", "Field 2", "Field 3"]
def test_clear(self, helper_table: PrettyTable) -> None:
helper_table.add_row([0, "a", "b", "c"], divider=True)
helper_table.clear()
assert helper_table.rows == []
assert helper_table.dividers == []
assert helper_table.field_names == []
|
