ShapeHandle shape_handle = c.output(i);
    TF_ShapeAndType& shape = output_shapes_result->items[i];
    shape.num_dims = c.Rank(shape_handle);
    if (shape.num_dims == InferenceContext::kUnknownRank) {
      shape.dims = nullptr;
      continue;
    }
    shape.dims = new int64_t[shape.num_dims];
    for (size_t j = 0; j < shape.num_dims; ++j) {
      shape.dims[j] = c.Value(c.Dim(shape_handle, j));
    }
  }

  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,

        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;

    }

    @Test
    @DisplayName("Test equals method with same netName")
    void testEqualsWithSameNetName() {
        SmbShareInfo info1 = new SmbShareInfo("Share1", 0, "Remark1");
        SmbShareInfo info2 = new SmbShareInfo("Share1", 1, "Remark2");

        // Same netName, different type and remark
        assertTrue(info1.equals(info2));
        assertTrue(info2.equals(info1));
    }

    @Test

    void testResourceCreationConsistency() throws CIFSException {
        // Test that resource creation works consistently
        String[] validUrls = { "smb://server1/share1/", "smb://server2/share2/file.txt" };

        for (String validUrl : validUrls) {
            // When
            SmbResource resource = context.get(validUrl);

            // Then

        srvsvc.ShareInfoCtr0 ctr0 = new srvsvc.ShareInfoCtr0();
        ctr0.count = 2;
        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

"$python" -m pip install *.whl $TFCI_PYTHON_VERIFY_PIP_INSTALL_ARGS
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();

      return samples;
    }

    @Override
    public Set<V> create(Object... elements) {
      @SuppressWarnings("unchecked")
      V[] valuesArray = (V[]) elements;

      // Start with a suitably shaped collection of entries
      Collection<Entry<K, V>> originalEntries = mapGenerator.getSampleElements(elements.length);

      // Create a copy of that, with the desired value for each value

    @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);