def test_gets_most_likely_ability(self):
self.get_sets_mock.return_value = {
constants.ABILITIES: {
"immunity": 10,
"sandveil": 9
}
}
p = Pokemon('gligar', 100)
p.ability = None
p.update_ability_for_random_battles()
self.assertEqual("immunity", p.ability)
def test_no_abilities_does_not_error(self):
self.get_sets_mock.return_value = {constants.ABILITIES: {}}
p = Pokemon('gligar', 100)
p.ability = None
p.update_ability_for_random_battles()
self.assertEqual(None, p.ability)
def test_does_not_get_ability_when_it_exists(self):
p = Pokemon('pikachu', 100)
p.ability = "some_ability"
p.update_ability_for_random_battles()
self.assertEqual("some_ability", p.ability)
def test_gets_item_when_item_is_unknown(self):
p = Pokemon('pikachu', 100)
p.ability = None
p.update_ability_for_random_battles()
self.assertIsNotNone(p.ability)
