class applyEffect(unittest.TestCase): """ Test cases of applyEffect """ def setUp(self): """ Build the Pkmn, Abilities and Effect for the test """ self.user = BuildPokemonBattleWrapper() self.target = BuildPokemonBattleWrapper() self.userAbility = Ability(None) self.targetAbility = Ability(None) self.user.setAbility(self.userAbility) self.target.setAbility(self.targetAbility) self.delegate = SwapAbilityDelegate() def swapped(self): """ Test that the abilities are swapped """ self.delegate.applyEffect(self.user, self.target, None) assert self.user.getAbility() is self.targetAbility, "User should have target's ability" assert self.target.getAbility() is self.userAbility, "Target should have user's ability" def message(self): """ Test message is returned correctly """ messages = self.delegate.applyEffect(self.user, self.target, None) assert messages == [SwapAbilityDelegate.message % (self.user.getHeader(), self.target.getHeader())], "Should have the Effect's message"
def setUp(self): """ Build the Pkmn, Abilities and Effect for the test """ self.user = BuildPokemonBattleWrapper() self.target = BuildPokemonBattleWrapper() self.userAbility = Ability(None) self.targetAbility = Ability(None) self.user.setAbility(self.userAbility) self.target.setAbility(self.targetAbility) self.delegate = SwapAbilityDelegate()