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()