Only move if the flags allow.

[naja.git] / naja / tests / test_player.py

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index c748c7295827b9e1355c6cb3f0e6182f049585b2..9939f58b31c4118953f0c0b6e5f5b6cc07dd4b66 100644 (file)
--- a/naja/tests/test_player.py
+++ b/naja/tests/test_player.py
@@ -106,3 +106,54 @@ class TestPlayer(TestCase):
             'bits': 0xaa,
             'position': [1, 2],
         })
+
+    def test_move(self):
+        player = Player(0x0f, (2, 2))
+        self.assertEqual(player.move(BITS.NORTH), True)
+        self.assertEqual(player.position, (2, 1))
+
+        player = Player(0x0f, (2, 2))
+        self.assertEqual(player.move(BITS.SOUTH), True)
+        self.assertEqual(player.position, (2, 3))
+
+        player = Player(0x0f, (2, 2))
+        self.assertEqual(player.move(BITS.WEST), True)
+        self.assertEqual(player.position, (1, 2))
+
+        player = Player(0x0f, (2, 2))
+        self.assertEqual(player.move(BITS.EAST), True)
+        self.assertEqual(player.position, (3, 2))
+
+    def test_move_flags_clear(self):
+        player = Player(0x00, (2, 2))
+        self.assertEqual(player.move(BITS.NORTH), False)
+        self.assertEqual(player.position, (2, 2))
+
+        player = Player(0x00, (2, 2))
+        self.assertEqual(player.move(BITS.SOUTH), False)
+        self.assertEqual(player.position, (2, 2))
+
+        player = Player(0x00, (2, 2))
+        self.assertEqual(player.move(BITS.WEST), False)
+        self.assertEqual(player.position, (2, 2))
+
+        player = Player(0x00, (2, 2))
+        self.assertEqual(player.move(BITS.EAST), False)
+        self.assertEqual(player.position, (2, 2))
+
+    def test_move_flags_edges(self):
+        player = Player(0x0f, (2, 0))
+        self.assertEqual(player.move(BITS.NORTH), False)
+        self.assertEqual(player.position, (2, 0))
+
+        player = Player(0x0f, (2, 4))
+        self.assertEqual(player.move(BITS.SOUTH), False)
+        self.assertEqual(player.position, (2, 4))
+
+        player = Player(0x0f, (0, 2))
+        self.assertEqual(player.move(BITS.WEST), False)
+        self.assertEqual(player.position, (0, 2))
+
+        player = Player(0x0f, (4, 2))
+        self.assertEqual(player.move(BITS.EAST), False)
+        self.assertEqual(player.position, (4, 2))