predict.weightedMeanLearner: 100% test coverage

imbs-hl · Jul 19, 2024 · 051f3d6 · 051f3d6
1 parent d287845
commit 051f3d6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.covrignore b/.covrignore
@@ -5,7 +5,6 @@
 ./R/NewLayer.R
 ./R/NewMetaLayer.R
 ./R/NewStudy.R
-./R/predict.weightedMeanLearner.R
 ./R/PredictData.R
 ./R/PredictLayer.R
 ./R/PredictMetaLayer.R

diff --git a/tests/testthat/test-weightedMeanLearner.R b/tests/testthat/test-weightedMeanLearner.R
@@ -4,6 +4,16 @@ test_that("weightedMeanLearner works", {
     x = data.frame(x1 = rnorm(50L))
     y = sample(x = 0L:1L, size = 50L, replace = TRUE)
     my_model = weightedMeanLearner(x = x, y = y)
+    x_new <- data.frame(x1 = rnorm(10))
+    my_predictions <- predict(object = my_model, data = x_new)
+  })
+  expect_error({
+    set.seed(20240624L)
+    x = data.frame(x1 = rnorm(50L))
+    y = sample(x = 0L:1L, size = 50L, replace = TRUE)
+    my_model = weightedMeanLearner(x = x, y = y)
+    x_new <- data.frame(x2 = rnorm(10))
+    my_predictions <- predict(object = my_model, data = x_new)
   })
   expect_error({
     set.seed(20240624L)