Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Uncategorized category for Army Battles by haulef
# Taken from mission The Warriors
class Warrior:
def __init__(self, health=50, attack=5):
self.health = health
self.attack = attack
@property
def is_alive(self):
return self.health > 0
pass
class Knight(Warrior):
def __init__(self, attack=7):
super().__init__(attack=attack)
pass
class Army(Knight, Warrior):
'''
adds the chosen amount of units to army.
first unit added, is the first unit to go fight
'''
def __init__(self):
self.units = list()
def add_units(self,unit_type, num):
for _ in range(num):
self.units.append(unit_type())
pass
class Battle:
'''
determines the strongest army
'''
def fight(self, unit_1, unit_2):
#i = j = 1 #this is for printing
while unit_1.units and unit_2.units:
unit_1_curr = unit_1.units[0]
unit_2_curr = unit_2.units[0]
#print(f'[in] --> unit1: {i},{unit_1_curr.health}, unit2:{j},{unit_2_curr.health}')
winner = fight(unit_1_curr, unit_2_curr)
#print(f'[out] --> unit1: {i},{unit_1_curr.health}, unit2:{j},{unit_2_curr.health}')
if winner:
unit_2.units.pop(0)
#j += 1 #this is for printing
else:
unit_1.units.pop(0)
#i+=1 #this is for printing
return True if unit_1.units else False
def fight(unit_1, unit_2):
#print('in fight function')
while unit_1.is_alive and unit_2.is_alive:
unit_2.health -= unit_1.attack
if unit_2.is_alive:
unit_1.health -= unit_2.attack
return unit_1.is_alive
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()
#print(isinstance(my_army.units[0], Knight))
#print(type(my_army.units[0]))
#print(my_army.units[0])
assert battle.fight(my_army, enemy_army) == True
assert battle.fight(army_3, army_4) == False
print("Coding complete? Let's try tests!")
March 6, 2020
