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"
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"