tests/testthat/test-predict.R

In ddtlcm: Latent Class Analysis with Dirichlet Diffusion Tree Process Prior

test_that("'predict' returns a valid prediction.", {
  data(result_diet_1000iters)
  N <- nrow(result_diet_1000iters$data)
  K <- result_diet_1000iters$setting$K
  burnin <- 50
  summarized_result <- summary(result_diet_1000iters, burnin, relabel = TRUE, be_quiet = TRUE)
  predicted <- predict(summarized_result, result_diet_1000iters$data)
  expect_length(predicted, 2)
  expect_length(predicted$class_assignments, N)
  expect_equal(dim(predicted$predictive_probs), c(N, K))

  expect_error(predict(result_diet_1000iters, result_diet_1000iters$data))
})