fix and update tests

mlr-org · Oct 19, 2023 · 81b587f · 81b587f
1 parent 304273f
commit 81b587f
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 4 deletions.
diff --git a/tests/testthat/test_glmnet_surv_cv_glmnet.R b/tests/testthat/test_glmnet_surv_cv_glmnet.R
@@ -12,7 +12,7 @@ test_that("autotest", {
 
 test_that("selected_features", {
   task = tsk("gbcs")
-  learner = lrn("surv.glmnet")
+  learner = lrn("surv.cv_glmnet")
   learner$train(task)
 
   expect_equal(

diff --git a/tests/testthat/test_glmnet_surv_glmnet.R b/tests/testthat/test_glmnet_surv_glmnet.R
@@ -12,7 +12,7 @@ test_that("autotest", {
 
 test_that("selected_features", {
   task = tsk("gbcs")
-  learner = lrn("surv.cv_glmnet")
+  learner = lrn("surv.glmnet")
   learner$train(task)
 
   expect_equal(

diff --git a/tests/testthat/test_paramtest_glmnet_surv_cv_glmnet.R b/tests/testthat/test_paramtest_glmnet_surv_cv_glmnet.R
@@ -20,7 +20,11 @@ test_that("predict surv.cv_glmnet", {
   exclude = c(
     "object", # handled via mlr3
     "newx", # handled via mlr3
-    "predict.gamma" # renamed from gamma
+    "predict.gamma", # renamed from gamma
+    "offset", # for distr prediction
+    "newoffset", # for distr prediction
+    "stype", # for distr prediction
+    "ctype" # for distr prediction
   )
 
   paramtest = run_paramtest(learner, fun, exclude, tag = "predict")

diff --git a/tests/testthat/test_paramtest_glmnet_surv_glmnet.R b/tests/testthat/test_paramtest_glmnet_surv_glmnet.R
@@ -26,7 +26,10 @@ test_that("predict surv.glmnet", {
     "object", # handled via mlr3
     "newx", # handled via mlr3
     "type", # handled via mlr3
-    "predict.gamma" # renamed from gamma
+    "predict.gamma", # renamed from gamma
+    "offset", # for distr prediction
+    "stype", # for distr prediction
+    "ctype" # for distr prediction
   )
 
   paramtest = run_paramtest(learner, fun, exclude, tag = "predict")