Write an iterator that outputs the value n number of times. If there is no n specified in the input arguments, then your iterator should act as an infinite iterator, always outputting the same element.

class Repeat(object):
    >>> r = Repeat(3, 5) # Will output 3, 5 times.
    >>> s = Repeat(3) # Will output 3 an infinite number of times.
    "***YOUR CODE HERE***"

Toggle Solution

class Repeat(object):

    def __init__(self, elem, n=None):
        self.elem = elem
        self.n = n
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.n is not None and self.index >= self.n:
            raise StopIteration
        self.index += 1
        return self.elem

The important thing to note in this question is the possibility of two different cases. On one hand, the iterator might be finite in which case you’d have to raise a StopIteration. On the other hand, the iterator might be infinite in which case you don’t have to raise the StopIteration.

I don't claim to be perfect so if you find an error on this page, please send me an email preferably with a link to this page so that I know what I need to fix!

comments powered by Disqus