def test_set_excessive_abilities(self):
error = "Replacing an elemental's abilities with an excessive set wasn't handled correctly"
species = self.get_species()
elemental = ElementalBuilder().with_level(5).with_species(
species).build()
abilities = [Slam(), RollingThunder(), Rend(), Fireball(), Claw()]
elemental.set_abilities(abilities)
ability_names = [
ability.name for ability in elemental.active_abilities
]
self.assertNotIn(Claw().name, ability_names, error)
self.assertEqual(4, len(elemental.active_abilities))
def test_set_abilities(self):
error = "Elemental abilities couldn't be replaced"
species = self.get_species()
elemental = ElementalBuilder().with_level(5).with_species(
species).build()
abilities = [Slam(), RollingThunder(), Rend(), Fireball()]
elemental.set_abilities(abilities)
correctly_added = 0
ability_names = [
ability.name for ability in elemental.active_abilities
]
for ability in abilities:
if ability.name in ability_names:
correctly_added += 1
self.assertEqual(correctly_added, len(abilities), error)