tests/testthat/test-best_models.R

In badp: Bayesian Averaging for Dynamic Panels

test_that(paste("best_models creates correct lists with graphs"), {

  data_prepared <- badp::economic_growth[,1:6] %>%
    badp::feature_standardization(
      excluded_cols = c(country, year, gdp)
    ) %>%
    badp::feature_standardization(
      group_by_col  = year,
      excluded_cols = country,
      scale         = FALSE
    )

  bma_results <- bma(small_model_space, round= 3, dilution = 0)

  best <- 5

  best_5_models <- best_models(bma_results, prior = "beta", best = best, estimate = TRUE, robust = TRUE)

  expect_equal(class(best_5_models), c("badp_best_models", "list"))
  expect_s3_class(best_5_models, "badp_best_models")
  expect_equal(length(best_5_models), 9)
  expect_true(isTRUE(attr(best_5_models, "estimate")))
  expect_true(isTRUE(attr(best_5_models, "robust")))
  expect_equal(class(best_5_models[[1]]), c("matrix", "array"))
  expect_equal(class(best_5_models[[2]]), c("matrix", "array"))
  expect_equal(class(best_5_models[[3]]), c("matrix", "array"))
  expect_equal(ncol(best_5_models[[1]]), best)
  expect_equal(ncol(best_5_models[[2]]), best)
  expect_equal(ncol(best_5_models[[3]]), best)
  expect_equal(class(best_5_models[[4]]), "knitr_kable")
  expect_equal(class(best_5_models[[5]]), "knitr_kable")
  expect_equal(class(best_5_models[[6]]), "knitr_kable")
  expect_equal(class(best_5_models[[7]]),
               c("badp_drawable_grob", "gTree", "grob", "gDesc"))
  expect_equal(class(best_5_models[[8]]),
               c("badp_drawable_grob", "gTree", "grob", "gDesc"))
  expect_equal(class(best_5_models[[9]]),
               c("badp_drawable_grob", "gTree", "grob", "gDesc"))
  # The drawable wrapper must still inherit from gTree so that existing
  # code calling `grid::grid.draw(best[[9]])` keeps working.
  expect_s3_class(best_5_models[[7]], "gTree")
  expect_s3_class(best_5_models[[8]], "gTree")
  expect_s3_class(best_5_models[[9]], "gTree")
})