tests/testthat/test-getyhat.R

In midr: Learning from Black-Box Models by Maximum Interpretation Decomposition

test_that("get.yhat() returns numeric vector of length n", {
  # test 1 : lm - regression
  model <- lm(Ozone ~ ., airquality)
  preds <- get.yhat(model, airquality)
  expect_vector(preds, ptype = double(), size = nrow(airquality))
  # test 2 : glm - regression
  model <- glm(Ozone ~ ., Gamma("log"), airquality)
  preds <- get.yhat(model, airquality)
  expect_vector(preds, ptype = double(), size = nrow(airquality))
})