def test_triangle_number(self): self.assertEqual(1, numbers.triangle_number(1)) self.assertEqual(3, numbers.triangle_number(2)) self.assertEqual(6, numbers.triangle_number(3)) self.assertEqual(10, numbers.triangle_number(4)) self.assertEqual(15, numbers.triangle_number(5)) self.assertEqual(21, numbers.triangle_number(6)) self.assertEqual(28, numbers.triangle_number(7)) self.assertEqual(36, numbers.triangle_number(8)) self.assertEqual(45, numbers.triangle_number(9)) self.assertEqual(55, numbers.triangle_number(10))
def problem012(n): i = 1 while len(numbers.factors(numbers.triangle_number(i))) < n: i = i + 1 return numbers.triangle_number(i)