add more tests

tests/testthat/test-estimators.R

@@ -22,3 +22,17 @@ test_that("Estimation with glmnet works in a simple case", {
     expect_equal(colnames(predictions$mediators), mediators(model))
     expect_equal(mediators(model), paste0("ASV", 1:5))
 })
+
+test_that("Can compile and estimate a BRMS model.", {
+    model <- multimedia(exper, brms_model())
+    expect_equal(outcome_model(model)@model_type, "brms_model()")
+    model <- estimate(model, exper)
+    expect_s4_class(model, "multimedia")
+    y_hat <- predict(model)
+    expect_named(y_hat, c("mediators", "outcomes"))
+    expect_named(y_hat$mediators, paste0("ASV", 1:5))
+
+    samples <- multimedia:::brms_sampler(model@outcome@estimates)
+    expect_named(samples, "PHQ")
+    expect_equal(dim(samples), c(100, 1))
+})

tests/testthat/test-treatment_profile.R

@@ -1,40 +1,56 @@
-
 exper <- demo_joy() |>
     mediation_data("PHQ", "treatment", starts_with("ASV"))
 fit <- multimedia(exper) |>
     estimate(exper)
 
 test_that("Can create a treatment profile from a data.frame.", {
-  t1 <- data.frame(treatment = factor(rep(c(0, 1), each = 5)))
-  profile <- setup_profile(fit, t_mediator = t1, t_outcome = t1)
-  expect_s4_class(profile, "treatment_profile")
-  expect_named(profile@t_mediator, paste0("ASV", 1:5))
-  expect_named(profile@t_outcome, "PHQ")
-  expect_equal(
-    unique(profile@t_mediator$ASV1$treatment),
-    as.factor(c("0", "1"))
-  )
-  expect_s3_class(profile@t_mediator$ASV1$treatment, "factor")
+    t1 <- data.frame(treatment = factor(rep(c(0, 1), each = 5)))
+    profile <- setup_profile(fit, t_mediator = t1, t_outcome = t1)
+    expect_s4_class(profile, "treatment_profile")
+    expect_named(profile@t_mediator, paste0("ASV", 1:5))
+    expect_named(profile@t_outcome, "PHQ")
+    expect_equal(
+        unique(profile@t_mediator$ASV1$treatment),
+        as.factor(c("0", "1"))
+    )
+    expect_s3_class(profile@t_mediator$ASV1$treatment, "factor")
 })
 
 test_that("Can create treatment profile from a vector.", {
-  t1 <- rep(0:1, 5)
-  profile <- setup_profile(fit, t_mediator = t1, t_outcome = t1)
-  expect_s4_class(profile, "treatment_profile")
-  expect_named(profile@t_mediator, paste0("ASV", 1:5))
-  expect_named(profile@t_outcome, "PHQ")
-  expect_equal(
-    unique(profile@t_mediator$ASV1$treatment),
-    c(0, 1)
-  )
-  expect_type(profile@t_mediator$ASV1$treatment, "integer")
-  expect_equal(profile@t_mediator$ASV1$treatment, t1)
+    t1 <- rep(0:1, 5)
+    profile <- setup_profile(fit, t_mediator = t1, t_outcome = t1)
+    expect_s4_class(profile, "treatment_profile")
+    expect_named(profile@t_mediator, paste0("ASV", 1:5))
+    expect_named(profile@t_outcome, "PHQ")
+    expect_equal(
+        unique(profile@t_mediator$ASV1$treatment),
+        c(0, 1)
+    )
+    expect_type(profile@t_mediator$ASV1$treatment, "integer")
+    expect_equal(profile@t_mediator$ASV1$treatment, t1)
 })
 
 test_that("Can create treatment profile from scratch.", {
-  t1 <- list(m = data.frame(treatment = rep(0:1, 5)))
-  profile <- new("treatment_profile", t_mediator = t1, t_outcome = t1)
-  expect_s4_class(profile, "treatment_profile")
-  t2 <- list(t = data.frame(treatment = rep(0:1, 2)))
-  expect_error(new("treatment_profile", t_mediator = t1, t_outcome = t2))
-})
+    t1 <- list(m = data.frame(treatment = rep(0:1, 5)))
+    profile <- new("treatment_profile", t_mediator = t1, t_outcome = t1)
+    expect_s4_class(profile, "treatment_profile")
+    t2 <- list(t = data.frame(treatment = rep(0:1, 2)))
+    expect_error(new("treatment_profile", t_mediator = t1, t_outcome = t2))
+})
+
+test_that("check_profile raises errors in invalid inputs.", {
+    setClass("TestProfile", slots = c(t_mediator = "list", t_outcome = "list"))
+    test_profile <- new(
+        "TestProfile",
+        t_mediator = list(),
+        t_outcome = list(data.frame(A = 1:10), data.frame(A = 1:2))
+    )
+    expect_error(multimedia:::check_profile(test_profile))
+
+    test_profile <- new(
+        "TestProfile",
+        t_mediator = list(data.frame(A = 1:10), data.frame(A = 1:10)),
+        t_outcome = list(data.frame(A = 1:10), data.frame(A = 1:2))
+    )
+    expect_error(multimedia:::check_profile(test_profile))
+})