diff --git a/squirrelbattle/tests/entities_test.py b/squirrelbattle/tests/entities_test.py index 5f9d298..341461f 100644 --- a/squirrelbattle/tests/entities_test.py +++ b/squirrelbattle/tests/entities_test.py @@ -96,10 +96,10 @@ class TestEntities(unittest.TestCase): self.map.add_entity(entity) self.map.add_entity(fam) - self.player.move(1,6) - entity.move(2,6) - fam.move(2,7) - + self.player.move(1, 6) + entity.move(2, 6) + fam.move(2, 7) + entity.health = 2 entity.paths = [] entity.recalculate_paths() @@ -107,8 +107,8 @@ class TestEntities(unittest.TestCase): self.map.tick(self.player) self.assertTrue(entity.dead) - self.player.move(5,5) - fam.move(4,5) + self.player.move(5, 5) + fam.move(4, 5) fam.target = self.player self.player.move_down() self.map.tick(self.player) @@ -245,4 +245,3 @@ class TestEntities(unittest.TestCase): player_state = player.save_state() self.assertEqual(player_state["current_xp"], 10) - diff --git a/squirrelbattle/tests/game_test.py b/squirrelbattle/tests/game_test.py index f8a6e93..8cd816b 100644 --- a/squirrelbattle/tests/game_test.py +++ b/squirrelbattle/tests/game_test.py @@ -599,5 +599,5 @@ class TestGame(unittest.TestCase): self.game.state = GameMode.CREDITS self.game.handle_key_pressed(KeyValues.ENTER) - + self.assertEqual(self.game.state, GameMode.MAINMENU)