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)
