def reset_hero_attributes(hero: Hero):
from items.weapon import Weapon
from items.spell import Spell
hero.health = hero.MAX_HEALTH
hero.mana = hero.MAX_MANA
hero.weapon = Weapon.create_weapon(random.choice(WEAPON_NAMES))
hero.spell = Spell.create_spell(random.choice(SPELL_NAMES))
def test_if_regular_fight_works_when_enemy_is_killed_during_the_fight(self):
health = 1
hero = Hero(health=health)
hero.weapon = Weapon(damage=1)
enemy = Enemy(health=health)
exp = 0
regular_fight(hero, enemy)
self.assertEqual(enemy.health, exp)
def test_if_attack_returns_weapon_damagae_if_it_is_stronger(self):
h = Hero()
h.spell = Spell('Fireball', 10, 5, 1)
h.weapon = Weapon('Bat', 20)
self.assertEqual(h.attack(), 20)