eightedMeanLearner: 100% test coverage

.covrignore

@@ -15,4 +15,3 @@
 ./R/TrainMetaLayer.R
 ./R/TrainStudy.R
 ./R/VarSel.R
-./R/weightedMeanLearner.R

R/weightedMeanLearner.R

@@ -14,9 +14,9 @@
 #' @export
 #'
 #' @examples
-#' set.seed(20240624)
-#' x = data.frame(x1 = rnorm(50))
-#' y = sample(x = 0:1, size = 50, replace = TRUE)
+#' 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)
 #'
 weightedMeanLearner = function (x, y) {

tests/testthat/test-weightedMeanLearner.R

@@ -1,3 +1,26 @@
-test_that("multiplication works", {
-  expect_equal(2 * 2, 4)
+test_that("weightedMeanLearner works", {
+  expect_no_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)
+  })
+  expect_error({
+    set.seed(20240624L)
+    x = data.frame(x1 = rnorm(50L))
+    y = sample(x = 0L:2L, size = 50L, replace = TRUE)
+    my_model = weightedMeanLearner(x = x, y = y)
+  })
+  expect_no_error({
+    set.seed(20240624L)
+    x = data.frame(x1 = rnorm(50L))
+    y = sample(x = c("control", "case"), size = 50L, replace = TRUE)
+    my_model = weightedMeanLearner(x = x, y = y)
+  })
+  expect_no_error({
+    set.seed(20240624L)
+    x = data.frame(x1 = rnorm(50L))
+    y = sample(x = c("0", "1"), size = 50L, replace = TRUE)
+    my_model = weightedMeanLearner(x = x, y = factor(y))
+  })
 })