tests/testthat/test-reconcile_clusterings.R

In EmilHvitfeldt/celery: A Common API to Clustering

test_that("reconciliation works with one-to-one", {
  primary_cluster_assignment <- c(
    "Apple",
    "Apple",
    "Carrot",
    "Carrot",
    "Banana",
    "Banana"
  )
  alt_cluster_assignment <- c("Dog", "Dog", "Cat", "Dog", "Fish", "Fish")

  res <- reconcile_clusterings_mapping(
    primary_cluster_assignment,
    alt_cluster_assignment
  )

  expect_equal(
    res$alt_recoded,
    c("Carrot", "Carrot", "Banana", "Carrot", "Apple", "Apple")
  )
})

test_that("reconciliation works with uneven numbers", {
  primary_cluster_assignment <- c(
    "Apple",
    "Apple",
    "Carrot",
    "Carrot",
    "Banana",
    "Banana"
  )
  alt_cluster_assignment <- c("Dog", "Dog", "Cat", "Dog", "Parrot", "Fish")

  expect_snapshot(
    error = TRUE,
    reconcile_clusterings_mapping(
      primary_cluster_assignment,
      alt_cluster_assignment,
      one_to_one = TRUE
    )
  )

  res <- reconcile_clusterings_mapping(
    primary_cluster_assignment,
    alt_cluster_assignment,
    one_to_one = FALSE
  )

  expect_equal(
    res$alt_recoded,
    c("Apple", "Apple", "Carrot", "Apple", "Banana", "Banana")
  )
})

test_that("reconciliation errors for uneven lengths", {
  expect_snapshot(
    error = TRUE,
    reconcile_clusterings_mapping(
      letters,
      letters[1:10]
    )
  )
})