def test_can_cast(self): h = Hero(name="Pesho", title="The Great One", health=100, mana=10, mana_regeneration_rate=30) s = Spell(name="Abrakadabra", damage=50, mana_cost=2, cast_range=4) h.learn(s) self.assertEqual(True, h.can_cast())
def test_learn(self): h = Hero(name="Pesho", title="The Great One", health=100, mana=10, mana_regeneration_rate=30) s = Spell(name="Abrakadabra", damage=50, mana_cost=20, cast_range=4) h.learn(s) self.assertEqual(h.equiped_spell.name, "Abrakadabra")