tests/testthat/test-model_performance.lm.R

In performance: Assessment of Regression Models Performance

test_that("model_performance.lm", {
  model <- lm(mpg ~ wt + cyl, data = mtcars)
  expect_equal(model_performance(model)$R2, 0.830, tolerance = 0.01)
  expect_equal(
    colnames(model_performance(model)),
    c("AIC", "AICc", "BIC", "R2", "R2_adjusted", "RMSE", "Sigma")
  )
})

test_that("model_performance.glm", {
  model <- glm(vs ~ wt + mpg, data = mtcars, family = "binomial")
  expect_equal(model_performance(model)$R2_Tjur, 0.478, tolerance = 0.01)
  expect_equal(
    colnames(model_performance(model)),
    c(
      "AIC", "AICc", "BIC", "R2_Tjur", "RMSE", "Sigma", "Log_loss", "Score_log",
      "Score_spherical", "PCP"
    )
  )
})

test_that("model_performance.glm-factor", {
  model <- glm(factor(vs, labels = c("automatic", "manual")) ~ wt + mpg, data = mtcars, family = "binomial")
  expect_equal(model_performance(model)$R2_Tjur, 0.478, tolerance = 0.01)
  expect_equal(
    colnames(model_performance(model)),
    c(
      "AIC", "AICc", "BIC", "R2_Tjur", "RMSE", "Sigma", "Log_loss", "Score_log",
      "Score_spherical", "PCP"
    )
  )
})