tests/testthat/test-estimate_ite_slearner.R

In CRE: Interpretable Discovery and Inference of Heterogeneous Treatment Effects

test_that("S-Learner ITE Estimated Correctly", {
  # Generate sample data
  skip_on_cran()
  set.seed(8697)
  dataset_cont <- generate_cre_dataset(n = 100, rho = 0, n_rules = 2, p = 10,
                                       effect_size = 0.5,
                                       binary_outcome = FALSE)
  y <- dataset_cont[["y"]]
  z <- dataset_cont[["z"]]
  X <- dataset_cont[["X"]]
  or_method <- "SL.xgboost"

  # Incorrect data inputs
  expect_error(estimate_ite_slearner(y = "test", z, X, or_method))
  expect_warning(expect_error(estimate_ite_slearner(y, z = "test", X,
                                                    or_method)))
  expect_error(estimate_ite_slearner(y, z, X = NA, or_method))

  # Correct outputs
  ite <- estimate_ite_slearner(y, z, X, or_method)
  expect_true(length(ite) == length(y))
  expect_true(class(ite) == "numeric")

  # Reproducible results
  expect_equal(ite[1], 0.4066854, tolerance = 0.00001)
  expect_equal(ite[11], -0.8926377, tolerance = 0.00001)
  expect_equal(ite[91], 1.136385, tolerance = 0.00001)
})