Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for Army Battles by anvarvaliev01
# Taken from mission The Warriors
class Warrior:
def __init__(self):
self.health = 50
self.is_alive = True
def attack(self):
return 5
def get_damage(self, damage):
self.health -= int(damage)
class Knight(Warrior):
def attack(self):
return 7
def fight(unit_1, unit_2):
while unit_1.health > 0 and unit_2.health > 0:
unit_2.get_damage(unit_1.attack())
if unit_2.health <= 0:
unit_2.is_alive = False
break
unit_1.get_damage(unit_2.attack())
if unit_1.health <= 0:
unit_1.is_alive = False
break
return unit_1.health > 0 and unit_2.health <= 0
class Army(Knight, Warrior):
def add_units(self, warrior_class, number):
battler = warrior_class
self.army = [battler() for _ in range(number)]
class Battle(Knight, Warrior):
def fight(self, army_1, army_2):
while True:
if fight(army_1.army[0], army_2.army[0]) == True:
del army_2.army[0]
if len(army_2.army) == 0: break
else:
del army_1.army[0]
if len(army_1.army) == 0: break
return len(army_1.army) > len(army_2.army)
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!")
Aug. 10, 2021