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
