class getStatMod(unittest.TestCase): """ Test that statMod returns the correct values for all stats """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() def checkStatMods(self): """ Test that stat modifiers are correct for Burn .5 for ATK, 1 for the rest """ for key in self.status.statMods: if key == "ATK": assert self.status.getStatMod(key) == .5, "ATK should be .5" else: assert self.status.getStatMod( key) == 1, "All stats, except ATK, should be 1"
class getStatMod(unittest.TestCase): """ Test that statMod returns the correct values for all stats """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() def checkStatMods(self): """ Test that stat modifiers are correct for Burn .5 for ATK, 1 for the rest """ for key in self.status.statMods: if key == "ATK": assert self.status.getStatMod(key) == 0.5, "ATK should be .5" else: assert self.status.getStatMod(key) == 1, "All stats, except ATK, should be 1"
class afterTurn(unittest.TestCase): """ Test that afterTurn works correctly """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() self.pkmn = BuildPokemonBattleWrapper() def damage(self): """ Test that the damage is done correctly """ self.pkmn.setStat("HP", 32) self.pkmn.setCurrHP(32) self.status.afterTurn(self.pkmn) damage = self.pkmn.getStat("HP") - self.pkmn.getCurrHP() assert damage == self.pkmn.getRatioOfHealth(Burn.ratio), "Damage should be Burn Ratio of Health" def message(self): """ Test that the message is returned correctly """ messages = self.status.afterTurn(self.pkmn) message = self.pkmn.getHeader() + Burn.intermittent assert len(messages) == 1, "Should get one message" assert messages[0] == message, "Message should be that the Pkmn was damaged by the Burn"
class immune(unittest.TestCase): """ Test that immune returns correctly """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() def immune(self): """ Test if it can correctly identify when the target is immune """ types = ["FIRE"] other = "ELECTRIC" assert self.status.immune(types, other), "Should be immune if FIRE" types = ["GROUND", "FIRE"] other = "ELECTRIC" assert self.status.immune(types, other), "Should be immune if FIRE regardless of other type" def notImmune(self): """ Test if it can correctly identify when the target is immune """ types = ["ELECTRIC"] other = "FIRE" assert not self.status.immune(types, other), "Should not be immune if not FIRE"
class immune(unittest.TestCase): """ Test that immune returns correctly """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() def immune(self): """ Test if it can correctly identify when the target is immune """ types = ["FIRE"] other = "ELECTRIC" assert self.status.immune(types, other), "Should be immune if FIRE" types = ["GROUND", "FIRE"] other = "ELECTRIC" assert self.status.immune( types, other), "Should be immune if FIRE regardless of other type" def notImmune(self): """ Test if it can correctly identify when the target is immune """ types = ["ELECTRIC"] other = "FIRE" assert not self.status.immune( types, other), "Should not be immune if not FIRE"
class afterTurn(unittest.TestCase): """ Test that afterTurn works correctly """ def setUp(self): """ Builds the Paralysis status""" self.status = Burn() self.pkmn = BuildPokemonBattleWrapper() def damage(self): """ Test that the damage is done correctly """ self.pkmn.setStat("HP", 32) self.pkmn.setCurrHP(32) self.status.afterTurn(self.pkmn) damage = self.pkmn.getStat("HP") - self.pkmn.getCurrHP() assert damage == self.pkmn.getRatioOfHealth( Burn.ratio), "Damage should be Burn Ratio of Health" def message(self): """ Test that the message is returned correctly """ messages = self.status.afterTurn(self.pkmn) message = self.pkmn.getHeader() + Burn.intermittent assert len(messages) == 1, "Should get one message" assert messages[ 0] == message, "Message should be that the Pkmn was damaged by the Burn"
def setUp(self): """ Builds the Paralysis status""" self.status = Burn() self.pkmn = BuildPokemonBattleWrapper()
def setUp(self): """ Builds the Paralysis status""" self.status = Burn()
def status(self): """ Test that damage is double the core when the target has a status """ self.target.setStatus(Burn()) damage = self.delegate.coreDamage(self.user, self.target) assert damage == 2 * self.core, "Damage should double when target has a status"