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

Learn how to build clean, secure, and scalable applications with Object-Oriented Programming in Python. Cover the fundamentals of classes and objects, then moves into advanced concepts such as inheritance, composition, polymorphism, and encapsulation. With clear explanations and practical examples, you'll develop the ability to design Python programs that are powerful, maintainable, and ready for real-world use.

Learn the core building blocks of object-oriented programming in Python. This section covers classes, objects, methods, and variables, giving you a solid base to understand how OOP organizes and structures code.

Explore how classes can share and extend functionality. You'll learn different inheritance patterns, from simple hierarchies to multiple inheritance and abstract classes, and see how they enable code reuse and scalability.

Discover how to build flexible systems by combining objects. Instead of relying only on inheritance, you'll see how composition lets you assemble behaviors in a modular, adaptable way.

Understand how different objects can share a common interface while behaving in their own way. This section shows how method overriding, interfaces, and protocols make your code more dynamic and reusable.

Learn how to safeguard internal data and control access to class attributes. You'll explore private attributes, getters and setters, and data-hiding techniques that keep your classes secure and maintainable.

Challenge: Building a Composed Warrior

Solution

Challenge: Building a Composed Warrior

Solution