import unittest
import importlib
import re
import io
import contextlib


class TestTask(unittest.TestCase):
    def test_classes_exist(self):
        import user_code
        importlib.reload(user_code)

        Weapon = getattr(user_code, "Weapon", None)
        Warrior = getattr(user_code, "Warrior", None)

        _dynamic_test(
            self,
            Weapon is not None and Warrior is not None,
            "Weapon and Warrior classes exist",
            "Weapon and Warrior classes missing",
        )

    def test_weapon_init(self):
        import user_code
        importlib.reload(user_code)

        Weapon = getattr(user_code, "Weapon", None)
        sword = Weapon("Sword", 5)

        _dynamic_test(
            self,
            hasattr(sword, "name") and hasattr(sword, "attack"),
            "Weapon has name and attack attributes",
            "Weapon missing name or attack attribute",
        )
        _dynamic_test(
            self,
            sword.name == "Sword" and sword.attack == 5,
            "Weapon stores name and attack correctly",
            f"Expected ('Sword', 5), got ({getattr(sword,'name',None)}, {getattr(sword,'attack',None)})",
        )

    def test_warrior_init(self):
        import user_code
        importlib.reload(user_code)

        Warrior = getattr(user_code, "Warrior", None)
        w = Warrior("Conan")

        _dynamic_test(
            self,
            hasattr(w, "name") and w.name == "Conan",
            "Warrior stores name correctly",
            "Warrior missing name or stores it incorrectly",
        )
        _dynamic_test(
            self,
            hasattr(w, "weapon") and w.weapon is None,
            "Warrior starts unarmed (weapon is None)",
            "Warrior weapon should start as None",
        )

    def test_equip_weapon_composition(self):
        import user_code
        importlib.reload(user_code)

        Weapon = getattr(user_code, "Weapon", None)
        Warrior = getattr(user_code, "Warrior", None)

        sword = Weapon("Sword", 5)
        w = Warrior("Conan")
        w.equip_weapon(sword)

        _dynamic_test(
            self,
            hasattr(w, "weapon") and w.weapon is sword,
            "Warrior equips weapon via composition",
            "Weapon not equipped correctly (should store the same object)",
        )

    def test_show_weapon_output_unarmed(self):
        import user_code
        importlib.reload(user_code)

        Warrior = getattr(user_code, "Warrior", None)
        w = Warrior("Conan")

        out = _capture_print(w.show_weapon)
        expected = normalize_text("Conan is unarmed")

        _dynamic_test(
            self,
            normalize_text(out) == expected,
            "show_weapon prints unarmed message",
            f"Expected '{expected}', got '{normalize_text(out)}'",
        )

    def test_show_weapon_output_armed(self):
        import user_code
        importlib.reload(user_code)

        Weapon = getattr(user_code, "Weapon", None)
        Warrior = getattr(user_code, "Warrior", None)

        sword = Weapon("Sword", 5)
        w = Warrior("Conan")
        w.equip_weapon(sword)

        out = _capture_print(w.show_weapon)
        expected = normalize_text("Conan is holding a Sword")

        _dynamic_test(
            self,
            normalize_text(out) == expected,
            "show_weapon prints armed message",
            f"Expected '{expected}', got '{normalize_text(out)}'",
        )


def _capture_print(func, *args, **kwargs) -> str:
    buf = io.StringIO()
    with contextlib.redirect_stdout(buf):
        func(*args, **kwargs)
    return buf.getvalue().strip()


def _dynamic_test(test_case, condition, success_message, failure_message):
    if condition:
        test_case._testMethodName = success_message
        test_case.assertTrue(True, success_message)
    else:
        test_case._testMethodName = failure_message
        test_case.fail(failure_message)


def normalize_text(text: str) -> str:
    text = text.strip().lower()
    text = re.sub(r"\s{2,}", " ", text)
    return text


if __name__ == "__main__":
    unittest.main()

test_main.py

Apprenez à créer des applications propres, sécurisées et évolutives avec la programmation orientée objet en Python. Couvrez les fondamentaux des classes et des objets, puis passez à des concepts avancés tels que l'héritage, la composition, le polymorphisme et l'encapsulation. Grâce à des explications claires et des exemples pratiques, vous développerez la capacité de concevoir des programmes Python puissants, maintenables et prêts pour une utilisation réelle.

Découvrez les éléments fondamentaux de la programmation orientée objet en Python. Cette section aborde les classes, les objets, les méthodes et les variables, vous offrant une base solide pour comprendre comment la POO organise et structure le code.

Découvrez comment les classes peuvent partager et étendre des fonctionnalités. Vous étudierez différents modèles d'héritage, des hiérarchies simples à l'héritage multiple et aux classes abstraites, et verrez comment ils permettent la réutilisation du code et l'évolutivité.

Découvrez comment créer des systèmes flexibles en combinant des objets. Plutôt que de s'appuyer uniquement sur l'héritage, vous verrez comment la composition permet d'assembler des comportements de manière modulaire et adaptable.

Comprendre comment différents objets peuvent partager une interface commune tout en ayant des comportements propres. Cette section montre comment la redéfinition de méthodes, les interfaces et les protocoles rendent le code plus dynamique et réutilisable.

Découvrez comment protéger les données internes et contrôler l'accès aux attributs de classe. Vous étudierez les attributs privés, les accesseurs et mutateurs, ainsi que les techniques de masquage des données qui assurent la sécurité et la maintenabilité de vos classes.

Challenge: Construire un Guerrier Composé

Solution