context("subgroup effect calculations")
test_that("test that subgroup effect calculations are correct", {
set.seed(123)
bene.score <- rnorm(10)
y <- rnorm(10)
trt <- c(rep(1, 5), rep(0, 5))
pi.x <- c(rep(0.75, 5), rep(0.25, 5))
sub.eff <- subgroup.effects(benefit.scores = bene.score,
y = y, trt = trt, pi.x = pi.x)
recom <- 1 * (bene.score > 0)
mean11 <- mean(y[trt == 1 & recom == 1])
mean01 <- mean(y[trt == 0 & recom == 1])
mean10 <- mean(y[trt == 1 & recom == 0])
mean00 <- mean(y[trt == 0 & recom == 0])
overall <- mean(y[(trt == 1 & recom == 1) |
(trt == 0 & recom == 0) ]) -
mean(y[(trt == 0 & recom == 1) |
(trt == 1 & recom == 0) ])
expect_equal(overall, sub.eff$overall.subgroup.effect)
expect_error(subgroup.effects(benefit.scores = bene.score,
y = y, trt = c(rep(1, 4), rep(0, 3), rep(2, 3)),
pi.x = c(rep(0.33, 4), rep(0.25, 3), rep(0.75, 3))
))
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.