Merge pull request #137 from n8thangreen/dev

update to latest across main and dev branches
n8thangreen · Aug 9, 2023 · 84657f9 · 84657f9
2 parents 8757a60 + cf868d2
commit 84657f9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/R/CreateInputs.R b/R/CreateInputs.R
@@ -14,6 +14,12 @@
 createInputs.default <- function(inputs,
                                  print_is_linear_comb = TRUE) {
 
+  # remove NA columns
+  if (sum(is.na(inputs)) > 0) {
+    inputs <- inputs[ , colSums(is.na(inputs)) == 0]
+    message("Dropped any columns containing NAs")
+  }
+
   if (!is.logical(print_is_linear_comb))
     stop("print_is_linear_comb must be logical.", call. = FALSE)
 

diff --git a/tests/testthat/test-createInputs.R b/tests/testthat/test-createInputs.R
@@ -1,9 +1,16 @@
 
 
-##TODO:
-
-
-
-
-
+test_that("createInputs bad arguments", {
+
+  data(Vaccine, package = "BCEA")
+  treats <- c("Status quo", "Vaccination")
+
+  m <- bcea(e.pts, c.pts, ref = 2, interventions = treats)
+
+  dat <- vaccine_mat[, 1:5]
+  datNA <-  cbind(dat, NA)
+
+  expect_equal(createInputs(dat), createInputs(datNA))
+  expect_message(createInputs(datNA), regexp = "Dropped any columns containing NAs")
+})