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)
def test_add_volatile_status(self): self.assertNotIn(StatusType.Attract, self.pokemon.volatile_status) Status.add_volatile_status(StatusType.Attract, self.pokemon) self.assertIn(StatusType.Attract, self.pokemon.volatile_status) check = Status.add_volatile_status(StatusType.Attract, self.pokemon) self.assertEqual(False, check)