test_semicont_multi.R
In hmi: Hierarchical Multiple Imputation

context("semicont_multi")
library(testthat)
library(hmi)

set.seed(123)
y_imp <- iris[, 1]
y_imp[sample(1:length(y_imp), size = 60)] <- 0
y_imp[sample(1:length(y_imp), size = 20)] <- NA
X_imp <- cbind(1, iris[, 2:4])
Z_imp1 <- data.frame(rep(1, nrow(iris)))
Z_imp2 <- data.frame(cbind(1, iris[, 2]))
clID <- iris$Species


imp1 <- imp_semicont_multi(y_imp = y_imp,
                           X_imp = X_imp,
                           Z_imp = Z_imp1,
                           clID = clID,
                           pvalue = 1)[[1]]

imp2 <- imp_semicont_multi(y_imp = y_imp,
                           X_imp = X_imp,
                           Z_imp = Z_imp2,
                           clID = clID,
                           pvalue = 1)[[1]]




#test_check("hmi")

test_that("semicont_multi returns plausible values", {
  expect_equal(class(imp1$y_ret), "array")
  expect_equal(sum(is.na(imp1$y_ret)), 0)
  expect_equal(class(imp2$y_ret), "array")
  expect_equal(sum(is.na(imp2$y_ret)), 0)
})