def test_round_fight_when_hero_hits_and_enemy_is_not_gone_return_enemy_strike_back(
self):
hero = Hero("Pesho", "Padawan")
enemy = Enemy(health=200)
weapon = Weapon("shodow", 100)
hero.equip(weapon)
create_fight = Fight(hero, enemy)
hero_result = create_fight.choose_fighting_tools(hero, True)
enemy_result = create_fight.choose_fighting_tools(enemy, True)
result = create_fight.round_fight(hero_result, enemy_result)
self.assertEqual(hero.get_health(), 80)
self.assertEqual(enemy.get_health(), 100)
def test_round_fight_when_hero_hits_and_with_spell_enemy_is_not_gone_return_enemy_move_to_hero(
self):
hero = Hero("Pesho", "Padawan")
enemy = Enemy(health=200)
weapon = Weapon("shodow", 100)
hero.equip(weapon)
spell = Spell("Abraka", 120)
hero.learn(spell)
create_fight = Fight(hero, enemy)
hero_result = create_fight.choose_fighting_tools(hero, False)
enemy_result = create_fight.choose_fighting_tools(enemy, True)
result = create_fight.round_fight(hero_result, enemy_result, True)
self.assertEqual(hero.get_health(), 100)
self.assertEqual(enemy.get_health(), 80)
