small updates

bluefoxr · May 21, 2024 · 338b21c · 338b21c
1 parent 6e409ff
commit 338b21c
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 3 deletions.
diff --git a/man/Treat.purse.Rd b/man/Treat.purse.Rd
diff --git a/man/n_scaled.Rd b/man/n_scaled.Rd
diff --git a/tests/testthat/test-treat.R b/tests/testthat/test-treat.R
@@ -78,3 +78,16 @@ test_that("treat_coin", {
   expect_identical(dset_t, dset_t2$x_treat)
 
 })
+
+test_that("disable_treat", {
+
+  purse <- build_example_purse(up_to = "Screen", quietly = T)
+
+  purse <- Treat(purse, dset = "Screened", disable = T)
+
+  d1 <- get_dset(purse, dset = "Treated")
+  d2 <- get_dset(purse, dset = "Screened")
+
+  expect_equal(d1, d2)
+
+})