+ self.assert_state(state_1, state_2, exclude=['health'])
+
+ def test_gain_health(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.health = 2
+ self.assertEqual(board.health, 2)
+ state_1 = board.export()
+
+ board.gain_health()
+ self.assertEqual(board.health, 3)
+ state_2 = board.export()
+
+ self.assert_state(state_1, state_2, exclude=['health'])
+
+ def test_gain_health_at_max(self):
+ board = GameBoard.new_game(self.make_deck())
+ self.assertEqual(board.health, 4)
+ state_1 = board.export()
+
+ board.gain_health()
+ self.assertEqual(board.health, 4)
+ state_2 = board.export()
+
+ self.assert_state(state_1, state_2)
+
+ def generate_locations(self, override_dict=None):
+ locations_dict = dict(((x, y), '%s%s' % (x, y))
+ for x in range(5) for y in range(5))
+ if override_dict:
+ locations_dict.update(override_dict)
+ return locations_dict
+
+ def test_shift_locations_north(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.shift_locations('NORTH')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (2, 0): '21', (2, 1): '23', (2, 3): '24', (2, 4): '20',
+ }))
+
+ def test_shift_locations_south(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.shift_locations('SOUTH')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (2, 0): '24', (2, 1): '20', (2, 3): '21', (2, 4): '23',
+ }))
+
+ def test_shift_locations_east(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.shift_locations('EAST')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (0, 2): '42', (1, 2): '02', (3, 2): '12', (4, 2): '32',
+ }))
+
+ def test_shift_locations_west(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.shift_locations('WEST')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (0, 2): '12', (1, 2): '32', (3, 2): '42', (4, 2): '02',
+ }))
+
+ def test_rotate_locations_anticlockwise(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('ANTICLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (1, 1): '21', (2, 1): '31', (3, 1): '32',
+ (1, 2): '11', (3, 2): '33',
+ (1, 3): '12', (2, 3): '13', (3, 3): '23',
+ }))
+
+ def test_rotate_locations_anticlockwise_top(self):
+ board = GameBoard.new_game(self.make_deck(), initial_pos=(2, 0))
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('ANTICLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (1, 0): '30', (3, 0): '31',
+ (1, 1): '10', (2, 1): '11', (3, 1): '21',
+ }))
+
+ def test_rotate_locations_anticlockwise_right(self):
+ board = GameBoard.new_game(self.make_deck(), initial_pos=(0, 2))
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('ANTICLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (0, 1): '11', (1, 1): '12',
+ (1, 2): '13',
+ (0, 3): '01', (1, 3): '03',
+ }))
+
+ def test_rotate_locations_anticlockwise_corner(self):
+ board = GameBoard.new_game(self.make_deck(), initial_pos=(0, 4))
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('ANTICLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (0, 3): '13', (1, 3): '14',
+ (1, 4): '03',
+ }))
+
+ def test_rotate_locations_clockwise(self):
+ board = GameBoard.new_game(self.make_deck())
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('CLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (1, 1): '12', (2, 1): '11', (3, 1): '21',
+ (1, 2): '13', (3, 2): '31',
+ (1, 3): '23', (2, 3): '33', (3, 3): '32',
+ }))
+
+ def test_rotate_locations_clockwise_1_3(self):
+ board = GameBoard.new_game(self.make_deck(), initial_pos=(1, 3))
+ board.board_locations = self.generate_locations()
+ board.rotate_locations('CLOCKWISE')
+ self.assertEqual(board.board_locations, self.generate_locations({
+ (0, 2): '03', (1, 2): '02', (2, 2): '12',
+ (0, 3): '04', (2, 3): '22',
+ (0, 4): '14', (1, 4): '24', (2, 4): '23',
+ }))
+
+ def test_allow_chess_move_knight(self):
+ board = GameBoard.new_game(self.make_deck())
+ 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(self.make_deck())
+ 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(self.make_deck())
+ board.allow_chess_move(MOVES.CASTLE)
+ self.assertEqual(board.player.movement_mode, MOVES.CASTLE)