python Python Class Iterator Square Root

Posted 2021-05-09

#!/usr/bin/env python

class Squared:
    def __init__(self):
        self.number = 0
    def __iter__(self):
        return self
    def __next__(self):
        self.number += 1
        return self.number * self.number
 
if __name__ == '__main__':
    squared = Squared()
    count = 0
 
    for number in squared:
        print(number)
        if count > 5:
            break
        count += 1