tests/testthat/test-pensynth.R

In pensynth: Penalized Synthetic Control Estimation

set.seed(123)

w <- c(.5, .4, .1, 0, 0, 0)
X0 <- matrix(rnorm(3*6), 3)
X1 <- X0 %*% w


test_that("Basic pensynth works", {
  res <- pensynth(X1, X0, 1, lambda = 0, verbose = FALSE)
  expect_length(res$w, 6)
  expect_lt(sum(abs(res$w - w)), 5e-3)
})

test_that("Pensynth with high lambda works", {
  res <- pensynth(X1, X0, 1, lambda = 100, verbose = FALSE)
  expect_gt(max(res$w), 0.99)
})


test_that("Pensynth with mutiple donors works", {
  w <- matrix(c(w, 0, 0, 0, .5, .4, .1), nrow = 6)
  X1 <- X0 %*% w
  res <- pensynth(X1, X0, 1, lambda = 1e-5, verbose = FALSE)
  expect_equal(dim(res$w), c(6, 2))
  expect_equal(dim(predict(res, X0)), dim(X1))
  expect_lt(sum(abs(res$w - w)), 5e-3)
})