DS-3488 Add more unit tests [revdep skip]

Displayr · Aug 25, 2021 · 2956841 · 2956841
1 parent 19a57c3
commit 2956841
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/testthat/test-checks.R b/tests/testthat/test-checks.R
@@ -163,4 +163,11 @@ test_that("DS-3488 Check dummy variable adjustment handled with and without outl
 
     expect_equal(CheckPredictionVariables(output, newdata = dummy.adj.model),
                  expected.output)
+    # Check method still works when formula not available
+    output[["formula"]] <- NULL
+    output[["outcome.name"]] <- "Y"
+    output[["model"]] <- missing.all.predictors
+    expected.output <- missing.all.predictors[-1, -1]
+    expect_equal(CheckPredictionVariables(output, newdata = dummy.adj.model),
+                 dummy.adj.model[, c("X1", "X2")])
 })