Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for The Defenders by Magu
class Warrior:
def __init__(self):
self.health = 50
self.attack = 5
self.is_alive = True
class Knight(Warrior):
def __init__(self):
super().__init__()
self.attack = 7
class Defender(Warrior):
def __init__(self):
super().__init__()
self.health = 60
self.attack = 3
self.defence = 2
class Army:
def __init__(self):
self.list = []
def add_units(self, unit, num):
for i in range(num):
self.list.append(unit())
class Battle:
def fight(self, army1, army2):
while army1.list and army2.list:
if bl := fight(army1.list[0], army2.list[0]):
army2.list.pop(0)
else:
army1.list.pop(0)
return bl # bool(army1.list)
def fight(unit_1, unit_2):
while True:
if isinstance(unit_2, Defender):
dmg = max(0, unit_1.attack - unit_2.defence)
unit_2.health -= dmg
else:
unit_2.health -= unit_1.attack
if unit_2.health <= 0:
unit_2.is_alive = False; break
if isinstance(unit_1, Defender):
dmg = max(0, unit_2.attack - unit_1.defence)
unit_1.health -= dmg
else:
unit_1.health -= unit_2.attack
if unit_1.health <= 0:
unit_1.is_alive = False; break
return unit_1.is_alive
if __name__ == "__main__":
# fight tests
chuck = Warrior()
bruce = Warrior()
carl = Knight()
dave = Warrior()
mark = Warrior()
bob = Defender()
mike = Knight()
rog = Warrior()
lancelot = Defender()
assert fight(chuck, bruce) == True
assert fight(dave, carl) == False
assert chuck.is_alive == True
assert bruce.is_alive == False
assert carl.is_alive == True
assert dave.is_alive == False
assert fight(carl, mark) == False
assert carl.is_alive == False
assert fight(bob, mike) == False
assert fight(lancelot, rog) == True
# battle tests
my_army = Army()
my_army.add_units(Defender, 1)
enemy_army = Army()
enemy_army.add_units(Warrior, 2)
army_3 = Army()
army_3.add_units(Warrior, 1)
army_3.add_units(Defender, 1)
army_4 = Army()
army_4.add_units(Warrior, 2)
battle = Battle()
assert battle.fight(my_army, enemy_army) == False
assert battle.fight(army_3, army_4) == True
print("Coding complete? Let's try tests!")
April 3, 2025
