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"])
