tests/testthat/test-predictions.R

In jolars/golem: Generalized Linear Models Regularized with the Sorted L1-Norm

test_that("predictions work for all models", {
  set.seed(1)

  for (family in c("gaussian", "binomial", "poisson", "multinomial")) {
    xy <- owl:::randomProblem(100, 10, response = family)
    x <- xy$x
    y <- xy$y

    fit <- owl(x, y, family = family, n_sigma = 5)

    for (type in c("link", "response", "class")) {
      if (type == "class" && family %in% c("gaussian", "poisson"))
        next

      expect_silent(predict(fit, x, type = type))
    }
  }
})