Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for Army Battles by liw_80
class Warrior:
def __init__(self):
self.is_alive = True
self.attack = 5
self.health = 50
class Knight(Warrior):
def __init__(self):
super().__init__()
self.attack = 7
class Army:
def __init__(self):
self.my_army = []
pass
def add_units(self, role, num):
self.my_army.extend([role() for i in range(num)])
class Battle:
def fight(self, unit_1, unit_2):
while len(unit_1.my_army) > 0 and len(unit_2.my_army) > 0:
res, last_health = fight(unit_1.my_army[0], unit_2.my_army[0], True)
if res:
unit_1.my_army[0].health = last_health
unit_2.my_army.pop(0)
else:
unit_2.my_army[0].health = last_health
unit_1.my_army.pop(0)
return True if len(unit_1.my_army) > 0 else False
def fight(unit_1, unit_2, isArmy = False):
i = 0
while unit_1.is_alive and unit_2.is_alive:
if i % 2 == 0:
unit_2.health -= unit_1.attack
else:
unit_1.health -= unit_2.attack
unit_1.is_alive = True if unit_1.health > 0 else False
unit_2.is_alive = True if unit_2.health > 0 else False
i += 1
return (True if unit_1.is_alive else False) if not isArmy else \
(True, unit_1.health) if unit_1.is_alive else (False, unit_2.health)
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!")
Jan. 13, 2020
