tests/testthat/test_train_model4.R

In markgene/yamatClassifier: Yet Another Methylation Array Toolkit: Classifier

context("Test train_model4()")
library(yamatClassifier)
library(glmnet)
library(randomForest)

test_that("train_model4() end-to-end with mtry not set", {
  require(datasets)
  utils::data(iris)
  tune_result <- yamatClassifier::train_model4(
    dat = iris,
    response_name = "Species",
    outer_cv_folds = 3,
    inner_cv_folds = 3,
    random_state = 1,
    mtry = NULL,
    save_level = 2,
    save_prefix = "train_model4",
    overwrite = FALSE,
    output = "output/train_model4",
    verbose = TRUE
  )
  expect_true(is.list(tune_result))
  expect_true(length(tune_result) == 1)
})


test_that("train_model4() end-to-end with mtry set to 2", {
  require(datasets)
  utils::data(iris)
  tune_result <- yamatClassifier::train_model4(
    dat = iris,
    response_name = "Species",
    outer_cv_folds = 3,
    inner_cv_folds = 3,
    random_state = 1,
    mtry = 2,
    save_level = 2,
    save_prefix = "train_model4",
    overwrite = FALSE,
    output = "output/train_model4",
    verbose = TRUE
  )
  expect_true(is.list(tune_result))
  expect_true(length(tune_result) == 1)
})