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

            // 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

    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

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

"$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.

    }

    @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

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

        // Protocol should be consistent

      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