tests/testthat/test-sim_out_miss_pattern.R

In power4mome: Power Analysis for Moderation and Mediation

library(testthat)

test_that("miss_pattern", {

set.seed(1234)
n <- 200
p <- 8
prop <- .50
dat <- matrix(rnorm(n * p),
              n, p)
dat[sample.int(n * p, ceiling(n * prop))] <- NA
# dat[sample.int(n, ceiling(n * (1 - prop))), ] <- 1
dat <- as.data.frame(dat)

out <- miss_pattern(dat)
chk0 <- !is.na(dat)
expect_equal(attr(out, "nvalid"),
             colSums(chk0))
expect_equal(as.numeric(rownames(out)[1]),
             sum(complete.cases(dat)))

# Complete data
set.seed(1234)
n <- 123
p <- 8
dat <- matrix(rnorm(n * p),
              n, p)
dat <- as.data.frame(dat)

out <- miss_pattern(dat)
expect_equal(nrow(out), 1)
expect_equal(colnames(out),
             colnames(dat))
expect_equal(unique(attr(out, "nvalid")),
             n)

})