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
