def test_create_table_with_default_truthy_out_of_range_column(headers, rows):
with pytest.raises(ValueError) as exc:
_ = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
truthy=5,
)
assert 'The column specified for truthy values does not exist. Column: 5' in str(
exc.value)
def test_create_table_with_default_truthy_not_enough_colors(headers, rows):
with pytest.raises(ValueError) as exc:
_ = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
colors=[pretty_tables.Colors.cyan],
truthy=3,
)
assert 'When using the truthy option, you must specify two colors, or no colors' in str(
exc.value)
def test_create_table(headers, rows):
output = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
)
# fmt: off
assert output == ('| ID | Name | Occupation | Employed |\n'
'| -- | ------ | ----------------- | -------- |\n'
'| 1 | Justin | Software Engineer | True |\n'
'| 2 | Misty | Receptionist | False |\n'
'| 3 | John | No data | False |')
def test_create_table_with_colors(headers, rows, colors):
output = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
colors=colors,
)
# fmt: off
assert output == (
'| \x1b[94mID\x1b[0m | \x1b[95mName \x1b[0m | \x1b[1mOccupation \x1b[0m | \x1b[92mEmployed\x1b[0m |\n'
'| -- | ------ | ----------------- | -------- |\n'
'| \x1b[94m1 \x1b[0m | \x1b[95mJustin\x1b[0m | \x1b[1mSoftware Engineer\x1b[0m | \x1b[92mTrue \x1b[0m |\n'
'| \x1b[94m2 \x1b[0m | \x1b[95mMisty \x1b[0m | \x1b[1mReceptionist \x1b[0m | \x1b[92mFalse \x1b[0m |\n'
'| \x1b[94m3 \x1b[0m | \x1b[95mJohn \x1b[0m | \x1b[1mNo data \x1b[0m | \x1b[92mFalse \x1b[0m |'
)
def test_create_table_with_default_truthy(headers, rows):
output = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
truthy=3,
)
# fmt: off
assert output == (
'| ID\x1b[0m | Name \x1b[0m | Occupation \x1b[0m | Employed\x1b[0m |\n'
'| -- | ------ | ----------------- | -------- |\n'
'| \x1b[92m1 \x1b[0m | \x1b[92mJustin\x1b[0m | \x1b[92mSoftware Engineer\x1b[0m | \x1b[92mTrue \x1b[0m |\n'
'| \x1b[91m2 \x1b[0m | \x1b[91mMisty \x1b[0m | \x1b[91mReceptionist \x1b[0m | \x1b[91mFalse \x1b[0m |\n'
'| \x1b[91m3 \x1b[0m | \x1b[91mJohn \x1b[0m | \x1b[91mNo data \x1b[0m | \x1b[91mFalse \x1b[0m |'
)
def test_create_table_with_custom_truthy_colors(headers, rows):
output = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='No data',
colors=[pretty_tables.Colors.cyan, pretty_tables.Colors.purple],
truthy=3,
)
# fmt: off
assert output == (
'| ID\x1b[0m | Name \x1b[0m | Occupation \x1b[0m | Employed\x1b[0m |\n'
'| -- | ------ | ----------------- | -------- |\n'
'| \x1b[96m1 \x1b[0m | \x1b[96mJustin\x1b[0m | \x1b[96mSoftware Engineer\x1b[0m | \x1b[96mTrue \x1b[0m |\n'
'| \x1b[95m2 \x1b[0m | \x1b[95mMisty \x1b[0m | \x1b[95mReceptionist \x1b[0m | \x1b[95mFalse \x1b[0m |\n'
'| \x1b[95m3 \x1b[0m | \x1b[95mJohn \x1b[0m | \x1b[95mNo data \x1b[0m | \x1b[95mFalse \x1b[0m |'
)
def generate_table(formulas: List) -> str:
"""Generates a pretty table which will be used in the README file."""
headers = ['Project', 'Description', 'Install']
rows = []
for formula in formulas:
rows.append(
[
f'[{formula["name"]}]({formula.get("homepage")})',
formula.get('desc'),
f'`brew install {formula["name"]}`',
]
)
table = pretty_tables.create(
headers=headers,
rows=rows,
empty_cell_placeholder='NA',
)
return table