X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_player.py;h=2e17379213db3d9e4a8f5f6d0603c199ddf88eb8;hb=266021313a1501c86c11c803851c2b4bb50a8c00;hp=d45c09eadcdad3957862611d5f337b41919c4f56;hpb=aac1783aa9c84031bb37d74616ee72a7b32ebeac;p=naja.git

diff --git a/naja/tests/test_player.py b/naja/tests/test_player.py
index d45c09e..2e17379 100644
--- a/naja/tests/test_player.py
+++ b/naja/tests/test_player.py
@@ -1,6 +1,6 @@
 from unittest import TestCase
 
-from naja.constants import BITS
+from naja.constants import BITS, MOVES
 from naja.player import PlayerBits, Player
 
 
@@ -130,17 +130,19 @@ class TestPlayer(TestCase):
 
     def test_legal_moves_castle(self):
         player = Player(0x0f, (1, 3), MOVES.CASTLE)
-        self.assertEqual(player.legal_moves(), [(0, 3), (1, 0), (1, 1), (1, 2),
-        (1, 3), (1, 4), (2, 3), (3, 3), (4, 3)])
+        self.assertEqual(player.legal_moves(), [
+            (0, 3), (1, 0), (1, 1), (1, 2), (1, 3), (1, 4), (2, 3), (3, 3),
+            (4, 3)])
 
     def test_legal_moves_bishop(self):
         player = Player(0x0f, (1, 3), MOVES.BISHOP)
-        self.assertEqual(player.legal_moves(), [(0, 2), (0, 4), (1, 3), (2, 2),
-        (2, 4), (3, 1), (3, 4), (4, 0)])
+        self.assertEqual(player.legal_moves(), [
+            (0, 2), (0, 4), (1, 3), (2, 2), (2, 4), (3, 1), (4, 0)])
 
     def test_legal_moves_knight(self):
         player = Player(0x0f, (1, 3), MOVES.KNIGHT)
-        self.assertEqual(player.legal_moves(), [(0, 1), (2, 1), (3, 2), (3, 4)])
+        self.assertEqual(player.legal_moves(), [
+            (0, 1), (1, 3), (2, 1), (3, 2), (3, 4)])
 
     def test_set_position(self):
         player = Player(0x0f, (3, 3), MOVES.BISHOP)