def test_guesses_choiceband_when_can_not_have_band_is_false(self):
p = Pokemon('pikachu', 100)
p.item = constants.UNKNOWN_ITEM
p.can_not_have_band = False
items = [
('lightball', 50),
('choiceband',
50), # should NOT be ignored because flag is set to False
]
possible_items = p.get_possible_items(items)
expected_items = ['lightball', 'choiceband']
self.assertEqual(expected_items, possible_items)
def test_can_not_have_choice_specs_flag_does_not_affect_choice_band_guess(
self):
p = Pokemon('pikachu', 100)
p.item = constants.UNKNOWN_ITEM
p.can_not_have_specs = True
p.can_not_have_band = False
items = [
('lightball', 50),
('choiceband',
50), # should NOT be ignored because flag is set to False
# choice_specs flag doesn't matter
]
possible_items = p.get_possible_items(items)
expected_items = ['lightball', 'choiceband']
self.assertEqual(expected_items, possible_items)
