189 lines
5.2 KiB
Python
189 lines
5.2 KiB
Python
# Taken from mission The Lancers
|
|
|
|
# Taken from mission The Vampires
|
|
# Taken from mission The Defenders
|
|
# Taken from mission Army Battles
|
|
# Taken from mission The Warriors
|
|
|
|
|
|
class Warrior:
|
|
def __init__(self):
|
|
self.health = 50
|
|
self.attack = 5
|
|
self.is_alive = True
|
|
self.defense = 0
|
|
self.vampirism = 0
|
|
self.healing = 0
|
|
|
|
def hit(self, other_unit) -> bool:
|
|
hit_points = self.attack - other_unit.defense
|
|
hit_points = hit_points if hit_points > 0 else 0
|
|
other_unit.health -= hit_points
|
|
other_unit.is_alive = other_unit.health > 0
|
|
|
|
self.health += (hit_points * self.vampirism)
|
|
return other_unit.is_alive
|
|
|
|
|
|
class Knight(Warrior):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.attack = 7
|
|
|
|
|
|
class Defender(Warrior):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.health = 60
|
|
self.attack = 3
|
|
self.defense = 2
|
|
|
|
|
|
class Vampire(Warrior):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.health = 40
|
|
self.attack = 4
|
|
self.vampirism = 0.5
|
|
|
|
|
|
class Rookie(Warrior):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__()
|
|
self.health = 50
|
|
self.attack = 1
|
|
|
|
|
|
class Lancer(Warrior):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.attack = 6
|
|
|
|
|
|
class Healer(Warrior):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.health = 60
|
|
self.attack = 0
|
|
self.healing = 2
|
|
|
|
def heal(self, unit: Warrior):
|
|
unit.health += self.healing
|
|
|
|
|
|
def fight(unit_1, unit_2):
|
|
while unit_1.is_alive and unit_2.is_alive:
|
|
if not unit_1.hit(unit_2):
|
|
return True
|
|
if not unit_2.hit(unit_1):
|
|
return False
|
|
|
|
|
|
class Army:
|
|
def __init__(self):
|
|
self.units = []
|
|
|
|
def add_units(self, unit_type, quantity):
|
|
for i in range(quantity):
|
|
self.units.append(unit_type())
|
|
|
|
|
|
class Battle:
|
|
@staticmethod
|
|
def fight(army_1: Army, army_2: Army) -> bool:
|
|
while len(army_1.units) > 0 and len(army_2.units) > 0:
|
|
current_unit_1 = army_1.units[0]
|
|
current_unit_2 = army_2.units[0]
|
|
while current_unit_1.is_alive and current_unit_2.is_alive:
|
|
current_unit_1_attack = current_unit_1.hit(current_unit_2)
|
|
if len(army_1.units) > 1 and isinstance(army_1.units[1], Healer):
|
|
current_unit_1.health += army_1.units[1].healing
|
|
if isinstance(current_unit_1, Lancer) and len(army_2.units) > 1:
|
|
army_2.units[1].health -= current_unit_1.attack * 0.5
|
|
if not current_unit_1_attack:
|
|
army_2.units.pop(0)
|
|
break
|
|
current_unit_2_attack = current_unit_2.hit(current_unit_1)
|
|
if len(army_2.units) > 1 and isinstance(army_2.units[1], Healer):
|
|
current_unit_2.health += army_2.units[1].healing
|
|
if isinstance(current_unit_2, Lancer) and len(army_1.units) > 1:
|
|
army_1.units[1].health -= current_unit_2.attack * 0.5
|
|
if not current_unit_2_attack:
|
|
army_1.units.pop(0)
|
|
break
|
|
return len(army_1.units) > 0
|
|
|
|
|
|
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()
|
|
eric = Vampire()
|
|
adam = Vampire()
|
|
richard = Defender()
|
|
ogre = Warrior()
|
|
freelancer = Lancer()
|
|
vampire = Vampire()
|
|
priest = Healer()
|
|
|
|
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
|
|
assert fight(eric, richard) == False
|
|
assert fight(ogre, adam) == True
|
|
assert fight(freelancer, vampire) == True
|
|
assert freelancer.is_alive == True
|
|
assert freelancer.health == 14
|
|
priest.heal(freelancer)
|
|
assert freelancer.health == 16
|
|
|
|
# battle tests
|
|
my_army = Army()
|
|
my_army.add_units(Defender, 2)
|
|
my_army.add_units(Healer, 1)
|
|
my_army.add_units(Vampire, 2)
|
|
my_army.add_units(Lancer, 2)
|
|
my_army.add_units(Healer, 1)
|
|
my_army.add_units(Warrior, 1)
|
|
|
|
enemy_army = Army()
|
|
enemy_army.add_units(Warrior, 2)
|
|
enemy_army.add_units(Lancer, 4)
|
|
enemy_army.add_units(Healer, 1)
|
|
enemy_army.add_units(Defender, 2)
|
|
enemy_army.add_units(Vampire, 3)
|
|
enemy_army.add_units(Healer, 1)
|
|
|
|
army_3 = Army()
|
|
army_3.add_units(Warrior, 1)
|
|
army_3.add_units(Lancer, 1)
|
|
army_3.add_units(Healer, 1)
|
|
army_3.add_units(Defender, 2)
|
|
|
|
army_4 = Army()
|
|
army_4.add_units(Vampire, 3)
|
|
army_4.add_units(Warrior, 1)
|
|
army_4.add_units(Healer, 1)
|
|
army_4.add_units(Lancer, 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!") |