class TestGameBoard(TestCase):
def test_export_new_board(self):
- board = GameBoard.new_game(4, 4, [])
- self.assertEqual(board.export(), {
+ board = GameBoard.new_game([{'actions': []}])
+ 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': [],
- 'board_locations': {},
+ 'locations': [{'actions': []}],
'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'], [])
+ self.assertTrue(2 <= len(location_state['bitwise_operand']) <= 3)
def test_lose_health(self):
- board = GameBoard.new_game(4, 4, [])
+ board = GameBoard.new_game([{'actions': []}])
self.assertEqual(board.health, 4)
state_1 = board.export()