tests/testthat/test-performance.R

In tune: Tidy Tuning Tools

test_that("prediction types", {
  expect_equal(tune:::pred_type(accuracy), "class")
  expect_equal(tune:::pred_type(roc_auc), "prob")
  expect_equal(tune:::pred_type(rsq), "numeric")
  expect_equal(
    tune:::pred_type(yardstick::metric_set(yardstick::accuracy, yardstick::roc_auc)),
    "unknown"
  )
})

test_that("metric information", {
  set_1 <- tune:::metrics_info(
    yardstick::metric_set(yardstick::accuracy, yardstick::kap, yardstick::roc_auc)
  )
  set_2 <- tune:::metrics_info(
    yardstick::metric_set(yardstick::rmse, yardstick::rsq)
  )
  expect_equal(
    set_1,
    tibble::tibble(
      .metric = c("accuracy", "kap", "roc_auc"),
      direction = rep("maximize", 3),
      type = c("class", "class", "prob")
    )
  )
  expect_equal(
    set_2,
    tibble::tibble(
      .metric = c("rmse", "rsq"),
      direction = c("minimize", "maximize"),
      type = c("numeric", "numeric")
    )
  )
})