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)