Fix broken game test

dungeonbattle/tests/game_test.py

@@ -4,6 +4,7 @@ import unittest
 from dungeonbattle.bootstrap import Bootstrap
 from dungeonbattle.display.display import Display
 from dungeonbattle.display.display_manager import DisplayManager
+from dungeonbattle.entities.player import Player
 from dungeonbattle.game import Game, KeyValues, GameMode
 from dungeonbattle.menus import MainMenuValues
 
@@ -90,6 +91,11 @@ class TestGame(unittest.TestCase):
         self.game.handle_key_pressed(KeyValues.ENTER)
         self.assertEqual(self.game.state, GameMode.PLAY)
 
+        # Kill entities
+        for entity in self.game.map.entities.copy():
+            if not isinstance(entity, Player):
+                self.game.map.remove_entity(entity)
+
         y, x = self.game.player.y, self.game.player.x
         self.game.handle_key_pressed(KeyValues.DOWN)
         new_y, new_x = self.game.player.y, self.game.player.x