class immune(unittest.TestCase): """ Test that immune returns the correct values """ def setUp(self): """ Build side for use in test cases """ self.pkmn = BuildPokemonBattleWrapper() self.parent = Attack() self.parent.type = "NORMAL" def buildDelegate(self, effect): """ Builds a staushit delegate and its parent attack """ self.parent.effectDelegates = [effect] self.delegate = StatusHitDelegate(self.parent, 100) def notImmune(self): """ Test an effect that is not immune returns correctly """ status = Status() self.pkmn.setStatus(status) effect = ApplyStatusDelegate(self.parent, "PAR", 1) self.buildDelegate(effect) assert not self.delegate.immune(self.pkmn, None), "Should not be immune" def immune(self): """ Test an effect that is immune will result in a miss """ status, message = StatusFactory.buildStatusFromAbbr("PAR") self.pkmn.setStatus(status) effect = ApplyStatusDelegate(self.parent, "PAR", 1) self.buildDelegate(effect) assert self.delegate.immune(self.pkmn, None), "Should be immune" def noImmune(self): """ Test an effect that has no immune function isn't immune """ effect = StatModDelegate("ATK", 1, 1) self.buildDelegate(effect) assert not self.delegate.immune(self.pkmn, None), "Should never be immune"
class immune(unittest.TestCase): """ Test that immune returns the correct values """ def setUp(self): """ Build side for use in test cases """ self.pkmn = BuildPokemonBattleWrapper() self.parent = Attack() self.parent.type ="NORMAL" def buildDelegate(self, effect): """ Builds a staushit delegate and its parent attack """ self.parent.effectDelegates = [effect] self.delegate = StatusHitDelegate(self.parent, 100) def notImmune(self): """ Test an effect that is not immune returns correctly """ status = Status() self.pkmn.setStatus(status) effect = ApplyStatusDelegate(self.parent, "PAR", 1) self.buildDelegate(effect) assert not self.delegate.immune(self.pkmn, None), "Should not be immune" def immune(self): """ Test an effect that is immune will result in a miss """ status, message = StatusFactory.buildStatusFromAbbr("PAR") self.pkmn.setStatus(status) effect = ApplyStatusDelegate(self.parent, "PAR", 1) self.buildDelegate(effect) assert self.delegate.immune(self.pkmn, None), "Should be immune" def noImmune(self): """ Test an effect that has no immune function isn't immune """ effect = StatModDelegate("ATK", 1, 1) self.buildDelegate(effect) assert not self.delegate.immune(self.pkmn, None), "Should never be immune"
def buildDelegate(self, effect): """ Builds a staushit delegate and its parent attack """ self.parent.effectDelegates = [effect] self.delegate = StatusHitDelegate(self.parent, 100)