diff --git a/tests/testthat/test_ensemble_fselect.R b/tests/testthat/test_ensemble_fselect.R index bf0d7512..19bd1240 100644 --- a/tests/testthat/test_ensemble_fselect.R +++ b/tests/testthat/test_ensemble_fselect.R @@ -15,8 +15,8 @@ test_that("ensemble feature selection works", { expect_character(efsr$man) expect_data_table(efsr$result, nrows = 4) expect_list(efsr$result$features, any.missing = FALSE, len = 4) - expect_vector(efsr$result$n_features, size = 4) - expect_vector(efsr$result$classif.ce, size = 4) + expect_numeric(efsr$result$n_features, len = 4) + expect_numeric(efsr$result$classif.ce, len = 4) expect_benchmark_result(efsr$benchmark_result) expect_equal(efsr$measure, "classif.ce") expect_equal(efsr$n_learners, 2) @@ -72,8 +72,8 @@ test_that("ensemble feature selection works without benchmark result", { expect_character(efsr$man) expect_data_table(efsr$result, nrows = 4) expect_list(efsr$result$features, any.missing = FALSE, len = 4) - expect_vector(efsr$result$n_features, size = 4) - expect_vector(efsr$result$classif.ce, size = 4) + expect_numeric(efsr$result$n_features, len = 4) + expect_numeric(efsr$result$classif.ce, len = 4) expect_null(efsr$benchmark_result) expect_equal(efsr$measure, "classif.ce") expect_equal(efsr$n_learners, 2) @@ -121,8 +121,8 @@ test_that("ensemble feature selection works with rfe", { expect_character(efsr$man) expect_data_table(efsr$result, nrows = 4) expect_list(efsr$result$features, any.missing = FALSE, len = 4) - expect_vector(efsr$result$n_features, size = 4) - expect_vector(efsr$result$classif.ce, size = 4) + expect_numeric(efsr$result$n_features, len = 4) + expect_numeric(efsr$result$classif.ce, len = 4) expect_list(efsr$result$importance, any.missing = FALSE, len = 4) expect_benchmark_result(efsr$benchmark_result) expect_equal(efsr$measure, "classif.ce")