Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for Army Battles by tigerhu3180
# Taken from mission The Warriors
class Warrior:
def __init__(self,health=50,attack=5):
self.health=health
self.attack=attack
self.is_alive=True
class Knight(Warrior):
def __init__(self):
super().__init__(attack=7)
def fight(unit_1, unit_2):
flag=1
while unit_1.health>0 and unit_2.health>0:
if flag:
unit_2.health-=unit_1.attack
if unit_2.health<=0:unit_2.is_alive=False
flag=0
else:
unit_1.health-=unit_2.attack
if unit_1.health<=0:unit_1.is_alive=False
flag=1
return unit_1.health>0
class Army:
def __init__(self):
self.array=[]
def add_units(self,unit,number):
while number>0:
self.array.append(unit())
number-=1
class Battle():
def fight(self,group1,group2):
unit1,unit2=group1.array[0],group2.array[0]
fight(unit1,unit2)
while True:
if unit1.is_alive:
group2.array.remove(unit2)
if len(group2.array)==0:
return True
unit2=group2.array[0]
else:
group1.array.remove(unit1)
if len(group1.array)==0:
return False
unit1=group1.array[0]
fight(unit1,unit2)
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. 12, 2018