tests/testthat/helper_script.R

In realgabon/insuRglm: Tools for GLM Modeling in Insurance Context

require("insuRglm")
require("dplyr")

data("freq_train")
data("sev_train")
data("bc_train")

data("freq_test")
data("sev_test")
data("bc_test")

no_print <- function(expr) {
  invisible(capture.output(expr))
}

freq_args <- list(
  data_train = freq_train,
  data_test = freq_test,
  target = "numclaims",
  weight = "exposure",
  family = "poisson",
  keep_cols = c("pol_nbr", "premium"),
  folder = tempdir()
)

sev_args <- list(
  data_train = sev_train,
  data_test = sev_test,
  target = "sev",
  weight = "numclaims",
  family = "gamma",
  keep_cols = c("pol_nbr", "exposure", "premium"),
  folder = tempdir()
)

bc_args <- list(
  data_train = bc_train,
  data_test = bc_test,
  target = "bc",
  weight = "exposure",
  family = "tweedie",
  tweedie_p = 1.5,
  keep_cols = c("pol_nbr", "exposure", "premium"),
  folder = tempdir()
)

no_print(freq_setup <- suppressMessages({do.call(setup, freq_args)}))
no_print(sev_setup <- do.call(setup, sev_args))
no_print(bc_setup <- do.call(setup, bc_args))