tests/testthat/test-calculate_aic.R

In reconhub/trendeval: Evaluate Trending Models

test_that("calculate_aic works as expected", {

  # setup
  x <- rnorm(100, mean = 0)
  y <- rpois(n = 100, lambda = exp(x + 1))
  dat <- data.frame(x = x, y = y)
  traditional <- glm(y ~ x, family = poisson)
  traditional2 <- lm(y ~ x)
  model <- glm_model(y ~ x, poisson)
  models <- list(
    poisson_model = glm_model(y ~ x, poisson),
    linear_model = lm_model(y ~ x)
  )
  fitted_model <- trending::fit(model, dat)
  fitted_model_tbl <- trending::fit(model, dat, as_tibble = TRUE)
  fitted_models <- trending::fit(models, dat)
  expected_AIC <- AIC(traditional)
  expected_AIC2 <- AIC(traditional2)

  # single model tests
  expect_equal(calculate_aic(traditional), expected_AIC)

  calc_model <- calculate_aic(model, dat)
  expect_equal(calc_model$result, expected_AIC)
  expect_equal(calc_model$metric, "aic")

  calc_model_tbl <- calculate_aic(model, dat, as_tibble = TRUE)
  expect_equal(calc_model_tbl$result, expected_AIC)
  expect_equal(calc_model_tbl$metric, "aic")

  calc_fitted_model <- calculate_aic(fitted_model)
  expect_equal(calc_fitted_model$result, expected_AIC)
  expect_equal(calc_fitted_model$metric, "aic")

  calc_fitted_model_tbl <- calculate_aic(fitted_model, as_tibble = TRUE)
  expect_equal(calc_fitted_model_tbl$result, expected_AIC)
  expect_equal(calc_fitted_model_tbl$metric, "aic")

  calc_fitted_model <- calculate_aic(fitted_model_tbl)
  expect_equal(calc_fitted_model$result, expected_AIC)
  expect_equal(calc_fitted_model$metric, "aic")

  calc_fitted_model_tbl <- calculate_aic(fitted_model_tbl, as_tibble = TRUE)
  expect_equal(calc_fitted_model_tbl$result, expected_AIC)
  expect_equal(calc_fitted_model_tbl$metric, "aic")

  # multiple model tests
  calc_fitted_models <- calculate_aic(models, dat)
  expect_equal(calc_fitted_models$result, c(expected_AIC, expected_AIC2))
  expect_equal(calc_fitted_models$metric, c("aic", "aic"))
  expect_equal(calc_fitted_models$model_name, c("poisson_model", "linear_model"))

  # should error
  expect_error(calculate_aic(list("bob")))

})