def test_generate_triangle_numbers(self): generated = [] for i in euler.generate_triangle_numbers(): generated.append(i) if len(generated) == 7: break self.assertEqual(generated, [1, 3, 6, 10, 15, 21, 28])
def euler_problem12(): for number in euler.generate_triangle_numbers(): if euler.count_divisors(number) > 500: return number