inf_seq.py

def infinite_seq():
    count = 1

    while True:
        
        yield count ** 2
        count = count + 1

infinite_stream = infinite_seq()

print( next(infinite_stream) )
print( next(infinite_stream) )
print( next(infinite_stream) )