class getTurns(unittest.TestCase):
""" Test cases of getTurns """
def setUp(self):
""" Build the Effect, Environment, and Pokemon for the test """
self.turnToAttack = 1
self.effect = WeatherChargeDelegate(2, self.turnToAttack, "Some Message", Hail.type)
self.environment = BattleEnvironment()
self.user = BuildPokemonBattleWrapper()
def rain(self):
""" Test that the weather can make the Attack be fully charged """
self.environment.weather.type = "RAIN"
self.effect.getTurns(self.environment)
assert self.effect.turns == 3, "Should last 2 turns"
assert self.effect.turnToAttack == 2, "Should attack on turn index 1"
def otherWeather(self):
""" Tests if getTurns returns correctly when it is charging """
self.effect.getTurns(self.environment)
assert self.effect.turns == 2, "Should last 2 turns"
assert self.effect.turnToAttack == 1, "Should attack on turn index 1"
class isCharging(unittest.TestCase):
""" Test cases of isCharging """
def setUp(self):
""" Build the Effect, Environment, and Pokemon for the test """
self.turnToAttack = 1
self.effect = WeatherChargeDelegate(2, self.turnToAttack, "Some Message", Hail.type)
self.environment = BattleEnvironment()
self.user = BuildPokemonBattleWrapper()
def weatherCharged(self):
""" Test that the weather can make the Attack be fully charged """
self.environment.setWeather(self.effect.weatherType)
assert not self.effect.isCharging(self.user, self.environment), "Effect should not be charging when the weather is right"
def isCharging(self):
""" Tests if isCharging returns correctly when it is charging """
self.effect.turnOn = self.turnToAttack - 1
assert self.effect.isCharging(self.user, self.environment), "Should be charging"
def isNotCharging(self):
""" Tests if isCharging returns correctly when it is not charging """
self.effect.turnOn = self.turnToAttack
assert not self.effect.isCharging(self.user, self.environment), "Should not be charging"
def setUp(self):
""" Build the Effect, Environment, and Pokemon for the test """
self.turnToAttack = 1
self.effect = WeatherChargeDelegate(2, self.turnToAttack, "Some Message", Hail.type)
self.environment = BattleEnvironment()
self.user = BuildPokemonBattleWrapper()