namespace tensorflow {

// Set the shapes and types of the output's handle.
//
// The lengths of the arrays pointed to by `shapes`, `ranks`, and `types` must
// all be equal to `num_shapes_and_types`. If `ranks[i] != -1`, (i.e., if the
// rank is known), then it must be equal to the length of `shapes[i]`; if
// `ranks[i] == 1`, then `shapes[i]` may be nullptr.
//

 public:
  CheckpointReader(const string& filename, TF_Status* status);

  bool HasTensor(const string& name) const;
  const string DebugString() const;

  // Returns a map from variable names to their shapes.  Slices of a partitioned
  // tensor are combined into a single entry.
  const TensorSliceReader::VarToShapeMap& GetVariableToShapeMap() const;

  // Returns a map from variable names to their data types.  Slices of a

        assertEquals(2, response.results.length, "Should have 2 share entries.");

        SmbShareInfo share1 = (SmbShareInfo) response.results[0];
        assertEquals("SHARE1", share1.getName().trim(), "Share 1 name should be correct.");
        assertEquals(0, share1.type, "Share 1 raw type should be 0.");
        assertEquals("Remark 1", share1.remark.trim(), "Share 1 remark should be correct.");

        // Create real ShareInfo1 objects instead of mocks
        srvsvc.ShareInfo1 shareInfo1 = new srvsvc.ShareInfo1();
        shareInfo1.netname = "Share1";
        shareInfo1.type = 0;
        shareInfo1.remark = "Remark for Share1";

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

    }

    /**
     * Test of equals method, of class SmbShareInfo.
     */
    @Test
    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();

        DfsEnumArray200 dfsEnumArray = new DfsEnumArray200();
        dfsEnumArray.count = 2;
        dfsEnumArray.s = new DfsInfo200[2];
        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;

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

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.

        // Session starts with usage count > 0
        assertTrue(session.isInUse(), "Session should be in use initially");

        // Create and release trees
        SmbTreeImpl tree = session.getSmbTree("share1", null);
        assertNotNull(tree, "Tree should be created");
        tree.release();

        // Release the session (back to initial state)
        session.release();

    @Test
    @DisplayName("Handler correctly identifies itself as SMB handler")
    void testHandlerEquality() throws Exception {
        // Create two URLs with the same handler
        URL url1 = new URL(null, "smb://host1/share1", handler);
        URL url2 = new URL(null, "smb://host2/share2", handler);

        // Both should use the same handler instance
        assertNotNull(url1);
        assertNotNull(url2);