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")