You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

118 lines
2.9 KiB

# Taken from mission Army Battles
# Taken from mission The Warriors
class Warrior:
def __init__(self):
self.health = 50
self.attack = 5
self.is_alive = True
self.defense = 0
def hit(self, other_unit) -> bool:
hit_points = self.attack - other_unit.defense
hit_points = hit_points if hit_points > 0 else 0
other_unit.health -= hit_points
other_unit.is_alive = other_unit.health > 0
return other_unit.is_alive
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.defense = 2
class Rookie(Warrior):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.health = 50
self.attack = 1
def fight(unit_1, unit_2):
while unit_1.is_alive and unit_2.is_alive:
if not unit_1.hit(unit_2):
return True
if not unit_2.hit(unit_1):
return False
class Army:
def __init__(self):
self.units = []
def add_units(self, unit_type, quantity):
for i in range(quantity):
self.units.append(unit_type())
class Battle:
def fight(self, army_1: Army, army_2: Army) -> bool:
while len(army_1.units) > 0 and len(army_2.units) > 0:
current_unit_1 = army_1.units[0]
current_unit_2 = army_2.units[0]
if fight(current_unit_1, current_unit_2):
army_2.units.pop(0)
else:
army_1.units.pop(0)
return len(army_1.units) > 0
if __name__ == '__main__':
# These "asserts" using only for self-checking and not necessary for auto-testing
# 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!")
unit_1 = Defender()
unit_2 = Rookie()
fight(unit_1, unit_2)
assert unit_1.health == 60