int rank = shape.dims();
  if (num_dims != rank) {
    status->status = InvalidArgument("Expected rank is ", num_dims,
                                     " but actual rank is ", rank);
    return;
  }
  for (int i = 0; i < num_dims; i++) {
    dims[i] = shape.dim_size(i);
  }
}

int TF_CheckpointReaderGetVariableNumDims(TF_CheckpointReader* reader,

    //                                                                        ============
    /**
     * Verifies that the CRUD mode matches the expected mode.
     *
     * @param crudMode the actual CRUD mode
     * @param expectedMode the expected CRUD mode
     */
    protected void verifyCrudMode(final int crudMode, final int expectedMode) {
        if (crudMode != expectedMode) {

				CalculatedSHA256: "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
			},
		},
		// Test case - 14.
		// Input with size more than the size of actual data inside the reader.
		{
			bucketName: bucket, objName: object, uploadID: uploadID, PartID: 1, inputReaderData: "abcd", inputMd5: "e2fc714c4727ee9395f324cd2e7f3335", inputDataSize: int64(len("abcd") + 1),

)

const (
	// MaxScanTokenSize is the maximum size used to buffer a token
	// unless the user provides an explicit buffer with [Scanner.Buffer].
	// The actual maximum token size may be smaller as the buffer
	// may need to include, for instance, a newline.
	MaxScanTokenSize = 64 * 1024

	startBufSize = 4096 // Size of initial allocation for buffer.
)

        // Decode and check return value
        int bytesConsumed = fileBothDirectoryInfo.decode(buffer, 0, buffer.length);

        // Verify bytes consumed matches the actual data size
        assertTrue(bytesConsumed < 0); // Return value is negative (start - bufferIndex)
        assertEquals(-94 - filename.length() * 2, bytesConsumed); // Base structure + filename length
    }

    @Test

        int wireRemarkOffset = actualRemarkOffset + converterValue;
        SMBUtil.writeInt4(wireRemarkOffset, buffer, bufferIndex);
        bufferIndex += 4;

        // Write remark at actual offset
        byte[] remarkBytes = remark.getBytes(StandardCharsets.US_ASCII);
        System.arraycopy(remarkBytes, 0, buffer, actualRemarkOffset, remarkBytes.length);
        buffer[actualRemarkOffset + remarkBytes.length] = 0;

        // Execute and ensure no exceptions are thrown
        QueryBuilder result = matchAllQueryCommand.execute(context, query, boost);

        assertNotNull(result);

        // Note: Actual log output verification would require a log appender mock
        // but we can at least ensure the method executes without errors
        logger.info("Execute method called with query: {} and boost: {}", query, boost);
    }

        void testGetHandleNcacnNpProtocol() {
            // When/Then: Should attempt to create DcerpcPipeHandle (may fail due to SMB URL creation)
            // This tests the protocol recognition logic, actual creation may fail due to missing SMB support
            assertThrows(Exception.class, () -> DcerpcHandle.getHandle("ncacn_np:\\\\server[endpoint=\\pipe\\srvsvc]", mockContext));
        }

        @Test

        assertEquals(expectedLength, ndrBuffer.getLength());

        // Verify content (simplified check, full verification would involve decoding)
        assertEquals(testString.length() + 1, Encdec.dec_uint32le(buffer, 0)); // Actual count
        assertEquals(0, Encdec.dec_uint32le(buffer, 4)); // Offset
        assertEquals(testString.length() + 1, Encdec.dec_uint32le(buffer, 8)); // Max count

        ndrBuffer.setIndex(0);

                .thenThrow(new SmbException("transport error", new jcifs.util.transport.TransportException()))
                .thenReturn(resp);

        setTree(c, tree);

        // Override connectHost to avoid actual network connection on retry
        doAnswer(invocation -> {
            SmbTreeHandleImpl handle = mock(SmbTreeHandleImpl.class);
            // Re-set the same tree to continue using our mock
            setTree(c, tree);