def test_print_columns_2(mock, capsys): columns = columnize(['a', 'b', 'ccc', 'dddd']) assert '\n'.join(columns) == "a \nb \nccc \ndddd"
def test_print_columns_3(mock, capsys): columns = list(columnize(['aaa', 'b', 'c', 'd', 'e', 'fff'])) assert max(map(len, columns)) <= 9 assert '\n'.join(columns) == "aaa d \nb e \nc fff"