def setUp(self): """ Build the Pkmn and Delegate for the test """ self.pkmn = BuildPokemonBattleWrapper() self.ratio = 2 self.damage = 10 self.delegate = HealByDamageRatioDelegate(self.ratio)
class heal(unittest.TestCase): """ Test cases of heal """ def setUp(self): """ Build the Pkmn and Delegate for the test """ self.pkmn = BuildPokemonBattleWrapper() self.ratio = 2 self.damage = 10 self.delegate = HealByDamageRatioDelegate(self.ratio) def heal(self): """ Test that it heals by the ratio of the damage done """ self.pkmn.setCurrHP(0) self.delegate.damage = self.damage self.delegate.heal(self.pkmn) hp = self.pkmn.getCurrHP() heal = self.damage/self.ratio assert hp == heal, "Should be healed by the ratio of the damage"