X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=naja%2Ftests%2Ftest_gameboard.py;h=5dae954fafaae21a532b7a7d842b408f6d210101;hb=9d16741e86873a84b2e518906adf0111b95103dd;hp=0ad0d28a5f218df26b9be12554094bd6a880d1ec;hpb=54b217b57e6b894e261d186319b1337fb53c0210;p=naja.git diff --git a/naja/tests/test_gameboard.py b/naja/tests/test_gameboard.py index 0ad0d28..5dae954 100644 --- a/naja/tests/test_gameboard.py +++ b/naja/tests/test_gameboard.py @@ -2,14 +2,127 @@ from unittest import TestCase from naja.constants import BITS from naja.gameboard import GameBoard, LocationCard +from naja import actions class TestGameBoard(TestCase): + def assert_state(self, state1, state2, exclude=(), player_exclude=()): + def filter_dict(source, exclude_keys): + return dict((k, v) for k, v in source.items() + if k not in exclude_keys) + + state1 = filter_dict(state1, exclude) + if 'player' in state1: + state1['player'] = filter_dict(state1['player'], player_exclude) + state2 = filter_dict(state2, exclude) + if 'player' in state2: + state2['player'] = filter_dict(state2['player'], player_exclude) + + self.assertEqual(state1, state2) + + def test_export_new_board(self): + board = GameBoard.new_game([{'actions': [{ + 'action_class': 'LoseHealthOrMSB', + 'required_bits': [], + }]}]) + exported_state = board.export() + board_locations = exported_state.pop('board_locations') + self.assertEqual(exported_state, { + 'max_health': 4, + 'health': 4, + 'wins_required': 4, + 'wins': 0, + 'locations': [{'actions': [{ + 'action_class': 'LoseHealthOrMSB', + 'required_bits': [], + }]}], + '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(): + self.assertEqual( + sorted(location_state.keys()), ['actions', 'bitwise_operand']) + self.assertEqual(location_state['actions'], [{ + 'action_class': 'LoseHealthOrMSB', + 'required_bits': [], + 'data': {}, + }]) + self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3) + def test_lose_health(self): - board = GameBoard(None, 4, 4, locations={}, state=None) + board = GameBoard.new_game([{'actions': []}]) self.assertEqual(board.health, 4) + state_1 = board.export() + board.lose_health() self.assertEqual(board.health, 3) + state_2 = board.export() + + self.assert_state(state_1, state_2, exclude=['health']) + + def test_gain_health(self): + board = GameBoard.new_game([{'actions': []}]) + 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([{'actions': []}]) + 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([{'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', + })) class TestLocationCard(TestCase): @@ -28,3 +141,17 @@ class TestLocationCard(TestCase): self.assertTrue(sizes[3] > 0) self.assertTrue(sizes[2] > sizes[3]) self.assertEqual(bits, set(BITS.values())) + + def test_new_location_no_actions(self): + location = LocationCard.new_location({'actions': []}) + [action] = location.actions + self.assertEqual(type(action), actions.DoNothing) + self.assertEqual(action.required_bits, set()) + + def test_new_location_one_action(self): + location = LocationCard.new_location({'actions': [ + {'required_bits': [], 'action_class': 'DoNothing'}, + ]}) + [action] = location.actions + self.assertEqual(type(action), actions.DoNothing) + self.assertEqual(action.required_bits, set())