class TestFight(unittest.TestCase): def setUp(self): self.pudge = Hero() self.magic = Spell(mana_cost=5, damage=33) self.sword = Weapon(damage=30) self.pudge.learn(self.magic) self.pudge.equip(self.sword) self.battle = Fight(self.pudge, (4, 6), (4, 5), 'walk') def test_direct_and_dist(self): direct_and_dist = self.battle.find_direct_and_dist() self.assertEqual(direct_and_dist[0], 'right') self.assertEqual(direct_and_dist[1], 1) def test_is_spell_more_eq_dmg(self): self.assertTrue(self.battle.is_spell_more_eq_dmg()) def test_fight_scenario(self): self.assertTrue(self.battle.fight_scenario()) print (self.battle) def test_combat_logg(self): pass