Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for Army Battles by l.szyman10
# Taken from mission The Warriors
class Warrior:
def __init__(self):
self.health = 50
self.attack = 5
def __str__(self):
return f'{self.__class__.__name__}, health: {self.health}, attack: {self.attack}'
@property
def is_alive(self):
return self.health > 0
class Knight(Warrior):
def __init__(self):
super().__init__()
self.attack = 7
def fight(unit_1, unit_2):
while (unit_1.health > 0):
unit_2.health -= unit_1.attack
if unit_2.health <= 0:
return True
unit_1.health -= unit_2.attack
return False
class Army:
def __init__(self):
self.list_of_warriors = []
def add_units(self, unit, amount):
for _ in range(amount):
self.list_of_warriors.append(unit())
def unit_death(self):
self.list_of_warriors.pop()
class Battle:
def fight(self, attacking, defending):
while len(attacking.list_of_warriors) > 0:
if fight(attacking.list_of_warriors[-1], defending.list_of_warriors[-1]):
defending.unit_death()
else:
attacking.unit_death()
if len(defending.list_of_warriors) == 0:
return True
return False
if __name__ == '__main__':
# These "asserts" using only for self-checking and not necessary for auto-testing
chuck = Warrior()
bruce = Warrior()
carl = Knight()
dave = Warrior()
mark = Warrior()
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
print("Coding complete? Let's try tests!")
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()
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
# battle tests
my_army = Army()
my_army.add_units(Knight, 3)
enemy_army = Army()
enemy_army.add_units(Warrior, 3)
army_3 = Army()
army_3.add_units(Warrior, 20)
army_3.add_units(Knight, 5)
army_4 = Army()
army_4.add_units(Warrior, 30)
battle = Battle()
assert battle.fight(my_army, enemy_army) == True
assert battle.fight(army_3, army_4) == False
print("Coding complete? Let's try tests!")
March 30, 2020
Comments:
