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"
