Classe itérable : Compteur

Pour qu'un objet soit utilisable dans une boucle for, il doit respecter le protocole itérable. Ça veut dire qu'il doit avoir une méthode __iter__() qui renvoie un itérateur. Et l'itérateur doit avoir une méthode __next__() qui renvoie la valeur suivante ou lève StopIteration quand c'est fini.

Le plus simple, c'est que la même classe fasse les deux : __iter__() renvoie self, et __next__() gère la séquence de valeurs.

Par exemple, voici comment Python utilise ton objet dans un for :

obj = MonObjet()
it = iter(obj) appelle obj.__iter__()
next(it) appelle it.__next__()
next(it) appelle it.__next__()
... jusqu'à StopIteration

Écris une classe Compteur qui prend une valeur de départ et une valeur de fin. Quand on itère dessus, elle produit les entiers de debut à fin inclus.

Exemple :
list(Compteur(1, 5)) renvoie [1, 2, 3, 4, 5]
list(Compteur(3, 3)) renvoie [3]

Tests (3/4)

Compteur 1 à 5

assert list(Compteur(1, 5)) == [1, 2, 3, 4, 5]

Compteur 3 à 3

assert list(Compteur(3, 3)) == [3]

Utilisable dans un for

total = 0
for x in Compteur(1, 3):
    total += x
assert total == 6

+ 0 tests cachés

Indices (3 disponibles)

Solution officielle

class Compteur:
    def __init__(self, debut, fin):
        self.current = debut
        self.fin = fin

    def __iter__(self):
        return self

    def __next__(self):
        if self.current > self.fin:
            raise StopIteration
        val = self.current
        self.current += 1
        return val

← Premiers pas avec iter et … Itérateur de nombres pairs →

solution.py

Bravo!

Classe itérable : Compteur

Tests (3/4)

Indices (3 disponibles)

Solution officielle