fix boost_spline tests

schalkdaniel · Jul 29, 2018 · cae39c5 · cae39c5
1 parent 65675bb
commit cae39c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/testthat/test_boost_spline.R b/tests/testthat/test_boost_spline.R
@@ -7,13 +7,13 @@ test_that("boostSpline function works", {
 
 	expect_length(mod$getBaselearnerNames(), 6)
   expect_length(mod$selected(), 100)
-  expect_length(mod$risk(), 101)
+  expect_length(mod$getInbagRisk(), 101)
   expect_length(mod$predict(), nrow(iris))
 
   expect_output(mod$train(150))
 
   expect_length(mod$selected(), 150)
-  expect_length(mod$risk(), 151)
+  expect_length(mod$getInbagRisk(), 151)
   expect_length(mod$predict(), nrow(iris))
   expect_equal(mod$predict(), mod$predict(iris))
 })