Test for effect estimation

tests/testthat/test-effect-estimation.R

@@ -0,0 +1,18 @@
+
+# Fit random forest on toy data
+xy_data <- demo_spline()
+exper <- mediation_data(xy_data, starts_with("outcome"), "treatment", "mediator")
+fit <- multimedia(exper, outcome_estimator = rf_model(num.trees = 1e3)) |>
+  estimate(exper)
+
+test_that("Can estimate direct effects on toy dataset.", {
+  effects <- direct_effect(fit) |>
+    effect_summary()
+  expect_type(effects, "data.frame")
+})
+
+test_that("Can estimate indirect effects on toy dataset.", {
+  effects <- indirect_overall(fit) |>
+    effect_summary()
+  expect_type(effects, "data.frame")
+})