Entities are living during two ticks, fixes #80

squirrelbattle/entities/items.py

@@ -232,13 +232,18 @@ class Explosion(Item):
     """
     When a bomb explodes, the explosion is displayed.
     """
-    def __init__(self, *args, **kwargs):
+    living_ticks: int
+
+    def __init__(self, living_ticks: int = 2, *args, **kwargs):
         super().__init__(name="explosion", *args, **kwargs)
+        self.living_ticks = living_ticks
 
     def act(self, m: Map) -> None:
         """
         The bomb disappears after exploding.
         """
+        self.living_ticks -= 1
+        if self.living_ticks <= 0:
             m.remove_entity(self)
 
     def hold(self, player: InventoryHolder) -> None:

squirrelbattle/tests/entities_test.py

@@ -209,7 +209,8 @@ class TestEntities(unittest.TestCase):
         self.assertNotIn(explosion, self.player.inventory)
         self.assertIsNone(explosion.held_by)
 
-        # The explosion disappears after one tick
+        # The explosion disappears after two ticks
+        explosion.act(self.map)
         explosion.act(self.map)
         self.assertNotIn(explosion, self.map.entities)
 

squirrelbattle/tests/game_test.py

@@ -255,6 +255,7 @@ class TestGame(unittest.TestCase):
         self.game.map.add_entity(explosion)
         self.assertIn(explosion, self.game.map.entities)
         self.game.handle_key_pressed(KeyValues.WAIT)
+        self.game.handle_key_pressed(KeyValues.WAIT)
         self.assertNotIn(explosion, self.game.map.entities)
 
         rabbit = Rabbit()