def test_one(self):
hero = Fighter("Hero", 8)
hero.set_attack(5)
hero.set_armor(5)
boss = Fighter("Boss", 12)
boss.set_attack(7)
boss.set_armor(2)
battle = Battle(hero, boss)
battle.start()
expected = hero
actual = battle.get_winner()
self.assertEqual(expected, actual)
