Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Speedy category for Army Battles by darkera
class Army():
def __init__(self):
self.units = []
def add_units(self, unit,cnt):
for i in range(0,cnt):
self.units.append(unit())
class Warrior:
def __init__(self, health=50, attack=5):
self.health = health
self.attack = attack
@property
def is_alive(self):
return True if self.health > 0 else False
def bit(self, hit):
self.health -= hit
class Knight(Warrior):
def __init__(self):
Warrior.__init__(self, attack=7)
class Battle():
def fight(self,army1,army2):
i=0
j=0
while army1.units[-1].is_alive and army2.units[-1].is_alive:
if fight(army1.units[i],army2.units[j]): j+=1
else: i+=1
if army1.units[-1].is_alive:return True
else: return False
def fight(unit_1, unit_2):
while unit_1.is_alive and unit_2.is_alive:
unit_2.bit(unit_1.attack)
if unit_2.is_alive == False: return True
unit_1.bit(unit_2.attack)
if unit_1.is_alive == False: return False
if __name__ == '__main__':
#These "asserts" using only for self-checking and not necessary for auto-testing
chuck = Warrior()
bruce = Warrior()
fight(chuck,bruce)
print( chuck.is_alive)
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. 18, 2021
