def test_prime():
g = gen.prime()
for val in [
2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61,
67, 71, 73, 79, 83, 89, 97
]:
assert next(g) == val
def test_prime(self):
gen = g.prime()
seq = [2, 3, 5, 7, 11, 13, 17, 19, 23]
for s in seq:
self.assertEqual(next(gen), s)