tests/testthat/test-AddSmPath.R

In eventstudyr: Estimation and Visualization of Linear Panel Event Studies

test_that("AddZerosCovar correctly adds zeros where normalized coefficient should be", {

  vcov_matrix_all <- diag(rep(3,10))
  eventstudy_coeffs  <- c(paste0("a", 1:2), paste0("a", 4:9))
  rownames(vcov_matrix_all) <- c(eventstudy_coeffs, "a10", "a11")
  colnames(vcov_matrix_all) <- c(eventstudy_coeffs, "a10", "a11")

  norm_column  <- "a3"
  coeffs_oder  <- paste0("a", 1:9)

  expected_matrix <- diag(c(rep(3,2), 0, rep(3,6)))
  rownames(expected_matrix) <- paste0("a", 1:9)
  colnames(expected_matrix) <- paste0("a", 1:9)

  covar <- AddZerosCovar(vcov_matrix_all, eventstudy_coeffs, norm_column, coeffs_oder)

  expect_equal(covar, expected_matrix)
})

test_that("AddZerosCovar correctly adds zeros where normalized coefficient should be with 2 normalizations", {

    vcov_matrix_all <- diag(rep(3,9))
    eventstudy_coeffs  <- c(paste0("a", 1:2), paste0("a", 4), paste0("a", 6:10))
    rownames(vcov_matrix_all) <- c(eventstudy_coeffs, "a11")
    colnames(vcov_matrix_all) <- c(eventstudy_coeffs, "a11")

    norm_column  <- c("a3", "a5")
    coeffs_oder  <- paste0("a", 1:10)

    expected_matrix <- diag(c(rep(3,2), 0, 3, 0, rep(3,5)))
    rownames(expected_matrix) <- paste0("a", 1:10)
    colnames(expected_matrix) <- paste0("a", 1:10)

    covar <- AddZerosCovar(vcov_matrix_all, eventstudy_coeffs, norm_column, coeffs_oder)

    expect_equal(covar, expected_matrix)
})