tests/testthat/test_s3.R

In bryandmartin/corncob: Count Regression for Correlated Observations with the Beta-Binomial

library(corncob)
context("Test S3")

set.seed(1)
seq_depth <- rpois(20, lambda = 10000)
my_counts <- rbinom(20, size = seq_depth, prob = 0.001) * 10
my_covariate <- cbind(rep(c(0,1), each = 10))
colnames(my_covariate) <- c("X1")

test_data <- data.frame("W" = my_counts, "M" = seq_depth, my_covariate)

out <- bbdml(formula = cbind(W, M - W) ~ X1,
        phi.formula = ~ X1,
        data = test_data,
        link = "logit",
        phi.link = "logit",
        nstart = 1)

out_fish <- bbdml(formula = cbind(W, M - W) ~ X1,
             phi.formula = ~ X1,
             data = test_data,
             link = "logit",
             phi.link = "fishZ",
             nstart = 1)

test_that("bbdml is S3", {
  expect_is(out, "bbdml")
  expect_is(out_fish, "bbdml")
})

test_that("bbdml S3 plotting works", {
  expect_is(plot(out, B = 10, facet = "X1"), "ggplot")
  expect_is(plot(out, B = 0), "ggplot")
  expect_is(plot(out, B = 10, color = "X1", shape = "X1", facet = "X1", total = TRUE), "ggplot")
  expect_is(plot(out, B = 10, color = as.character(1:20), shape = 1:20, sample_names = FALSE), "ggplot")
  expect_error(plot(out, B = 10, color = c(1,2)))
  expect_error(plot(out, B = 10, shape = c(1,2)))
})

test_that("bbdml print generic works", {
  expect_output(expect_null(print(out)))
})

test_that("summary function works", {
  expect_is(summary(out), "summary.bbdml")
})

test_that("summary printing works", {
  expect_output(expect_null(print(summary(out))))
})