def test_print_table_max_columns(self): table = Table(self.rows, self.columns) output = six.StringIO() table.print_table(max_columns=2, output=output) lines = output.getvalue().split('\n') self.assertEqual(len(lines), 8) self.assertEqual(len(lines[0]), 23)