tests/testthat/test-lead_panel.R

In panelWranglR: Panel Data Wrangling Tools

test_that("Leading a panel works", {


  X <- matrix(rnorm(800000),8000,100)
  tim <- seq(1:4000)
  geo_AT <- rep(c("AT"), length = 4000)
  geo_NO <- rep(c("NO"), length = 4000)
  both_vec_1 <- cbind(tim,geo_NO)
  both_vec_2 <- cbind(tim,geo_AT)
  both <- rbind(both_vec_1,both_vec_2)
  names(both[,"geo_NO"]) <- "geo"
  X <- cbind(both,X)

  lead_result_1 <- panel_lead(data = X,
                              cross.section = "geo_NO",
                              time.variable = "tim",
                              leads = 5,
                              keep.original = TRUE)

  expect_equal(ncol(lead_result_1), 602)
  expect_equal(nrow(lead_result_1), 8000)
  expect_equal(sum(is.na(lead_result_1)), 3000)
  expect_equal(unname(unlist(lead_result_1[1,103])), as.numeric(unlist(X[2,3])))

  lead_result_2 <- panel_lead(data = X,
                              cross.section = "geo_NO",
                              leads = 5,
                              keep.original = FALSE)

  expect_equal(ncol(lead_result_2), 505)
  expect_equal(nrow(lead_result_2), 8000)
  expect_equal(sum(is.na(lead_result_2)), 3030)
  expect_equal(unname(unlist(lead_result_2[1,6])), as.numeric(unlist(X[2,3])))

  lead_result_3 <- panel_lead(data = X,
                              cross.section = "geo_NO",
                              leads = 5,
                              variables.selected = c("tim","V5"),
                              keep.original = FALSE)

  expect_equal(ncol(lead_result_3), 10)
  expect_equal(nrow(lead_result_3), 8000)
  expect_equal(sum(is.na(lead_result_3)), 60)
  expect_equal(unname(unlist(lead_result_3[1,6])), as.numeric(unlist(X[2,5])))


})