def test_apply_poisoning_effect(self):
Status.apply_non_volatile_status(StatusType.Psn, self.pokemon)
print(self.pokemon.stats.get_actual_hp())
Status.apply_poisoning_effect(self.pokemon)
print(self.pokemon.stats.get_actual_hp())
check = Status.apply_bad_poisoning_effect(self.pokemon)
self.assertFalse(check)
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_apply_non_volatile_status(self):
check = Status.apply_non_volatile_status(StatusType.Brn, self.pokemon)
self.assertEqual(False, check)
self.assertEqual(self.pokemon.non_volatile_status, StatusType.Normal)
Status.apply_non_volatile_status(StatusType.Slp, self.pokemon)
self.assertEqual(self.pokemon.non_volatile_status, StatusType.Slp)
def test_apply_fainted_effect(self):
Status.apply_non_volatile_status(StatusType.Fnt, self.pokemon)
self.assertEqual(0, self.pokemon.stats.get_actual_hp())
def test_remove_non_volatile_status(self):
Status.apply_non_volatile_status(StatusType.Par, self.pokemon)
Status.remove_non_volatile_status(self.pokemon)
self.assertEqual(self.pokemon.stats.get_actual(StatsType.Spe), 167)
def test_apply_paralysis_effect(self):
Status.apply_non_volatile_status(StatusType.Par, self.pokemon)
self.assertEqual(self.pokemon.stats.get_actual(StatsType.Spe), 84)
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)