class checkCurable(unittest.TestCase): """ Test that checkCurable actually cures a status when possible """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.statusAbbr = "PAR" self.status = Paralysis() self.statusAbbr2 = "FRZ" self.delegate = CureStatusDelegate(self.statusAbbr, 1) self.delegate2 = CureStatusDelegate(self.statusAbbr2, 1) def isCurable(self): """ Tests if checkCurable cures the status when it can """ self.pkmn.setStatus(self.status) self.delegate.checkCurable(self.pkmn) assert self.pkmn.getStatus() != self.status, "Status should be cured" def notCurable(self): """ Tests if checkCurable cures the status when it can """ self.pkmn.setStatus(self.status) self.delegate2.checkCurable(self.pkmn) assert self.pkmn.getStatus() == self.status, "Status should not be cured"
class checkCurable(unittest.TestCase): """ Test that checkCurable actually cures a status when possible """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.statusAbbr = "PAR" self.status = Paralysis() self.statusAbbr2 = "FRZ" self.delegate = CureStatusDelegate(self.statusAbbr, 1) self.delegate2 = CureStatusDelegate(self.statusAbbr2, 1) def isCurable(self): """ Tests if checkCurable cures the status when it can """ self.pkmn.setStatus(self.status) self.delegate.checkCurable(self.pkmn) assert self.pkmn.getStatus() != self.status, "Status should be cured" def notCurable(self): """ Tests if checkCurable cures the status when it can """ self.pkmn.setStatus(self.status) self.delegate2.checkCurable(self.pkmn) assert self.pkmn.getStatus( ) == self.status, "Status should not be cured"
class immobilized(unittest.TestCase): """ Test that immobilized performs correctly """ def setUp(self): """ Builds the Sleep status""" self.status = Sleep() self.pkmn = BuildPokemonBattleWrapper() self.pkmn.setStatus(self.status) self.turns = 1 def notDone(self): """ Test if immobilized returns correctly when it is not done """ self.status.turns = self.turns immobilized, messages = self.status.immobilized(self.pkmn) assert immobilized, "Should be immobilized" assert messages == [self.pkmn.getHeader() + Sleep.intermittent],\ "Should return Sleep's intermittent message" assert self.status.turns == self.turns - 1, "Turns should be decremented" def done(self): """ Test if immobilized returns correctly when it is done """ self.status.turns = 0 assert self.pkmn.getStatus( ) == self.status, "Should have status to start" immobilized, messages = self.status.immobilized(self.pkmn) assert not immobilized, "Should not be immobilized" assert len(messages) == 2, "Should have two messages" assert messages[0] == self.pkmn.getHeader() + Sleep.intermittent,\ "Should return Sleep's intermittent message" assert messages[1] == self.pkmn.getHeader() + Sleep.done,\ "Should return Sleep's done message" assert self.pkmn.getStatus() != self.status, "Status should be removed"
class immobilized(unittest.TestCase): """ Test that immobilized performs correctly """ def setUp(self): """ Builds the Sleep status""" self.status = Sleep() self.pkmn = BuildPokemonBattleWrapper() self.pkmn.setStatus(self.status) self.turns = 1 def notDone(self): """ Test if immobilized returns correctly when it is not done """ self.status.turns = self.turns immobilized, messages = self.status.immobilized(self.pkmn) assert immobilized, "Should be immobilized" assert messages == [self.pkmn.getHeader() + Sleep.intermittent],\ "Should return Sleep's intermittent message" assert self.status.turns == self.turns - 1, "Turns should be decremented" def done(self): """ Test if immobilized returns correctly when it is done """ self.status.turns = 0 assert self.pkmn.getStatus() == self.status, "Should have status to start" immobilized, messages = self.status.immobilized(self.pkmn) assert not immobilized, "Should not be immobilized" assert len(messages) == 2, "Should have two messages" assert messages[0] == self.pkmn.getHeader() + Sleep.intermittent,\ "Should return Sleep's intermittent message" assert messages[1] == self.pkmn.getHeader() + Sleep.done,\ "Should return Sleep's done message" assert self.pkmn.getStatus() != self.status, "Status should be removed"
class applyStatus(unittest.TestCase): """ Test that applyStatus actually applies a status """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.status = "PAR" attack = Attack() attack.type = "" self.delegate = ApplyStatusDelegate(attack, self.status, 1) def appliesStatusUser(self): """ Tests if applyStatus applies the staus and returns the method """ self.delegate.affectUser = 1 self.delegate.applyStatus(self.pkmn) assert self.pkmn.getStatus( ).abbr == self.status, "Status should be PAR on the user pkmn"
class applyStatus(unittest.TestCase): """ Test that applyStatus actually applies a status """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.status = "PAR" attack = Attack() attack.type = "" self.delegate = ApplyStatusDelegate(attack, self.status, 1) def appliesStatusUser(self): """ Tests if applyStatus applies the staus and returns the method """ self.delegate.affectUser = 1 self.delegate.applyStatus(self.pkmn) assert self.pkmn.getStatus().abbr == self.status, "Status should be PAR on the user pkmn"