tests/testthat/test_feature_selection_ranger.R

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

test_that("feature_selection_ranger()", {
  require(mlbench)
  utils::data(Sonar)
  set.seed(42)
  selected_features <- yamatClassifier::select_features_ranger(dat = Sonar,
                                                               response_name = "Class",
                                                               top_n = 10)
  expect_true(is.vector(selected_features))
  expect_true(length(selected_features) == 10)
})
markgene/yamatClassifier documentation built on Oct. 14, 2024, 2:36 a.m.