def test_health_setter(self): survivor = Survivor('test', 20) survivor.health = 60 self.assertEqual(survivor.health, 60) survivor.health += 20 self.assertEqual(survivor.health, 80) survivor.health += 40 self.assertEqual(survivor.health, 100) with self.assertRaises(ValueError) as ex: survivor.health = -50 self.assertEqual(str(ex.exception), "Health not valid!")
def test_health_raises(self): s = Survivor("test", 10) with self.assertRaises(ValueError) as ex: s.health = -10 self.assertEqual(str(ex.exception), "Health not valid!")
def apply(self, survivor: Survivor): if survivor.health + self.health_increase > 100: survivor.health = 100 else: survivor.health += self.health_increase