from unittest import TestCase
-from naja.constants import BITS
+from naja.constants import BITS, MOVES
from naja.gameboard import GameBoard, LocationCard
from naja import actions
(0, 2): '12', (1, 2): '32', (3, 2): '42', (4, 2): '02',
}))
+ def test_allow_chess_move_knight(self):
+ board = GameBoard.new_game([{'actions': []}])
+ board.board_locations = self.generate_locations()
+ board.allow_chess_move(MOVES.KNIGHT)
+ self.assertEqual(board.player.movement_mode, MOVES.KNIGHT)
+
+ def test_allow_chess_move_bishop(self):
+ board = GameBoard.new_game([{'actions': []}])
+ board.board_locations = self.generate_locations()
+ board.allow_chess_move(MOVES.BISHOP)
+ self.assertEqual(board.player.movement_mode, MOVES.BISHOP)
+
+ def test_allow_chess_move_castle(self):
+ board = GameBoard.new_game([{'actions': []}])
+ board.board_locations = self.generate_locations()
+ board.allow_chess_move(MOVES.CASTLE)
+ self.assertEqual(board.player.movement_mode, MOVES.CASTLE)
+
+
class TestLocationCard(TestCase):
def test_generate_bitwise_operand(self):
[action] = location.actions
self.assertEqual(type(action), actions.DoNothing)
self.assertEqual(action.required_bits, set())
+
+ def test_parse_bits(self):
+ self.assertEqual(
+ LocationCard.parse_bits([]), frozenset([]))
+ self.assertEqual(
+ LocationCard.parse_bits(['RED']), frozenset([BITS.RED]))
+ self.assertEqual(
+ LocationCard.parse_bits([BITS.BLUE]), frozenset([BITS.BLUE]))
+ self.assertEqual(
+ LocationCard.parse_bits([BITS.NORTH, 'MSB']),
+ frozenset([BITS.NORTH, BITS.MSB]))