def test_prepare_max_column_widths_given_max_column_widths_is_string_returns_max_col_widths_string(
):
expected_widths = "(30, 10)"
max_column_widths = Table.prepare_max_column_widths(["a", "b", "c"],
"(30, 10)", None)
assert max_column_widths == expected_widths
def test_prepare_max_column_widths_given_max_column_widths_is_none_returns_infinity(
):
expected_widths = [float('inf'), float('inf'), float('inf')]
max_column_widths = Table.prepare_max_column_widths(["a", "b", "c"], None,
None)
assert max_column_widths == expected_widths
def test_prepare_max_column_widths_given_max_column_widths_is_set_returns_max_col_widths_as_list(
):
expected_widths = [30, 10]
max_column_widths = Table.prepare_max_column_widths(["a", "b", "c"],
(30, 10), None)
assert max_column_widths == expected_widths
def test_prepare_max_column_widths_given_max_column_widths_is_none_returns_max_col_widths_of_max_length(
):
expected_widths = [20, 20, 20]
max_column_widths = Table.prepare_max_column_widths(["a", "b", "c"], None,
20)
assert max_column_widths == expected_widths