CommencerCommencer gratuitement

Surcharge de l'égalité

Lorsque vous comparez deux objets d'une classe personnalisée à l'aide de ==, Python compare par défaut uniquement les références des objets, et non les données qu'ils contiennent. Pour remplacer ce comportement, la classe peut implémenter la méthode spéciale __eq__(), qui accepte deux arguments - les objets à comparer - et renvoie True ou False. Cette méthode sera implicitement appelée lorsque deux objets seront comparés.

La classe BankAccount du chapitre précédent est disponible dans le panneau de script. Il possède un attribut, balance, et une méthode withdraw(). Deux comptes bancaires ayant le même solde ne sont pas nécessairement le même compte, mais un compte bancaire a généralement un numéro de compte et deux comptes ayant le même numéro de compte doivent être considérés comme identiques.

Cet exercice fait partie du cours

Programmation orientée objet en Python

Afficher le cours

Instructions

Essayez de sélectionner le code des lignes 1 à 7 et d'appuyer sur le bouton "Exécuter le code". Essayez ensuite de créer quelques objets BankAccount dans la console et comparez-les.

  • Modifiez la méthode __init__() pour accepter un nouveau paramètre - number - et initialiser un nouvel attribut number.
  • Définissez une méthode __eq__() qui renvoie True si l'attribut number de deux objets est égal.
  • Examinez les instructions d'impression et la sortie dans la console.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

class BankAccount:
   # MODIFY to initialize a number attribute
    def __init__(self, balance=0):
        self.balance = balance
      
    def withdraw(self, amount):
        self.balance -= amount 
    
    # Define __eq__ that returns True if the number attributes are equal 
    def ____(____, ____):
        return ____.number == ____.____   

# Create accounts and compare them       
acct1 = BankAccount(123, 1000)
acct2 = BankAccount(123, 1000)
acct3 = BankAccount(456, 1000)
print(acct1 == acct2)
print(acct1 == acct3)
Modifier et exécuter le code