def test_four_times_table_format(self): table = [[1, 2, 3, 4], [2, 4, 6, 8], [3, 6, 9, 12], [4, 8, 12, 16]] self.assertEquals( list(times_table.table_lines(self.nums[:4], table)), [ " | 1 2 3 4", "--+-----------", "1 | 1 2 3 4", "2 | 2 4 6 8", "3 | 3 6 9 12", "4 | 4 8 12 16", ], )
def lines(gen, n): nums = list(gen(n)) table = times_table.table(nums) return times_table.table_lines(nums, table)
def test_two_times_table_format(self): table = [[1, 2], [2, 4]] self.assertEquals(list(times_table.table_lines(self.nums[:2], table)), [" |1 2", "--+---", "1 |1 2", "2 |2 4"])