Problem in the last assert
My code:
class Warrior:
def __init__(self):
self.health = 50
self.max_hp = 50
self.attack = 5
self.defens = 0
self.vampirism = 0
self.is_alive = True
class Knight(Warrior):
def __init__(self):
super().__init__()
self.attack = 7
class Defender(Warrior):
def __init__(self):
super().__init__()
self.health = 60
self.max_hp = 60
self.attack = 3
self.defens = 2
class Vampire(Warrior):
def __init__(self):
super().__init__()
self.health = 40
self.max_hp = 40
self.attack = 4
self.vampirism = 50
class Lancer(Warrior):
def __init__(self):
super().__init__()
self.attack = 6
self.attack2 = 3
class Healer(Warrior):
def __init__(self):
super().__init__()
self.attack = 0
self.health = 60
self.max_hp = 60
self.up_hp = 2
def heal(self, prev_unit):
prev_unit.health += self.up_hp
if prev_unit.health > prev_unit.max_hp:
prev_unit.health = prev_unit.max_hp
return prev_unit.health
class Army:
def __init__(self):
self.units = []
def add_units(self, type_unit, solders):
for i in range(solders):
self.units.append(type_unit())
return self.units
class Battle:
def fight(self, army1, army2):
x = 0
y = 0
while x < (len(army1.units)) and y < len(army2.units):
if fight(army1.units[x], army2.units[y]):
if type(army1.units[x]) is Lancer and y != (len(army2.units) - 1):
if army1.units[x].attack2 > army2.units[y + 1].defens:
army2.units[y + 1].health -= army1.units[x].attack2 - army2.units[y + 1].defens
elif x != (len(army1.units) - 1) and type(army1.units[x+1]) is Healer:
army1.units[x].health += army1.units[x+1].heal(army1.units[x])
y += 1
else:
x += 1
return x < (len(army1.units))
def fight(unit_1, unit_2):
while unit_1.is_alive and unit_2.is_alive:
if unit_1.attack > unit_2.defens:
unit_2.health -= unit_1.attack - unit_2.defens
unit_1.health += int((unit_1.attack - unit_2.defens) * (unit_1.vampirism / 100))
if unit_1.health > unit_1.max_hp:
unit_1.health = unit_1.max_hp
if unit_2.health <= 0:
unit_2.is_alive = False
break
if unit_2.attack > unit_1.defens:
unit_1.health -= unit_2.attack - unit_1.defens
unit_2.health += int((unit_2.attack - unit_1.defens) * (unit_2.vampirism / 100))
if unit_2.health > unit_2.max_hp:
unit_2.health = unit_2.max_hp
if unit_1.health <= 0:
unit_1.is_alive = False
return unit_1.is_alive
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!")
All tests are correct except the last. I can’t understand where the problem is. Help, pls.
task.the-healers