brm_plot_compare()

openpharma · Jun 8, 2023 · 4cd50d0 · 4cd50d0
1 parent 59ef917
commit 4cd50d0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 34 deletions.
diff --git a/R/brm_plot_compare.R b/R/brm_plot_compare.R
@@ -11,15 +11,18 @@
 #'   response variable.
 #' @examples
 #' set.seed(0L)
-#' sim <- brm_simulate()
-#' data <- sim$data
+#' data <- brm_data(
+#'   data = tibble::as_tibble(brm_simulate()$data),
+#'   outcome = "response",
+#'   role = "response",
+#'   group = "group",
+#'   time = "time",
+#'   patient = "patient"
+#' )
 #' data$group <- paste("treatment", data$group)
 #' data$time <- paste("visit", data$time)
 #' formula <- brm_formula(
-#'   response = "response",
-#'   group = "group",
-#'   time = "time",
-#'   patient = "patient",
+#'   data = data,
 #'   effect_base = FALSE,
 #'   interaction_base = FALSE
 #' )
@@ -38,20 +41,12 @@
 #' )
 #' draws <- brm_marginal_draws(
 #'   model = model,
-#'   group = "group",
-#'   time = "time",
-#'   patient = "patient",
+#'   data = data,
 #'   control = "treatment 1",
-#'   baseline = "visit 1",
-#'   outcome = "response"
+#'   baseline = "visit 1"
 #' )
 #' suppressWarnings(summaries_draws <- brm_marginal_summaries(draws))
-#' summaries_data <- brm_marginal_data(
-#'   data,
-#'   response = "response",
-#'   group = "group",
-#'   time = "time"
-#' )
+#' summaries_data <- brm_marginal_data(data)
 #' brm_plot_compare(
 #'   model1 = summaries_draws,
 #'   model2 = summaries_draws,

diff --git a/tests/testthat/test-brm_plot_compare.R b/tests/testthat/test-brm_plot_compare.R
@@ -1,14 +1,17 @@
 test_that("brm_plot_compare()", {
   set.seed(0L)
-  sim <- brm_simulate()
-  data <- sim$data
+  data <- brm_data(
+    data = tibble::as_tibble(brm_simulate()$data),
+    outcome = "response",
+    role = "response",
+    group = "group",
+    time = "time",
+    patient = "patient"
+  )
   data$group <- paste("treatment", data$group)
   data$time <- paste("visit", data$time)
   formula <- brm_formula(
-    response = "response",
-    group = "group",
-    time = "time",
-    patient = "patient",
+    data = data,
     effect_base = FALSE,
     interaction_base = FALSE
   )
@@ -27,20 +30,12 @@ test_that("brm_plot_compare()", {
   )
   draws <- brm_marginal_draws(
     model = model,
-    group = "group",
-    time = "time",
-    patient = "patient",
+    data = data,
     control = "treatment 1",
-    baseline = "visit 1",
-    outcome = "response"
+    baseline = "visit 1"
   )
   suppressWarnings(summaries_draws <- brm_marginal_summaries(draws))
-  summaries_data <- brm_marginal_data(
-    data,
-    response = "response",
-    group = "group",
-    time = "time"
-  )
+  summaries_data <- brm_marginal_data(data)
   out <- brm_plot_compare(
     summaries_draws = summaries_draws,
     summaries_data = summaries_data