def test_guesses_assultvest_when_can_have_assultvest_flag_is_true(self):
p = Pokemon('pikachu', 100)
p.item = constants.UNKNOWN_ITEM
p.can_have_assaultvest = True
items = [
('lightball', 50),
('assaultvest', 50),
]
possible_items = p.get_possible_items(items)
expected_items = ['lightball', 'assaultvest']
self.assertEqual(expected_items, possible_items)
def test_does_not_guess_assultvest_when_can_have_assultvest_flag_is_false(
self):
p = Pokemon('pikachu', 100)
p.item = constants.UNKNOWN_ITEM
p.can_have_assaultvest = False
items = [
('lightball', 50),
('assaultvest',
50), # should be ignored because flag is set to False
]
possible_items = p.get_possible_items(items)
expected_items = [
'lightball',
]
self.assertEqual(expected_items, possible_items)
