def test_apply_infatuation(self): check = Status.apply_infatuation(self.pokemon, self.other_pokemon) self.assertEqual(True, check) check = Status.apply_infatuation(self.pokemon, self.other_pokemon) self.assertEqual(False, check) Status.remove_volatile_status(StatusType.Attract, self.pokemon) check = Status.apply_infatuation(self.pokemon, self.pokemon) self.assertEqual(False, check)
def test_remove_volatile_status(self): Status.add_volatile_status(StatusType.Attract, self.pokemon) Status.remove_volatile_status(StatusType.Attract, self.pokemon) self.assertNotIn(StatusType.Attract, self.pokemon.volatile_status)