X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_gameboard.py;h=e3aa32e03be48a27e0da319219fa880e25491c77;hb=75d1f94ef500df4bea81d8988757d37221a310dd;hp=bbc51284cfcb9847b6cfe8e8e194d482f126c802;hpb=4c5f45d39c034c4fc78de5f5a75e16e6ad918b95;p=naja.git diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py index bbc5128..e3aa32e 100644 --- a/naja/tests/test_gameboard.py +++ b/naja/tests/test_gameboard.py @@ -1,6 +1,6 @@ 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 @@ -21,10 +21,15 @@ class TestGameBoard(TestCase): self.assertEqual(state1, state2) def test_export_new_board(self): - board = GameBoard.new_game([{'actions': [{ + board = GameBoard.new_game([{'actions': [ + { 'action_class': 'LoseHealthOrMSB', 'required_bits': [], - }]}]) + }, { + 'action_class': 'GainHealth', + 'required_bits': [BITS.RED], + }, + ]}]) exported_state = board.export() board_locations = exported_state.pop('board_locations') self.assertEqual(exported_state, { @@ -32,24 +37,36 @@ class TestGameBoard(TestCase): 'health': 4, 'wins_required': 4, 'wins': 0, - 'locations': [{'actions': [{ - 'action_class': 'LoseHealthOrMSB', - 'required_bits': [], - }]}], + 'locations': [{'actions': [ + { + 'action_class': 'LoseHealthOrMSB', + 'required_bits': [], + }, { + 'action_class': 'GainHealth', + 'required_bits': [BITS.RED], + }, + ]}], 'player': board.player.export(), }) - self.assertEqual( - set(board_locations.keys()), - set((x, y) for x in range(5) for y in range(5))) - for location_state in board_locations.values(): + positions = [] + for position, location_state in board_locations: + positions.append(position) self.assertEqual( sorted(location_state.keys()), ['actions', 'bitwise_operand']) - self.assertEqual(location_state['actions'], [{ - 'action_class': 'LoseHealthOrMSB', - 'required_bits': [], - 'data': {}, - }]) + self.assertEqual(location_state['actions'], [ + { + 'action_class': 'LoseHealthOrMSB', + 'required_bits': [], + 'data': {}, + }, { + 'action_class': 'GainHealth', + 'required_bits': [BITS.RED], + 'data': {}, + }, + ]) self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3) + self.assertEqual( + positions, sorted((x, y) for x in range(5) for y in range(5))) def test_lose_health(self): board = GameBoard.new_game([{'actions': []}]) @@ -85,6 +102,108 @@ class TestGameBoard(TestCase): 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([{'actions': []}]) + 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([{'actions': []}]) + 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([{'actions': []}]) + 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([{'actions': []}]) + 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_clockwise(self): + board = GameBoard.new_game([{'actions': []}]) + board.board_locations = self.generate_locations() + board.rotate_locations('CLOCKWISE') + 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_clockwise_top(self): + board = GameBoard.new_game([{'actions': []}], initial_pos=(2, 0)) + board.board_locations = self.generate_locations() + board.rotate_locations('CLOCKWISE') + 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_clockwise_right(self): + board = GameBoard.new_game([{'actions': []}], initial_pos=(0, 2)) + board.board_locations = self.generate_locations() + board.rotate_locations('CLOCKWISE') + 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_clockwise_corner(self): + board = GameBoard.new_game([{'actions': []}], initial_pos=(0, 4)) + board.board_locations = self.generate_locations() + board.rotate_locations('CLOCKWISE') + self.assertEqual(board.board_locations, self.generate_locations({ + (0, 3): '13', (1, 3): '14', + (1, 4): '03', + })) + + def test_rotate_locations_anticlockwise(self): + board = GameBoard.new_game([{'actions': []}]) + board.board_locations = self.generate_locations() + board.rotate_locations('ANTICLOCKWISE') + 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_allow_chess_move_knight(self): + board = GameBoard.new_game([{'actions': []}]) + 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.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.allow_chess_move(MOVES.CASTLE) + self.assertEqual(board.player.movement_mode, MOVES.CASTLE) + class TestLocationCard(TestCase): def test_generate_bitwise_operand(self): @@ -116,3 +235,14 @@ class TestLocationCard(TestCase): [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]))