def __init__(self, game=None): if game is None: self.game = GamePlay() else: self.game = game self.clear_board()
class TestBoard(object): def __init__(self, game=None): if game is None: self.game = GamePlay() else: self.game = game self.clear_board() def clear_board(self): self.game.state.num_bottom_pieces = 0 self.game.state.num_top_pieces = 0 self.game.board.pieces = [[None] * Board.DEFAULT_WIDTH for row in self.game.board.pieces] def place_piece(self, row, col, origin): self.game.set_piece(row, col, Piece(origin)) def place_king(self, row, col, origin): self.place_piece(row, col, origin); self.game.board.get_piece(row, col).set_king(True)
def setUp(self): self.game = GamePlay() self.tboard = TestBoard()
class BoardTest(unittest.TestCase): def setUp(self): self.game = GamePlay() self.tboard = TestBoard() def test_board_initial_configuration_correct(self): for col in range (0, Board.DEFAULT_WIDTH): [self.check_piece(row, col) for row in range(0, Board.DEFAULT_HEIGHT)] def check_piece(self, row, col): piece = self.game.board.get_piece(row, col) if row < 3 and (row + col) % 2 == 0: self.assertIsNotNone(piece) self.assertTrue(piece.is_from_top()) elif row > 4 and (row + col) % 2 == 0: self.assertIsNotNone(piece) self.assertTrue(piece.is_from_bottom()) else: self.assertIsNone(piece) def test_move_piece_to_invalid_location_does_nothing(self): original_board = Board() self.game.move_piece((2, 2), (5, 6)) for row in range (0, Board.DEFAULT_HEIGHT): [self.check_pieces_equal(self.game.board.get_piece(row, col), original_board.get_piece(row, col)) for col in range(0, Board.DEFAULT_WIDTH)] def check_pieces_equal(self, piece1, piece2): if piece1 is None or piece2 is None: self.assertIsNone(piece1) self.assertIsNone(piece2) return self.assertEqual(piece1.get_origin(), piece2.get_origin()) def test_move_piece_to_valid_location_only_works_if_location_is_in_move_list(self): self.assertIsNone(self.game.board.get_piece(3, 1)) self.game.move_piece((2, 0), (3, 1)) self.assertIsNone(self.game.board.get_piece(2, 0)) self.assertIsNotNone(self.game.board.get_piece(3, 1)) self.assertTrue(self.game.board.get_piece(3, 1).is_from_top()) self.assertFalse(self.game.is_game_over()) def test_game_over_when_one_type_of_piece_runs_out_simple_board(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ _ _ _ _ _ _ _# #1 x _ x _ _ _ _ _# #2 _ t _ _ _ _ _ _# #3 x _ B _ _ _ _ _# #4 _ _ _ x _ _ _ _# #5 _ _ _ _ _ _ _ _# #6 _ _ _ _ _ _ _ _# #7 _ _ _ _ _ _ _ _# # # # # # # # # # # self.tboard.place_king(2, 1, origin.TOP) self.tboard.place_piece(3, 2, origin.BOTTOM) self.assertFalse(self.tboard.game.is_game_over()) self.tboard.game.move_piece((2, 1), (4, 3)) self.assertTrue(self.tboard.game.is_game_over()) self._check_board_empty_except([(4, 3)]) self.assertTrue(self.tboard.game.is_game_over()) self.assertEqual(1, self.tboard.game.state.num_top_pieces) self.assertEqual(0, self.tboard.game.state.num_bottom_pieces) def _check_board_empty_except(self, list_of_exceptions): for row in range(0, Board.DEFAULT_HEIGHT): for col in range (0, Board.DEFAULT_WIDTH): if (row, col) in list_of_exceptions: continue self.assertIsNone(self.tboard.game.get_piece(row, col), "There is a piece at " + str(row) + ", " + str(col)) for loc in list_of_exceptions: self.assertIsNotNone(self.tboard.game.get_piece(loc[0], loc[1]), "There is NO piece at " + str(row) + ", " + str(col)) def test_game_over_when_one_type_of_piece_runs_out_less_simple_board(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ _ _ _ _ _ _ _# #1 x _ x _ _ _ _ _# #2 _ t _ _ _ _ _ _# #3 x _ B _ _ _ _ _# #4 _ _ _ x _ B _ _# #5 _ _ _ _ _ _ _ _# #6 _ _ _ _ _ _ _ _# #7 _ _ _ _ _ _ _ _# # # # # # # # # # # self.tboard.place_king(2, 1, origin.TOP) self.tboard.place_piece(3, 2, origin.BOTTOM) self.tboard.place_piece(4, 5, origin.BOTTOM) self.assertFalse(self.tboard.game.is_game_over()) ## The counts work because when you move a piece, you're removing it from the board.-- ## And when you put it back somewhere else, you're incrementing once more.++ self.tboard.game.move_piece((2, 1), (4, 3)) self._check_board_configuration(num_top=1, num_bottom=1, non_empty_cells=[(4, 3), (4, 5)]) self.tboard.game.move_piece((4, 5), (3, 4)) self._check_board_configuration(num_top=1, num_bottom=1, non_empty_cells=[(4, 3), (3, 4)]) self.tboard.game.move_piece((4, 3), (2, 5)) self._check_board_configuration(num_top=1, num_bottom=0, non_empty_cells=[(2, 5)]) def test_board_has_current_origin_pass_if_no_moves_for_that_origin(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ T _ T _ T _ _# #1 T _ T _ _ _ _ _# #2 _ B _ _ _ _ _ _# #3 _ _ _ _ _ _ _ _# #4 _ _ _ _ _ _ _ _# #5 _ _ _ _ _ _ _ _# #6 _ _ _ _ _ _ _ _# #7 _ _ _ _ _ _ _ _# # # # # # # # # # # self.tboard.place_piece(0, 1, origin.TOP) self.tboard.place_piece(0, 3, origin.TOP) self.tboard.place_piece(0, 5, origin.TOP) self.tboard.place_piece(1, 0, origin.TOP) self.tboard.place_piece(1, 2, origin.TOP) self.tboard.place_piece(2, 1, origin.BOTTOM) self.tboard.game.move_piece((0, 5), (1, 4)) self.assertEqual(origin.TOP, self.tboard.game.current_turn.origin) def test_game_over_if_two_passes_occur_in_a_row(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ _ _ _ _ _ _ _# #1 _ _ _ _ _ _ _ _# #2 _ _ T _ _ _ _ _# #3 _ _ _ _ _ _ _ _# #4 T _ T _ T _ T _# #5 _ T _ T _ T _ T# #6 B _ B _ B _ B _# #7 _ B _ B _ B _ B# # # # # # # # # # # self.tboard.place_piece(2, 2, origin.TOP) self.tboard.place_piece(4, 0, origin.TOP) self.tboard.place_piece(4, 2, origin.TOP) self.tboard.place_piece(4, 4, origin.TOP) self.tboard.place_piece(4, 6, origin.TOP) self.tboard.place_piece(5, 1, origin.TOP) self.tboard.place_piece(5, 3, origin.TOP) self.tboard.place_piece(5, 5, origin.TOP) self.tboard.place_piece(5, 7, origin.TOP) self.tboard.place_piece(6, 0, origin.BOTTOM) self.tboard.place_piece(6, 2, origin.BOTTOM) self.tboard.place_piece(6, 4, origin.BOTTOM) self.tboard.place_piece(6, 6, origin.BOTTOM) self.tboard.place_piece(7, 1, origin.BOTTOM) self.tboard.place_piece(7, 3, origin.BOTTOM) self.tboard.place_piece(7, 5, origin.BOTTOM) self.tboard.place_piece(7, 7, origin.BOTTOM) self.tboard.game.move_piece((2, 2), (3, 1)) self.assertTrue(self.tboard.game.is_game_over()) def _check_board_configuration(self, num_top, num_bottom, non_empty_cells): self._check_board_empty_except(non_empty_cells) self.assertEqual(num_top, self.tboard.game.state.num_top_pieces) self.assertEqual(num_bottom, self.tboard.game.state.num_bottom_pieces) self.assertEqual(num_top == 0 or num_bottom == 0, self.tboard.game.is_game_over())
def setUp(self): self.tboard = TestBoard() self.game = GamePlay() self.slotting = slotting.Slotting(self.game)
class SlottingTest(unittest.TestCase): def setUp(self): self.tboard = TestBoard() self.game = GamePlay() self.slotting = slotting.Slotting(self.game) def test_piece_gets_selected_and_slotted_when_slot_is_valid_move_and_when_move_is_complete_turn_is_ended(self): self.assertIsNone(self.game.current_turn.piece) self.slotting.select_piece(MockEvent(120, 120)) self.assertEqual(2, self.slotting.start_row) self.assertEqual(2, self.slotting.start_col) self.slotting.release_piece(MockEvent(100, 200)) self.assertIsNone(self.game.get_piece(2, 2)) self.assertIsNotNone(self.game.get_piece(3, 1)) self.assertIsNone(self.game.current_turn.piece) def test_piece_does_not_get_placed_in_new_slot_when_invalid_move_and_start_row_and_col_reset(self): self.slotting.select_piece(MockEvent(100, 100)) self.assertEqual(1, self.slotting.start_row) self.assertEqual(1, self.slotting.start_col) self.slotting.release_piece(MockEvent(300, 300)) self.assertIsNotNone(self.game.get_piece(1, 1)) self.assertIsNotNone(self.game.get_piece(2, 2)) self.assertIsNone(self.game.get_piece(3, 1)) self.assertIsNone(self.game.get_piece(4, 4)) self.assertIsNone(self.slotting.start_row) self.assertIsNone(self.slotting.start_col) def test_when_doing_consecutive_moves_starting_row_and_col_changes(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ _ _ _ _ _ _ _# #1 _ _ _ _ _ _ _ _# #2 _ T _ _ _ _ _ _# #3 x _ B _ _ _ _ _# #4 _ _ _ x _ _ _ _# #5 _ _ _ _ B _ _ _# #6 _ _ _ _ _ x _ _# #7 _ _ _ _ _ _ _ _# # # # # # # # # # # self.tboard.place_piece(2, 1, origin.TOP) self.tboard.place_piece(3, 2, origin.BOTTOM) self.tboard.place_piece(5, 4, origin.BOTTOM) self.slotting = slotting.Slotting(self.tboard.game) self.board = self.tboard.game self.slotting.select_piece(MockEvent(100, 140)) self.assertEqual(2, self.slotting.start_row) self.assertEqual(1, self.slotting.start_col) self.slotting.release_piece(MockEvent(200, 250)) self.assertIsNone(self.board.get_piece(2, 1)) self.assertIsNotNone(self.board.get_piece(3, 2)) self.assertIsNotNone(self.board.get_piece(4, 3)) self.assertIsNotNone(self.board.current_turn.piece) self.slotting.select_piece(MockEvent(200, 250)) self.assertEqual(4, self.slotting.start_row) self.assertEqual(3, self.slotting.start_col) self.slotting.release_piece(MockEvent(330,400)) self.assertIsNone(self.board.current_turn.piece) self.assertIsNone(self.board.get_piece(3, 2)) self.assertIsNone(self.board.get_piece(5, 4)) self.assertIsNotNone(self.board.get_piece(6, 5)) def test_slotting_releases_piece_fully_and_does_not_grab_new_piece_in_same_loc(self): # # # # # # # # # # # 0 1 2 3 4 5 6 7# #0 _ _ _ _ _ _ _ _# #1 _ _ _ _ _ _ _ _# #2 _ _ _ _ _ _ _ _# #3 _ _ _ T _ _ _ _# #4 _ _ B _ _ _ _ _# #5 _ x _ _ _ _ _ _# #6 _ _ T _ _ _ _ _# #7 _ _ _ B _ _ _ _# # # # # # # # # # # self.tboard.place_piece(3, 3, origin.TOP) self.tboard.place_piece(4, 2, origin.BOTTOM) self.tboard.place_piece(6, 2, origin.TOP) self.tboard.place_piece(7, 3, origin.BOTTOM) self.slotting = slotting.Slotting(self.tboard.game) self.board = self.tboard.game self.slotting.select_piece(MockEvent(200, 200)) self.assertEqual(3, self.slotting.start_row) self.assertEqual(3, self.slotting.start_col) self.slotting.release_piece(MockEvent(80, 330)) self.assertIsNone(self.board.get_piece(3, 3)) self.assertIsNone(self.board.get_piece(4, 2)) self.assertIsNotNone(self.board.get_piece(5, 1)) self.assertFalse(self.slotting.is_holding_piece()) def test_slotting_is_holding_piece(self): self.assertFalse(self.slotting.is_holding_piece()) self.slotting.start_row = 5 self.slotting.start_col = 5 self.assertTrue(self.slotting.is_holding_piece())