class hasHeal(unittest.TestCase): """ Test that hasHeal returns appropriately for when there is and isn't periodic heal effects """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.delegate = PeriodicHealDelegate("", "") self.heal = PeriodicHeal("") def hasHeal(self): """ Tests if hasHeal returns true when there is a heal """ self.pkmn.secondaryEffects = [self.heal] assert self.delegate.hasHeal(self.pkmn), "Should have a periodic heal effect" def noHeal(self): """ Tests if hasHeal returns false when there is no heal """ self.pkmn.secondaryEffects = [] assert not self.delegate.hasHeal(self.pkmn), "Should not have a periodic heal effect"
class hasHeal(unittest.TestCase): """ Test that hasHeal returns appropriately for when there is and isn't periodic heal effects """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.delegate = PeriodicHealDelegate("", "") self.heal = PeriodicHeal("") def hasHeal(self): """ Tests if hasHeal returns true when there is a heal """ self.pkmn.secondaryEffects = [self.heal] assert self.delegate.hasHeal( self.pkmn), "Should have a periodic heal effect" def noHeal(self): """ Tests if hasHeal returns false when there is no heal """ self.pkmn.secondaryEffects = [] assert not self.delegate.hasHeal( self.pkmn), "Should not have a periodic heal effect"