tests/testthat/test-agglomerative.R

In cuda.ml: R Interface for the RAPIDS cuML Suite of Libraries

skip_if_not(run_gpu_tests, "requires the GPU test environment")

context("Agglomerative Clustering")

test_that("cuda_ml_agglomerative_clustering() works as expected", {
  set.seed(0)

  blob_sz <- 50
  blobs <- gen_blobs(blob_sz)
  clusters <- blobs |>
    cuda_ml_agglomerative_clustering(metric = "euclidean", n_clusters = 3L)

  expect_equal(
    clusters$labels,
    c(1L, 2L, 0L) |>
      purrr::map(\(x) rep(x, blob_sz)) |>
      unlist(use.names = FALSE)
  )
})