// Given: Create a new pool for this test to ensure isolation
        SmbTransportPoolImpl testPool = new SmbTransportPoolImpl();
        when(ctx.getTransportPool()).thenReturn(testPool);

        // Create an existing connection
        SmbTransportImpl first = testPool.getSmbTransport(ctx, address, 445, false);

        // When: Request another connection

            // Domains: single domain named TESTDOM
            lsarpc.LsarRefDomainList domains = new lsarpc.LsarRefDomainList();
            domains.count = 1;
            domains.max_count = 1;
            domains.domains = new lsarpc.LsarTrustInformation[1];
            domains.domains[0] = new lsarpc.LsarTrustInformation();
            domains.domains[0].name = new UnicodeString("TESTDOM", false);

        assertTrue(isPerfectSquare);
      } catch (ArithmeticException e) {
        assertFalse(isPerfectSquare);
      }
    }
  }

  @GwtIncompatible // 2147483646^2 expected=4
  public void testPow() {
    for (int i : ALL_INTEGER_CANDIDATES) {
      for (int pow : EXPONENTS) {
        assertEquals(i + "^" + pow, BigInteger.valueOf(i).pow(pow).intValue(), IntMath.pow(i, pow));
      }
    }
  }

        assertTrue(isPerfectSquare);
      } catch (ArithmeticException e) {
        assertFalse(isPerfectSquare);
      }
    }
  }

  @GwtIncompatible // 2147483646^2 expected=4
  public void testPow() {
    for (int i : ALL_INTEGER_CANDIDATES) {
      for (int pow : EXPONENTS) {
        assertEquals(i + "^" + pow, BigInteger.valueOf(i).pow(pow).intValue(), IntMath.pow(i, pow));
      }
    }
  }

        assertTrue(isPerfectSquare);
      } catch (ArithmeticException e) {
        assertFalse(isPerfectSquare);
      }
    }
  }

  @GwtIncompatible // TODO
  public void testPow() {
    for (long i : ALL_LONG_CANDIDATES) {
      for (int exp : EXPONENTS) {
        assertEquals(LongMath.pow(i, exp), valueOf(i).pow(exp).longValue());
      }
    }
  }

        assertEquals("/etc/passwd", AdminLogAction.sanitizeFilename("../../etc/passwd"));
        assertEquals("test.log", AdminLogAction.sanitizeFilename("..test.log"));
        assertEquals("testlog", AdminLogAction.sanitizeFilename("test..log"));
    }

    @Test
    public void test_sanitizeFilename_removesMultipleDoubleDots() {
        // "...." becomes "" (two ".." are removed)

    for (long a : TEST_LONGS) {
      assertThrows(
          ArithmeticException.class,
          () -> UnsignedLong.fromLongBits(a).dividedBy(UnsignedLong.ZERO));
    }
  }

  public void testMod() {
    for (long a : TEST_LONGS) {
      for (long b : TEST_LONGS) {
        if (b != 0) {
          UnsignedLong aUnsigned = UnsignedLong.fromLongBits(a);

  }

  public void testAbsent() {
    Optional<String> optionalName = Optional.absent();
    assertFalse(optionalName.isPresent());
  }

  public void testOf() {
    assertThat(Optional.of("training").get()).isEqualTo("training");
  }

  public void testOf_null() {
    assertThrows(NullPointerException.class, () -> Optional.of(null));
  }

    }

    // Test getThumbnailFile with existing file
    @Test
    public void test_getThumbnailFile_exists() throws IOException {
        Map<String, Object> docMap = new HashMap<>();
        docMap.put("_id", "testdoc");

        String filename = thumbnailManager.getImageFilename(docMap);
        File thumbnailFile = new File(tempDir, filename);
        thumbnailFile.getParentFile().mkdirs();
        thumbnailFile.createNewFile();

    ],
)

tf_custom_op_library(
    name = "test_op1.so",
    srcs = ["test_op1.cc"],
)

tf_kernel_library(
    name = "test_op_kernel",
    srcs = ["test_op.cc"],
    deps = [
        "//tensorflow/core:framework",
        "//tensorflow/core:lib",
    ],
    alwayslink = 1,
)

tf_cuda_cc_test(
    name = "env_test",
    size = "medium",