def test_format_max_columns(self): table = Table(self.rows, self.columns) output = table.format(max_columns=2) lines = output.split('\n') self.assertEqual(len(lines), 7) self.assertEqual(len(lines[0]), 22)