tests/testthat/test-bootstrap.R

In audreyrenson/didgformula: The difference-in-differences g-formula

test_that("bootstrap_se returns results", {
  set.seed(173)

  Tt = 4
  N = 100
  Beta = generate_parameters(Tt = Tt,
                             mu_Beta_Y = c(.2,.2,.2,0,0,.2), sd_Beta_Y = c(.2,.2,.2,0,0,.2),
                             mu_Beta_A = c(.2,.2,.2,0,0), sd_Beta_A = c(.2,.2,.2,0,0))
  df = generate_data(N = N,
                     Tt = Tt,
                     Beta = Beta)

  boot_iptw = bootstrap_se(
    data = df,
    nboots = 3,
    estimator = iptw_pipeline,
    den_formula = '~L{t}',
    Tt = Tt
  )
  boot_ice = bootstrap_se(
    data = df,
    nboots = 3,
    estimator = ice_pipeline,
    inside_formula_t = '~L{t}',
    inside_formula_tmin1 = '~L{t-1}',
    outside_formula = '~L{k}',
    Tt = Tt
  )
  boot_or = bootstrap_se(
    data = df,
    nboots = 3,
    estimator = or_pipeline,
    yt_formula = '~L{t}',
    ytmin1_formula = '~L{t-1}',
    l_formula = '~1',
    nreps = 100,
    Tt = Tt
  )

  expect_equal(dim(boot_iptw), c(Tt, 2))
  expect_equal(dim(boot_ice), c(Tt, 2))
  expect_equal(dim(boot_or), c(Tt, 2))
})