def test0044(self): from id_0044 import is_pentagonal l = [] for n in range(2, 20): l.append((3 * n - 1) * n / 2) # print l for n in xrange(2, l[17] + 1): self.assertEqual(is_pentagonal(n), n in l)