generator = sequence_generator(3) try: print(next(generator)) print(next(generator)) print(next(generator)) print(next(generator)) # This call to next will raise an exception. except StopIteration: print("Generator exhausted") >>> 0 >>> 1 >>> 2 >> Generator exhausted