- 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))
+ player.bits.clear_bit(BITS.NORTH)
+ self.assertEqual(player.legal_moves(), [(0, 2), (0, 3), (1, 2)])
+
+ 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)])
+
+ 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), (4, 0)])
+
+ def test_legal_moves_knight(self):
+ player = Player(0x0f, (1, 3), MOVES.KNIGHT)
+ 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)
+ player.set_position((4, 4))
+ self.assertEqual(player.movement_mode, MOVES.ADJACENT)