data("occup", package = "cat2cat")
data("trans", package = "cat2cat")

occup_old <- occup[occup$year == 2008, ]
occup_new <- occup[occup$year == 2010, ]

testthat::test_that("validate_mappings incorrect", {
  testthat::expect_error(
    validate_mappings(list()), 'all\\(c\\("trans", "direction"\\) %in% names\\(mappings\\)\\) is not TRUE'
  )
})

testthat::test_that("validate_mappings correct", {
  mappings_simple_back <- list(trans = trans, direction = "backward")
  mappings_simple_for <- list(trans = trans, direction = "forward")
  testthat::expect_silent(validate_mappings(mappings_simple_back))
  testthat::expect_silent(validate_mappings(mappings_simple_for))
})

testthat::test_that("validate_data incorrect", {
  testthat::expect_error(validate_data(list()), 'inherits\\(data\\$old, "data.frame"\\) is not TRUE')

  data_simple <- list(old = occup_old, new = occup_new, cat_var_old = "code", cat_var_new = "code", time_var = "WRONG")

  testthat::expect_error(
    validate_data(data_simple),
    'all\\(c\\(data\\$cat_var_old, data\\$time_var\\) \\%in\\% colnames\\(data\\$old\\)\\) is not TRUE'
  )

  data_simple <- list(old = occup_old, new = occup_new, cat_var_old = "WRONG", cat_var_new = "code", time_var = "year")

  testthat::expect_error(
    validate_data(data_simple),
    'all\\(c\\(data\\$cat_var_old, data\\$time_var\\) \\%in\\% colnames\\(data\\$old\\)\\) is not TRUE'
  )

  data_simple <- list(old = occup_old, new = occup_new, cat_var_old = "code", cat_var_new = "WRONG", time_var = "year")

  testthat::expect_error(
    validate_data(data_simple),
    'all\\(c\\(data\\$cat_var_new, data\\$time_var\\) \\%in\\% colnames\\(data\\$new\\)\\) is not TRUE'
  )

})

testthat::test_that("validate_data correct", {
  data_simple <- list(
    old = occup_old, new = occup_new, cat_var_old = "code", cat_var_new = "code", time_var = "year"
  )
  testthat::expect_silent(validate_data(data_simple))
})

testthat::test_that("validate_ml incorrect", {
  testthat::expect_error(
    validate_ml(list()), 'all\\(c\\("method", "features", "data"\\) %in% names\\(ml\\)\\) is not TRUE'
  )
})

testthat::test_that("validate_ml correct", {
  ml <- list(
    data = occup_new,
    cat_var = "code",
    method = c("knn", "rf", "lda"),
    features = c("age", "sex", "edu", "exp", "parttime", "salary"),
    args = list(k = 10, ntree = 30)
  )
  testthat::expect_silent(validate_ml(ml))
})