tests/testthat/test-run_es.R

In fixes: Tools for Creating and Visualizing Fixed-Effects Event Study Models

test_that("run_es produces correct output with interval parameter", {
  # Create test data
  df <- tibble::tibble(
    id      = rep(1:5, each = 11),
    year    = rep(2000:2010, times = 5),
    y       = rnorm(55),
    treated = dplyr::if_else(id %in% c(1, 3, 5), 1, 0)
  )

  # Run the function with default interval (1-year steps)
  result_default <- df |>
    run_es(y, treated, year, 2005, 3, 2, id + year, "id")

  # Verify output for default interval
  expect_true("relative_time" %in% colnames(result_default))
  expect_equal(min(result_default$relative_time), -3)
  expect_equal(max(result_default$relative_time), 2)

  # Run the function with interval = 5 (5-year steps)
  result_interval <- df |>
    run_es(y, treated, year, 2005, 1, 1, id + year, "id", interval = 5)

  # Verify output for 5-year interval
  expect_true("relative_time" %in% colnames(result_interval))
  expect_equal(min(result_interval$relative_time), -5)
  expect_equal(max(result_interval$relative_time), 5)

  # Suppress warnings for out-of-range lead/lag
  expect_warning(
    run_es(df, y, treated, year, 2005, 10, 10, id + year, "id"),
    regexp = "exceeds the available range"
  )
})

test_that("run_es handles expression in outcome (e.g., log(y)) correctly", {
  # Create test data ensuring y is always positive for log()
  df2 <- tibble::tibble(
    id      = rep(1:5, each = 5),
    year    = rep(2000:2004, times = 5),
    y       = abs(rnorm(25)) + 0.1,  # shift so y > 0
    treated = dplyr::if_else(id %in% c(1, 3, 5), 1, 0)
  )

  # Run run_es() with log(y) as the outcome expression
  result_log <- df2 |>
    run_es(log(y), treated, year, 2002, lead_range = 1, lag_range = 1,
           fe = id, cluster = "id", interval = 1)

  # Check that the function produces a result without error
  expect_true("relative_time" %in% colnames(result_log))
  expect_true("estimate" %in% colnames(result_log))

  # Additional checks for baseline and term names
  expect_true(any(result_log$term == "lag0"))
  expect_true(any(result_log$term == "lead1"))
})