def test_guesses_most_likely_item(self): self.get_sets_mock.return_value = { constants.ITEMS: { "item_one": 10, "item_two": 9 } } p = Pokemon('pikachu', 100) p.item = constants.UNKNOWN_ITEM p.update_item_for_random_battles() self.assertEqual("item_one", p.item)
def test_does_not_guess_item_when_it_is_none(self): p = Pokemon('pikachu', 100) p.item = None p.update_item_for_random_battles() self.assertIsNone(p.item)
def test_does_not_guess_item_when_it_exists(self): p = Pokemon('pikachu', 100) p.item = 'some_item' p.update_item_for_random_battles() self.assertEqual('some_item', p.item)
def test_gets_item_when_item_is_unknown(self): p = Pokemon('pikachu', 100) p.item = constants.UNKNOWN_ITEM p.update_item_for_random_battles() self.assertNotEqual(p.item, constants.UNKNOWN_ITEM)