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())
