blob: 37e86bfe2bedbc45fa2a7cef8b2364af7e71514a (
plain)
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
|
import itertools
from rich.style import Style
from rich.table import _Cell
from rich.tabulate import tabulate_mapping
def test_tabulate_mapping():
# TODO: tabulate_mapping may not be needed shortly
table = tabulate_mapping({"foo": "1", "bar": "2"})
assert len(table.columns) == 2
assert len(table.columns[0]._cells) == 2
assert len(table.columns[1]._cells) == 2
# add tests for title and caption justification
test_title = "Foo v. Bar"
test_caption = "approximate results"
for title_justify, caption_justify in itertools.product(
[None, "left", "center", "right"], repeat=2
):
table = tabulate_mapping(
{"foo": "1", "bar": "2"},
title=test_title,
caption=test_caption,
title_justify=title_justify,
caption_justify=caption_justify,
)
expected_title_justify = (
title_justify if title_justify is not None else "center"
)
expected_caption_justify = (
caption_justify if caption_justify is not None else "center"
)
assert expected_title_justify == table.title_justify
assert expected_caption_justify == table.caption_justify
|
