tests/testthat/test-stages_kmeans.R

In stagedtrees: Staged Event Trees

test_that("stages_kmeans", {
  fl <- full(PhDArticles[, 1:4], lambda = 0.5)
  expect_silent(mod <- stages_kmeans(join_unobserved(fl, name = "NA"),
    k = c(2, 3), ignore = "NA"
  ))
  expect_silent(mod <- stages_kmeans(join_unobserved(fl, name = "NA"),
    k = c(2, 3),
    scope = "Kids"
  ))
  expect_silent(mod <- stages_kmeans(fl, transform = function(x) x^2))
  expect_silent(mod <- stages_kmeans(fl, transform = NULL))
  expect_silent(mod <- stages_kmeans(fl))
  ll1 <- logLik(mod)
  mod$ll <- NULL
  ll2 <- logLik(mod)
  expect_equal(ll1, ll2)
})

test_that("stages_kmeans fails if not numeric k", {
  mod <- full(Titanic, lambda = 1)
  expect_error(stages_kmeans(mod, k = "a"))
})

test_that("stages_kmeans fails if transofrm is not function or NULL", {
  mod <- full(Titanic, lambda = 1)
  expect_error(stages_kmeans(mod, k = 2, transform = "aaa"))
})