tests/testthat/test-lag-lead.R

In tidytable: Tidy Interface to 'data.table'

test_that("lag works on a vector", {

  x <- c(1:5)
  vec <- lag(x,1)

  expect_equal(vec, c(NA, 1,2,3,4))
})

test_that("lead works on a vector", {

  x <- c(1:5)
  vec <- lead(x,1)

  expect_equal(vec, c(2,3,4,5, NA))
})

test_that("lag works with mutate. on df", {

  df <- data.table(x = c(4,3,9,7), y = 1:4)
  df <- df %>%
    mutate(x_lag = lag(x,1))

  expect_equal(df$x_lag, c(NA, 4,3,9))

})

test_that("lead works with mutate on df", {

  df <- data.table(x = c(4,3,9,7), y = 1:4)
  df <- df %>%
    mutate(x_lead = lead(x,1))

  expect_equal(df$x_lead, c(3,9,7,NA))
})

test_that("lags. works on a vector", {

  x <- c(1:5)
  vec <- lags.(x, 1) %>% suppressWarnings()

  expect_equal(vec, c(NA, 1,2,3,4))
})

test_that("leads. works on a vector", {

  x <- c(1:5)
  vec <- leads.(x, 1) %>% suppressWarnings()

  expect_equal(vec, c(2,3,4,5, NA))
})