class applyEffect(unittest.TestCase): """ Test that applyEffect actually adds a Periodic Heal effect """ def setUp(self): """ Builds the delegate and pkmn for use in the tests """ self.pkmn = BuildPokemonBattleWrapper() self.delegate = PeriodicHealDelegate("", "") def appliesPeriodicHeal(self): """ Tests if applyEffect applies the heal """ self.pkmn.secondaryEffects = [] self.delegate.applyEffect(self.pkmn, None, None) assert isinstance(self.pkmn.secondaryEffects[0], PeriodicHeal), "Should have a periodic heal effect"