test_that("klaR::rda model object", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  # formula method
  expect_no_error(f_fit <- fit(rda_spec, Type ~ ., data = glass_tr))
  expect_equal(f_fit$fit$covpooled, exp_f_fit_rda$covpooled)
  expect_equal(f_fit$fit$means, exp_f_fit_rda$means)

  # x/y method
  expect_no_error(
    xy_fit <- fit_xy(rda_spec, x = glass_tr[, -10], y = glass_tr$Type)
  )
  expect_equal(xy_fit$fit$covpooled, exp_f_fit_rda$covpooled)
  expect_equal(xy_fit$fit$means, exp_f_fit_rda$means)
})

# ------------------------------------------------------------------------------

test_that("klaR::rda class predictions", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  # formula method
  expect_no_error(f_fit <- fit(rda_spec, Type ~ ., data = glass_tr))
  f_pred <- predict(f_fit, glass_te)
  exp_f_pred <- predict(exp_f_fit_rda, glass_te)$class

  expect_s3_class(f_pred, "tbl_df")
  expect_true(all(names(f_pred) == ".pred_class"))
  expect_equal(f_pred$.pred_class, exp_f_pred)

  # x/y method
  expect_no_error(
    xy_fit <- fit_xy(rda_spec, x = glass_tr[, -10], y = glass_tr$Type)
  )
  xy_pred <- predict(xy_fit, glass_te)
  # See bug note above
  # exp_xy_pred <- predict(exp_xy_fit, glass_te)

  expect_s3_class(xy_pred, "tbl_df")
  expect_true(all(names(xy_pred) == ".pred_class"))
  expect_equal(xy_pred$.pred_class, exp_f_pred)
})

# ------------------------------------------------------------------------------

test_that("klaR::rda prob predictions", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  # formula method
  expect_no_error(f_fit <- fit(rda_spec, Type ~ ., data = glass_tr))
  f_pred <- predict(f_fit, glass_te, type = "prob")
  exp_f_pred <- probs_to_tibble(
    predict(exp_f_fit_rda, glass_te, type = "posterior")$posterior
  )

  expect_s3_class(f_pred, "tbl_df")
  expect_equal(names(f_pred), glass_prob_names)
  expect_equal(f_pred, exp_f_pred)

  # x/y method
  expect_no_error(
    xy_fit <- fit_xy(rda_spec, x = glass_tr[, -10], y = glass_tr$Type)
  )
  xy_pred <- predict(xy_fit, glass_te, type = "prob")
  expect_s3_class(xy_pred, "tbl_df")
  expect_equal(names(xy_pred), glass_prob_names)
  expect_equal(xy_pred, exp_f_pred)
})

# ------------------------------------------------------------------------------

test_that("klaR::rda missing data", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  expect_no_error(f_fit <- fit(rda_spec, Type ~ ., data = glass_tr))
  f_pred <- predict(f_fit, glass_na, type = "prob")

  exp_f_pred <- probs_to_tibble(
    predict(exp_f_fit_rda, glass_na, type = "posterior")$posterior
  )

  expect_s3_class(f_pred, "tbl_df")
  expect_true(nrow(f_pred) == nrow(glass_te))
  expect_equal(names(f_pred), glass_prob_names)
  expect_equal(f_pred, exp_f_pred)
})

# ------------------------------------------------------------------------------

test_that("klaR::rda printing", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  expect_snapshot(print(rda_spec))
})

# ------------------------------------------------------------------------------

test_that("klaR::rda updating", {
  skip_if_not_installed("klaR")
  skip_if_not_installed("mlbench")
  # exp_* objects in helper-object.R

  rda_spec_2 <-
    discrim_regularized(frac_common_cov = 1, frac_identity = 1) |>
    set_engine("klaR")
  rda_spec_3 <- update(rda_spec, frac_common_cov = 1, frac_identity = 1)
  expect_equal(rda_spec_2, rda_spec_3)

  prior_spec_2 <- discrim_regularized(frac_common_cov = 1) |>
    set_engine("klaR", prior = rep(1 / 6, 6))
  prior_spec_3 <- update(prior_spec_rda, frac_common_cov = 1)
  expect_equal(
    prior_spec_2,
    prior_spec_3,
    ignore_function_env = TRUE,
    ignore_formula_env = TRUE
  )
})