class getMod(unittest.TestCase):
""" Test the getMod function """
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
mod = self.effectivenessTable.getMod("GHOST", "NORMAL")
assert mod is 0, "GHOST on NORMAL should be immune"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
mod = self.effectivenessTable.getMod("FIRE", "FIRE")
assert mod == .5, "FIRE on FIRE should be ineffective"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
mod = self.effectivenessTable.getMod("NORMAL", "NORMAL")
assert mod is 1, "NORMAL on NORMAL should be effective"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
mod = self.effectivenessTable.getMod("FIRE", "ICE")
assert mod is 2, "FIRE on ICE should be super-effective"
class getMod(unittest.TestCase):
""" Test the getMod function """
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
mod = self.effectivenessTable.getMod("GHOST", "NORMAL")
assert mod is 0, "GHOST on NORMAL should be immune"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
mod = self.effectivenessTable.getMod("FIRE", "FIRE")
assert mod == .5, "FIRE on FIRE should be ineffective"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
mod = self.effectivenessTable.getMod("NORMAL", "NORMAL")
assert mod is 1, "NORMAL on NORMAL should be effective"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
mod = self.effectivenessTable.getMod("FIRE", "ICE")
assert mod is 2, "FIRE on ICE should be super-effective"
def setUp(self):
self.typeImmune = self.getPkmnWithTypes(["GHOST", "NORMAL"])
self.typeIneffective = self.getPkmnWithTypes(["FIRE"])
self.typeIneffective2 = self.getPkmnWithTypes(["FIRE", "ROCK"])
self.typeEffective = self.getPkmnWithTypes(["NORMAL"])
self.typeSuperEffective = self.getPkmnWithTypes(["ICE"])
self.typeSuperEffective2 = self.getPkmnWithTypes(["ICE", "GRASS"])
self.effectivenessTable = Effectiveness()
def setUp(self):
self.typeImmune = self.getPkmnWithTypes(["GHOST", "NORMAL"])
self.typeIneffective = self.getPkmnWithTypes(["FIRE"])
self.typeIneffective2 = self.getPkmnWithTypes(["FIRE", "ROCK"])
self.typeEffective = self.getPkmnWithTypes(["NORMAL"])
self.typeSuperEffective = self.getPkmnWithTypes(["ICE"])
self.typeSuperEffective2 = self.getPkmnWithTypes(["ICE", "GRASS"])
self.effectivenessTable = Effectiveness()
class getEffectiveness(unittest.TestCase):
""" Test the getMod function """
def setUp(self):
self.typeImmune = self.getPkmnWithTypes(["GHOST", "NORMAL"])
self.typeIneffective = self.getPkmnWithTypes(["FIRE"])
self.typeIneffective2 = self.getPkmnWithTypes(["FIRE", "ROCK"])
self.typeEffective = self.getPkmnWithTypes(["NORMAL"])
self.typeSuperEffective = self.getPkmnWithTypes(["ICE"])
self.typeSuperEffective2 = self.getPkmnWithTypes(["ICE", "GRASS"])
self.effectivenessTable = Effectiveness()
def getPkmnWithTypes(self, types):
""" Returns a Pokemon Battle Wrapper with the given types """
pkmn = BuildPokemonBattleWrapper()
pkmn.setTypes(types)
return pkmn
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness(
"GHOST", self.typeImmune)
assert mod is 0, "GHOST on NORMAL should be immune"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness(
"FIRE", self.typeIneffective)
assert mod == .5, "FIRE on FIRE should be ineffective"
mod, message = self.effectivenessTable.getEffectiveness(
"FIRE", self.typeIneffective2)
assert mod == .25, "FIRE on FIRE/ROCK should be 2x ineffective"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness(
"NORMAL", self.typeEffective)
assert mod is 1, "NORMAL on NORMAL should be effective"
assert message is None, \
"No message should be returned when the attack is unmodified"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness(
"FIRE", self.typeSuperEffective)
assert mod is 2, "FIRE on ICE should be super-effective"
mod, message = self.effectivenessTable.getEffectiveness(
"FIRE", self.typeSuperEffective2)
assert mod is 4, "FIRE on ICE/GRASS should be 2x super-effective"
class getEffectiveness(unittest.TestCase):
""" Test the getMod function """
def setUp(self):
self.typeImmune = self.getPkmnWithTypes(["GHOST", "NORMAL"])
self.typeIneffective = self.getPkmnWithTypes(["FIRE"])
self.typeIneffective2 = self.getPkmnWithTypes(["FIRE", "ROCK"])
self.typeEffective = self.getPkmnWithTypes(["NORMAL"])
self.typeSuperEffective = self.getPkmnWithTypes(["ICE"])
self.typeSuperEffective2 = self.getPkmnWithTypes(["ICE", "GRASS"])
self.effectivenessTable = Effectiveness()
def getPkmnWithTypes(self, types):
""" Returns a Pokemon Battle Wrapper with the given types """
pkmn = BuildPokemonBattleWrapper()
pkmn.setTypes(types)
return pkmn
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness("GHOST", self.typeImmune)
assert mod is 0, "GHOST on NORMAL should be immune"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness("FIRE", self.typeIneffective)
assert mod == .5, "FIRE on FIRE should be ineffective"
mod, message = self.effectivenessTable.getEffectiveness("FIRE", self.typeIneffective2)
assert mod == .25, "FIRE on FIRE/ROCK should be 2x ineffective"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness("NORMAL", self.typeEffective)
assert mod is 1, "NORMAL on NORMAL should be effective"
assert message is None, \
"No message should be returned when the attack is unmodified"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
mod, message = self.effectivenessTable.getEffectiveness("FIRE", self.typeSuperEffective)
assert mod is 2, "FIRE on ICE should be super-effective"
mod, message = self.effectivenessTable.getEffectiveness("FIRE", self.typeSuperEffective2)
assert mod is 4, "FIRE on ICE/GRASS should be 2x super-effective"
class respond(unittest.TestCase):
""" Test the respondfunction """
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
message = self.effectivenessTable.respond(0)
assert message == Effectiveness.immuneResponse, \
"Response should be the immune response"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
message = self.effectivenessTable.respond(.25)
assert message == Effectiveness.ineffectiveResponse, \
"Response should be the ineffective response"
message = self.effectivenessTable.respond(.5)
assert message == Effectiveness.ineffectiveResponse, \
"Response should be the ineffective response"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
message = self.effectivenessTable.respond(1)
assert message == Effectiveness.effectiveResponse, \
"Response should be the effective response"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
message = self.effectivenessTable.respond(2)
assert message == Effectiveness.superEffectiveResponse, \
"Response should be the super-effective response"
message = self.effectivenessTable.respond(4)
assert message == Effectiveness.superEffectiveResponse, \
"Response should be the super-effective response"
class respond(unittest.TestCase):
""" Test the respondfunction """
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()
def immune(self):
""" Test that the correct values are returned when the
Pokemon is immune to the Attack Type """
message = self.effectivenessTable.respond(0)
assert message == Effectiveness.immuneResponse, \
"Response should be the immune response"
def ineffective(self):
""" Test that the correct values are returned when the
Pokemon is resistant to the Attack Type """
message = self.effectivenessTable.respond(.25)
assert message == Effectiveness.ineffectiveResponse, \
"Response should be the ineffective response"
message = self.effectivenessTable.respond(.5)
assert message == Effectiveness.ineffectiveResponse, \
"Response should be the ineffective response"
def effective(self):
""" Test that the correct values are returned when the
Pokemon is indifferent to the Attack Type """
message = self.effectivenessTable.respond(1)
assert message == Effectiveness.effectiveResponse, \
"Response should be the effective response"
def supereffective(self):
""" Test that the correct values are returned when the
Pokemon is weak to the Attack Type """
message = self.effectivenessTable.respond(2)
assert message == Effectiveness.superEffectiveResponse, \
"Response should be the super-effective response"
message = self.effectivenessTable.respond(4)
assert message == Effectiveness.superEffectiveResponse, \
"Response should be the super-effective response"
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()
def setUp(self):
""" """
self.effectivenessTable = Effectiveness()