def test_transport1(self):
self.board[0, 0] = knight.Piece.transport
self.board[1, 1] = knight.Piece.transport
self.transports = knight._get_transports(self.board)
knight._undo_move(self.board, self.last_move, self.original_board,
self.transports, self.start_x, self.start_y)
self.assertEqual(self.board[0, 2], knight.Piece.current)
self.assertEqual(self.board[1, 4], self.original_board[1, 4])
def test_transport5(self):
self.last_move = 4
self.board[0, 0] = knight.Piece.transport
self.original_board[0, 0] = knight.Piece.transport
self.original_board[1, 4] = knight.Piece.transport
self.transports = [(1, 4), (0, 0)] # manually reverse their order
knight._undo_move(self.board, self.last_move, self.original_board,
self.transports, self.start_x, self.start_y)
self.assertEqual(self.board[1, 2], knight.Piece.current)
self.assertEqual(self.board[1, 4], knight.Piece.transport)
self.assertEqual(self.board[0, 0], knight.Piece.transport)
def test_transport4(self):
self.last_move = 4
self.board[0, 0] = knight.Piece.transport
self.original_board[0, 0] = knight.Piece.transport
self.original_board[1, 4] = knight.Piece.transport
self.transports = [(0, 0), (1, 4)]
knight._undo_move(self.board, self.last_move, self.original_board,
self.transports, self.start_x, self.start_y)
self.assertEqual(self.board[1, 2], knight.Piece.current)
self.assertEqual(self.board[1, 4], knight.Piece.transport)
self.assertEqual(self.board[0, 0], knight.Piece.transport)
def test_other_piece(self):
self.original_board[1, 4] = knight.Piece.water
knight._undo_move(self.board, self.last_move, self.original_board,
self.transports, self.start_x, self.start_y)
self.assertEqual(self.board[0, 2], knight.Piece.current)
self.assertEqual(self.board[1, 4], self.original_board[1, 4])