Skip to content

Commit 9d1de71

Browse files
committed
tests: surrogate default values
1 parent c042e05 commit 9d1de71

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/testthat/test_mbo_defaults.R

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ test_that("default_surrogate", {
6363
surrogate = default_surrogate(MAKE_INST(OBJ_1D_MIXED, search_space = PS_1D_MIXED))
6464
expect_r6(surrogate, "SurrogateLearner")
6565
expect_r6(surrogate$learner, "LearnerRegrRanger")
66-
expect_equal_sorted(surrogate$learner$param_set$values,
66+
expect_equal_sorted(surrogate$learner$param_set$values[c("num.threads", "num.trees", "keep.inbag", "se.method")],
6767
list(num.threads = 1L, num.trees = 100L, keep.inbag = TRUE, se.method = "jack"))
6868
expect_equal(surrogate$learner$encapsulation, c(train = "evaluate", predict = "evaluate"))
6969
expect_r6(surrogate$learner$fallback, "LearnerRegrRanger")
@@ -72,7 +72,7 @@ test_that("default_surrogate", {
7272
surrogate = default_surrogate(MAKE_INST(OBJ_1D_2_MIXED, search_space = PS_1D_MIXED))
7373
expect_r6(surrogate, "SurrogateLearnerCollection")
7474
expect_list(surrogate$learner, types = "LearnerRegrRanger")
75-
expect_equal_sorted(surrogate$learner[[1L]]$param_set$values,
75+
expect_equal_sorted(surrogate$learner[[1L]]$param_set$values[c("num.threads", "num.trees", "keep.inbag", "se.method")],
7676
list(num.threads = 1L, num.trees = 100L, keep.inbag = TRUE, se.method = "jack"))
7777
expect_equal(surrogate$learner[[1L]]$encapsulation, c(train = "evaluate", predict = "evaluate"))
7878
expect_r6(surrogate$learner[[1L]]$fallback, "LearnerRegrRanger")

0 commit comments

Comments
 (0)