import sieve_gen find = 100 gen = sieve_gen.sieve() for i in range(find - 1): found = next(gen) print found assert found == 541
import sieve_gen for i in sieve_gen.sieve(): print i if i >= 10: break assert i == 11
import sieve_gen for n, i in zip(range(20), sieve_gen.sieve()): print i