hp-oop-3-6.py

# hogwarts_magic.py

class Wizard:
    def __init__(self, name, patronus, birth_year):
        self.name = name
        self.patronus = patronus
        self.birth_year = birth_year
        self.house = None
        self.wand = None

    def assign_wand(self, wand):
        self.wand = wand
        print(f"{self.name} has a {self.wand} wand.")

    def assign_house(self, house):
        self.house = house
        house.add_member(self)

class House:
    # ...

    def add_member(self, member):
        self.members.append(member)

    # ...