tests/testthat/test-mappings-and-scores-work.R

In USCbiostats/partition: Agglomerative Partitioning Framework for Dimension Reduction

set.seed(1234)
prt <- partition(df, threshold = .6)

test_that("mappings work", {
  key <- mapping_key(prt)
  expect_s3_class(key, "tbl")
  expect_length(key, 4)
  expect_equal(nrow(key), ncol(partition_scores(prt)))

  long_key <- unnest_mappings(prt)
  expect_s3_class(long_key, "tbl")
  expect_length(long_key, 4)
  expect_equal(nrow(long_key), ncol(df))
  expect_equal(tidyr::unnest(key, cols = c(mapping, indices)), long_key)

  expect_true(nrow(long_key) > nrow(key))
  expect_true(nrow(unnest_mappings(prt)) > nrow(unnest_reduced(prt)))

  gnames <- mapping_groups(prt)
  gind <- mapping_groups(prt, indices = TRUE)
  expect_equal(length(gnames), length(gind))
  expect_equal(length(gnames), nrow(key))
  expect_equal(length(unlist(gnames)), nrow(long_key))
  expect_type(gnames[[1]], "character")
  expect_type(gind[[1]], "integer")
})


test_that("reduced scores work", {
  scores <- partition_scores(prt)
  expect_s3_class(scores, "tbl")
  expect_equal(nrow(scores), nrow(df))
})