Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Clear category for The Defenders by aya.kanazawa
# Taken from mission Army Battles
class Warrior:
def __init__(self):
self.health = 50
self.attack = 5
self.defence = 0
self.is_alive = True
def attacked(self, attacker):
self.health -= (attacker.attack - self.defence) if attacker.attack > self.defence else 0
if self.health <= 0:
self.is_alive = False
class Knight(Warrior):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.attack = 7
class Defender(Warrior):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.health = 60
self.attack = 3
self.defence = 2
class Rookie(Warrior):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.health = 50
self.attack = 1
def fight(unit_1, unit_2):
while unit_1.is_alive and unit_2.is_alive:
unit_2.attacked(unit_1)
if not unit_2.is_alive:
break
unit_1.attacked(unit_2)
return unit_1.is_alive and not unit_2.is_alive
class Army:
def __init__(self):
self.units = []
def add_units(self, ones_class, number):
for i in range(number):
self.units.append(ones_class())
def is_someone_alive(self):
for one in self.units:
if one.is_alive:
return True
return False
class Battle:
def fight(self, army1, army2):
index1 = 0
index2 = 0
while army1.is_someone_alive() and army2.is_someone_alive():
if fight(army1.units[index1], army2.units[index2]):
index2 += 1
else:
index1 += 1
return army1.is_someone_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
# 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!")
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()
bob = Defender()
mike = Knight()
rog = Warrior()
lancelot = Defender()
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
assert fight(bob, mike) == False
assert fight(lancelot, rog) == True
#battle tests
my_army = Army()
my_army.add_units(Defender, 1)
enemy_army = Army()
enemy_army.add_units(Warrior, 2)
army_3 = Army()
army_3.add_units(Warrior, 1)
army_3.add_units(Defender, 1)
army_4 = Army()
army_4.add_units(Warrior, 2)
battle = Battle()
assert battle.fight(my_army, enemy_army) == False
assert battle.fight(army_3, army_4) == True
print("Coding complete? Let's try tests!")
unit_1 = Defender()
unit_2 = Rookie()
fight(unit_1, unit_2)
print(unit_1.health)
May 20, 2019
