tests/testthat/test-makeModel.R

In dymium-org/dymiumCore: A Toolkit for Building a Dynamic Microsimulation Model for Integrated Urban Modelling

test_that("makeModel", {

  # binary choice model
  bc_model <- create_caret_binary_model()
  Mod <- makeModel(bc_model)
  checkmate::expect_r6(Mod, classes = "ModelBinaryChoice")
  checkmate::expect_numeric(predict(Mod, newdata = toy_individuals), finite = T, any.missing = FALSE)
  checkmate::expect_numeric(summary(Mod), names = "named")

  # mlogit model
  mlogit_model <- create_mlogit_model()
  Mod <- makeModel(mlogit_model)
  checkmate::expect_r6(Mod, classes = "ModelMultinomialLogit")
  # predict(Mod, newdata = Fishing)
  # checkmate::expect_numeric(predict(Mod, newdata = toy_individuals), finite = T, any.missing = FALSE)
  checkmate::expect_numeric(summary(Mod), names = "named")
})