tests/testthat/test-prob_lmod.R

In FrankLef/fciR: Data and functions used with Fundamentals of Causal Inference by Babette Brumback

test_that("prob_lmod", {
  data("whatifdat")
  data <- whatifdat

  out <- prob_lmod(data,
                   formula = Y ~ `T` + A + H,
                   condition.names = c("T", "A"))
  out$estimate[1] <- round(out$estimate[1], 4)
  # cat("\n")
  # print(out)
  # cat("\n")

  target <- c("logitP" = round(qlogis(0.09001343), 4))
  target <- data.frame(
    "term" = "logitP",
    "estimate" = round(qlogis(0.09001343), 4),
    "std.err" = NA_real_
  )
  # cat("\n")
  # print(target)
  # cat("\n")

  expect_identical(out, target)
})


test_that("prob_lmod_td", {
  data("whatifdat")
  data <- whatifdat

  out <- prob_lmod_td(data, formula = Y ~ `T` + A + H,
                      condition.names = c("T", "A"))
  # cat("\n")
  # print(out)
  # cat("\n")
  out <- round(c("logitP" = out$estimate), 6)

  target <- c("logitP" = round(qlogis(0.09001343), 6))
  # cat("\n")
  # print(target)
  # cat("\n")

  expect_identical(out, target)
})