38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Test the vertical table formatter."""
|
|
|
|
from textwrap import dedent
|
|
|
|
from cli_helpers.compat import text_type
|
|
from cli_helpers.tabular_output import vertical_table_adapter
|
|
|
|
|
|
def test_vertical_table():
|
|
"""Test the default settings for vertical_table()."""
|
|
results = [('hello', text_type(123)), ('world', text_type(456))]
|
|
|
|
expected = dedent("""\
|
|
***************************[ 1. row ]***************************
|
|
name | hello
|
|
age | 123
|
|
***************************[ 2. row ]***************************
|
|
name | world
|
|
age | 456""")
|
|
assert expected == "\n".join(
|
|
vertical_table_adapter.adapter(results, ('name', 'age')))
|
|
|
|
|
|
def test_vertical_table_customized():
|
|
"""Test customized settings for vertical_table()."""
|
|
results = [('john', text_type(47)), ('jill', text_type(50))]
|
|
|
|
expected = dedent("""\
|
|
-[ PERSON 1 ]-----
|
|
name | john
|
|
age | 47
|
|
-[ PERSON 2 ]-----
|
|
name | jill
|
|
age | 50""")
|
|
assert expected == "\n".join(vertical_table_adapter.adapter(
|
|
results, ('name', 'age'), sep_title='PERSON {n}',
|
|
sep_character='-', sep_length=(1, 5)))
|