shareInfo1.type = 0;
        shareInfo1.remark = "Remark for Share1";

        srvsvc.ShareInfo1 shareInfo2 = new srvsvc.ShareInfo1();
        shareInfo2.netname = "Share2";
        shareInfo2.type = 1;
        shareInfo2.remark = "Remark for Share2";

        // Create ShareInfoCtr1 with the shares
        srvsvc.ShareInfoCtr1 shareInfoCtr1 = new srvsvc.ShareInfoCtr1();
        shareInfoCtr1.count = 2;

  CHECK_EQ(TF_OK, TF_GetCode(status_)) << TF_Message(status_);

  // Infer shape when everything is known.
  CheckOutputShapes(matmul_op,
                    /*input_shapes*/ {make_shape({3, 2}), make_shape({2, 4})},
                    /*input_tensors*/ {},
                    /*expected_shape*/ make_shape({3, 4}));

  // Infer shape when second operand has unknown shape.
  CheckOutputShapes(matmul_op,

        dfsEnumArray.s[0] = new DfsInfo200();
        dfsEnumArray.s[0].dfs_name = "\\domain\share1";
        dfsEnumArray.s[1] = new DfsInfo200();
        dfsEnumArray.s[1].dfs_name = "\\domain\share2";

        // Manually set the DfsEnumArray200 to the info struct
        dfsRootEnum.info.e = dfsEnumArray;

        // When
        FileEntry[] entries = dfsRootEnum.getEntries();

        // Then

    void testEquals() {
        SmbShareInfo instance1 = new SmbShareInfo("SHARE1", 0, "remark1");
        SmbShareInfo instance2 = new SmbShareInfo("SHARE1", 1, "remark2");
        SmbShareInfo instance3 = new SmbShareInfo("SHARE2", 0, "remark1");
        Object notAShareInfo = new Object();

        assertTrue(instance1.equals(instance2)); // Should be equal based on netName
        assertFalse(instance1.equals(instance3)); // Should not be equal

  if (reader_ != nullptr) {
    status = reader_->GetTensor(name, out_tensor);
  } else {
    tensorflow::DataType dtype;
    tensorflow::TensorShape shape;
    status = v2_reader_->LookupDtypeAndShape(name, &dtype, &shape);
    if (status.ok()) {
      out_tensor->reset(new Tensor(dtype, shape));
      status = v2_reader_->Lookup(name, out_tensor->get());
      if (!status.ok()) out_tensor->reset();
    }
  }
  if (!status.ok()) {

            // Setup second referral
            DfsReferralDataImpl second = new DfsReferralDataImpl();
            when(mockReferral.getNode()).thenReturn("\\server2\\share2\\path2");
            second = DfsReferralDataImpl.fromReferral(mockReferral, "\\\\server2\\share2\\path2", System.currentTimeMillis() + 20000, 15);
            second.setDomain("DOMAIN");

            // Combine

        ctr0.array = new srvsvc.ShareInfo0[2];
        ctr0.array[0] = new srvsvc.ShareInfo0();
        ctr0.array[0].netname = "Share1";
        ctr0.array[1] = new srvsvc.ShareInfo0();
        ctr0.array[1].netname = "Share2";

        // Set up mocks
        mockDeferredBuffer.index = 0; // Set field directly
        mockDeferredBuffer.deferred = mockDeferredBuffer;

        ctr0.encode(mockNdrBuffer);

fi
if [[ "$TFCI_WHL_IMPORT_TEST_ENABLE" == "1" ]]; then
  "$python" -c 'import tensorflow as tf; t1=tf.constant([1,2,3,4]); t2=tf.constant([5,6,7,8]); print(tf.add(t1,t2).shape)'
  "$python" -c 'import sys; import tensorflow as tf; sys.exit(0 if "keras" in tf.keras.__name__ else 1)'
fi
# Import tf nightly wheel built with numpy2 from PyPI in numpy1 env for testing.

  TF_AddInput(desc, r);
  return TF_FinishOperation(desc, s);
}

TF_Operation* RandomUniform(TF_Operation* shape, TF_DataType dtype,
                            TF_Graph* graph, TF_Status* s) {
  TF_OperationDescription* desc =
      TF_NewOperation(graph, "RandomUniform", "random_uniform");
  TF_AddInput(desc, {shape, 0});
  TF_SetAttrType(desc, "dtype", dtype);
  return TF_FinishOperation(desc, s);
}

    }

    @Test
    void testRegistrationIdUniqueness() throws Exception {
        WitnessRegistration registration2 =
                new WitnessRegistration("\\\\server\\share2", InetAddress.getByName("192.168.1.101"), WitnessServiceType.CLUSTER_WITNESS);

        assertNotEquals(registration.getRegistrationId(), registration2.getRegistrationId());
    }

    @Test