tests/testthat/test-pois-glm.R

In aterui/avpc: Average Predictive Comparison

# setup -------------------------------------------------------------------

  context("test poisson glm coefficient equivalency")

  pacman::p_load(tidyverse)
  ilogit <- function(x) 1 / (1 + exp(-x))


# test dataset ------------------------------------------------------------

  n_sample <- 100

  x1 <- rnorm(n = n_sample)
  x2 <- rnorm(n = n_sample)
  x3 <- sample(letters[1:5], size = n_sample, replace = TRUE)
  mat <- stats::model.matrix(model.frame(~ x1 + x2 + x3))
  v_b <- runif(n = ncol(mat), -1, 1)
  y <- rpois(n = n_sample, lambda = exp(mat %*% v_b))


# run model ---------------------------------------------------------------

  m <- glm(y ~ x1 + x2 + x3, family = poisson)
  beta <- coef(m)
  names(beta) <- NULL


# test --------------------------------------------------------------------

  test_that("compare coefficients", {
    expect_equal(apcomp(m, u = "x1", y_scale = "identity")$estimate,
                 beta[2])
    expect_equal(apcomp(m, u = "x2", y_scale = "identity")$estimate,
                 beta[3])
    expect_equal(apcomp(m, u = "x3b", y_scale = "identity")$estimate,
                 beta[4])
  })