def createGenerator() :
    mylist = range(3)
    for i in mylist :
        yield i*i

gen = createGenerator()
print(gen)

for i in gen:
    print(i)