def test_factors(self): self.assertEqual([1], numbers.factors(1)) self.assertEqual([1, 3], numbers.factors(3)) self.assertEqual([1, 2, 3, 6], numbers.factors(6)) self.assertEqual([1, 2, 5, 10], numbers.factors(10)) self.assertEqual([1, 3, 5, 15], numbers.factors(15)) self.assertEqual([1, 3, 7, 21], numbers.factors(21)) self.assertEqual([1, 2, 4, 7, 14, 28], numbers.factors(28))
def problem012(n): i = 1 while len(numbers.factors(numbers.triangle_number(i))) < n: i = i + 1 return numbers.triangle_number(i)