def __init__(self): self.health = HealthAspect()
class HealthAspectTest(unittest.TestCase): def __init__(self): self.health = HealthAspect() def test_heal(self): self.health.health = 0 self.health.add_health(10) self.assertEqual(self.health.health, 10) def test_heal_at_max(self): self.health.add_health(100) self.assertEqual(self.health.health, self.health.max_health) def test_take_damage(self): self.health.sub_health(1) self.assertEqual(self.health.health, self.health.max_health - 1) def test_take_too_much_damage(self): self.health.sub_health(self.health.max_health + 1) self.assertEqual(self.health.health, 0) def test_is_dead(self): self.assertFalse(self.health.is_dead()) self.health.sub_health(self.health.max_health) self.assertTrue(self.health.is_dead()) def test_reset_health(self): self.health.health = 0 self.health.reset_health() self.assertEqual(self.health.health, self.health.max_health) def test_is_at_full_health(self): self.assertTrue(self.health.is_full_health()) self.health.health = 0 self.assertFalse(self.health.is_full_health())