Fix broken game test

dungeonbattle/tests/entities_test.py

@@ -28,7 +28,7 @@ class TestEntities(unittest.TestCase):
         Test some random stuff with fighting entities.
         """
         entity = Hedgehog()
-        self.assertIsNone(entity.act(self.map))
+        self.map.add_entity(entity)
         self.assertEqual(entity.maxhealth, 10)
         self.assertEqual(entity.maxhealth, entity.health)
         self.assertEqual(entity.strength, 3)
@@ -41,6 +41,12 @@ class TestEntities(unittest.TestCase):
         self.assertIsNone(entity.hit(entity))
         self.assertTrue(entity.dead)
 
+        entity = Hedgehog()
+        self.map.add_entity(entity)
+        entity.move(2, 6)
+        self.map.tick()
+        self.assertFalse(entity.y == 2 and entity.x == 6)
+
     def test_items(self) -> None:
         """
         Test some random stuff with items.

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