def test_generate_next_line(self):
for n in (10, 100, 1000, 10000):
expected = [
a + b for a, b in zip(chain(zero(), pascal(n)),
chain(pascal(n), zero()))
]
result = list(pascal(n + 1))
self.assertEqual(result, expected)
def test_n1(self):
self.assertEqual(list(pascal(1)), [1, 1])
def test_n0(self):
self.assertEqual(list(pascal(0)), [1])
def test_n5(self):
self.assertEqual(list(pascal(5)), [1, 4, 6, 4, 1])
def test_alternate_sum(self):
for n in (10, 100, 1000, 10000):
self.assertEqual(sum(alternate(pascal(n))), 0)
def test_sum(self):
for n in (10, 100, 1000, 10000):
self.assertEqual(sum(pascal(n)), 2**n)
def test_n5(self):
self.assertEqual(list(pascal(5)), [1, 5, 10, 10, 5, 1])
