X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_player.py;h=31cc8173395b53d7ae9a6147d06938a0e20cb6a9;hb=0cef06f581d48315a59a19ac06ca1d2be14cbb4c;hp=9939f58b31c4118953f0c0b6e5f5b6cc07dd4b66;hpb=692a111f56f0e0edd19376e447b530ff4cb21521;p=naja.git

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index 9939f58..31cc817 100644
--- 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, 2), (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, 2), (0, 3), (1, 2)])