Get legal moves from player.

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

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index 9939f58b31c4118953f0c0b6e5f5b6cc07dd4b66..cbae8cc23c03580a14f26f9f9043a90ba124b680 100644 (file)
--- a/naja/tests/test_player.py
+++ b/naja/tests/test_player.py
@@ -157,3 +157,13 @@ class TestPlayer(TestCase):
         player = Player(0x0f, (4, 2))
         self.assertEqual(player.move(BITS.EAST), False)
         self.assertEqual(player.position, (4, 2))
+
+    def test_legal_moves_all_available(self):
+        player = Player(0x0f, (2, 2))
+        self.assertEqual(
+            player.legal_moves(), [(2, 1), (2, 3), (3, 2), (1, 2)])
+
+    def test_legal_moves_some_unavailable(self):
+        player = Player(0x0f, (0, 2))
+        player.bits.clear_bit(BITS.NORTH)
+        self.assertEqual(player.legal_moves(), [(0, 3), (1, 2)])