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