runner: Running Operations for Vectors

set.seed(11)
x1 <- sample(c(1, 2, 3), 100, replace = TRUE)
x2 <- sample(c(NA, 1, 2, 3), 100, replace = TRUE)
k <- sample(1:100, 100, replace = TRUE)
lag <- sample(-15:15, 100, replace = TRUE)
idx <- cumsum(sample(c(1, 2, 3, 4), 100, replace = TRUE))
mean2 <- function(x, na_rm = TRUE) {
  if (all(is.na(x))) return(NA) else mean(x, na.rm = na_rm)
}

#       |--------]-------> ------
expect_equal(
  mean_run(x2),
  runner(x2, f = mean2)
)

expect_equal(
  mean_run(x2, na_pad = TRUE),
  runner(x2, f = mean2, na_pad = TRUE)
)

expect_identical(
  mean_run(x2, na_rm = FALSE),
  runner(x2, function(x) mean2(x, na_rm = FALSE))
)

#   [...|----]---+-------> ------
expect_equal(
  mean_run(x2, lag = 3),
  runner(x2, lag = 3, f = mean2))

expect_equal(
  mean_run(x2, lag = 3, na_pad = TRUE),
  runner(x2, lag = 3, f = mean2, na_pad = TRUE))

#       |--------+---]---> ------
expect_equal(
  mean_run(x2, lag = -3),
  runner(x2, lag = -3, f = mean2))

expect_equal(
  mean_run(x2, lag = -3, na_pad = TRUE),
  runner(x2, lag = -3, f = mean2, na_pad = TRUE))

#  [...]|--------+-------> ------
expect_equal(
  mean_run(x2, lag = 100),
  as.numeric(runner(x2, lag = 100, f = mean2))
)

expect_equal(
  mean_run(x2, lag = 100, na_pad = TRUE),
  as.numeric(
    runner(x2, lag = 100, f = mean2, na_pad = TRUE)
  )
)

expect_equal(
  mean_run(x2, lag = -100),
  runner(x2, lag = -100, f = mean2))

expect_equal(
  mean_run(x2, lag = -100, na_pad = TRUE),
  as.numeric(
    runner(x2, lag = -100, f = mean2, na_pad = TRUE)
  )
)

#       |----[...]-------> ------
expect_equal(
  mean_run(x2, k = 3),
  runner(x2, k = 3, f = mean2))

expect_equal(
  mean_run(x2, k = 3, na_pad = TRUE),
  runner(x2, k = 3, f = mean2, na_pad = TRUE))

#       [...|--------+-------[...] ------
expect_equal(
  mean_run(x2, k = 1),
  runner(x2, k = 1, f = mean2))

expect_equal(
  mean_run(x2, k = 1, na_pad = TRUE),
  runner(x2, k = 1, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 99),
  runner(x2, k = 99, f = mean2))

expect_equal(
  mean_run(x2, k = 99, na_pad = TRUE),
  runner(x2, k = 99, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 100),
  runner(x2, k = 100, f = mean2))

expect_equal(
  mean_run(x2, k = 100, na_pad = TRUE),
  runner(x2, k = 100, f = mean2, na_pad = TRUE))

#       [...|----]---+-------> ------
expect_equal(
  mean_run(x2, k = 5, lag = 3),
  runner(x2, k = 5, lag = 3, f = mean2))

expect_equal(
  mean_run(x2, k = 5, lag = 3, na_pad = TRUE),
  runner(x2, k = 5, lag = 3, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 5, lag = 3, na_rm = FALSE),
  runner(x2, k = 5, lag = 3, f = mean))

expect_equal(
  mean_run(x2, k = 5, lag = 3, na_pad = TRUE, na_rm = FALSE),
  runner(x2, k = 5, lag = 3, f = mean, na_pad = TRUE))

#       |-----[--+---]---> ------
expect_equal(
  mean_run(x2, k = 5, lag = -3),
  runner(x2, k = 5, lag = -3, f = mean2))

expect_equal(
  mean_run(x2, k = 5, lag = -3, na_pad = TRUE),
  runner(x2, k = 5, lag = -3, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 5, lag = -3, na_rm = FALSE),
  runner(x2, k = 5, lag = -3, f = mean))

expect_equal(
  mean_run(x2, k = 5, lag = -3, na_pad = TRUE, na_rm = FALSE),
  runner(x2, k = 5, lag = -3, f = mean, na_pad = TRUE))

#       |--------+-[---]-> ------
expect_equal(
  mean_run(x2, k = 5, lag = -7),
  runner(x2, k = 5, lag = -7, f = mean2))

expect_equal(
  mean_run(x2, k = 5, lag = -7, na_pad = TRUE),
  runner(x2, k = 5, lag = -7, f = mean2, na_pad = TRUE))

#       |--------+[]-----> ------
expect_equal(
  mean_run(x2, k = 1, lag = -1),
  runner(x2, k = 1, lag = -1, f = mean2))

expect_equal(
  mean_run(x2, k = 1, lag = -1, na_pad = TRUE),
  runner(x2, k = 1, lag = -1, f = mean2, na_pad = TRUE))

#       |------[]+-------> ------
expect_equal(
  mean_run(x2, k = 1, lag = 1),
  runner(x2, k = 1, lag = 1, f = mean2))

expect_equal(
  mean_run(x2, k = 1, lag = 1, na_pad = TRUE),
  runner(x2, k = 1, lag = 1, f = mean2, na_pad = TRUE))

#various ------
expect_equal(
  mean_run(x2, k = k, lag = 1),
  runner(x2, k = k, lag = 1, f = mean2))

expect_equal(
  mean_run(x2, k = k, lag = 1, na_pad = TRUE),
  runner(x2, k = k, lag = 1, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, k = 3, lag = lag),
  runner(x2, k = 3, lag = lag, f = mean2))

expect_equal(
  mean_run(x2, k = 3, lag = lag, na_pad = TRUE),
  runner(x2, k = 3, lag = lag, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = k, lag = lag),
  runner(x2, k = k, lag = lag, f = mean2))

expect_equal(
  mean_run(x2, k = k, lag = lag, na_pad = TRUE),
  runner(x2, k = k, lag = lag, f = mean2, na_pad = TRUE))

#date window ------
expect_equal(
  mean_run(x2, lag = 3, idx = idx, na_pad = FALSE),
  runner(x2, lag = 3, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = 3, idx = idx, na_pad = TRUE),
  runner(x2, lag = 3, idx = idx, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, lag = -3, idx = idx, na_pad = FALSE),
  runner(x2, lag = -3, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = -3, idx = idx, na_pad = TRUE),
  runner(x2, lag = -3, idx = idx, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 3, idx = idx, na_pad = FALSE),
  runner(x2, k = 3, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, k = 3, idx = idx, na_pad = TRUE),
  runner(x2, k = 3, idx = idx, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, lag = -1, idx = idx, na_pad = FALSE),
  runner(x2, lag = -1, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = -1, idx = idx, na_pad = TRUE),
  runner(x2, lag = -1, idx = idx, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, lag = 100, idx = idx, na_pad = FALSE),
  runner(x2, lag = 100, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = 100, idx = idx, na_pad = TRUE),
  runner(x2, lag = 100, idx = idx, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, lag = -100, idx = idx, na_pad = FALSE),
  runner(x2, lag = -100, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = -100, idx = idx, na_pad = TRUE),
  runner(x2, lag = -100, idx = idx, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, lag = lag, idx = idx, na_pad = FALSE),
  runner(x2, lag = lag, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, lag = lag, idx = idx, na_pad = TRUE),
  runner(x2, lag = lag, idx = idx, f = mean2, na_pad = TRUE))

expect_equal(
  mean_run(x2, k = 3, lag = 4, idx = idx, na_pad = FALSE),
  runner(x2, k = 3, lag = 4, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, k = 3, lag = 4, idx = idx, na_pad = TRUE),
  runner(x2, k = 3, lag = 4, idx = idx, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, k = 3, lag = -4, idx = idx, na_pad = FALSE),
  runner(x2, k = 3, lag = -4, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, k = 3, lag = -4, idx = idx, na_pad = TRUE),
  runner(x2, k = 3, lag = -4, idx = idx, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, k = k, lag = -4, idx = idx, na_pad = FALSE),
  runner(x2, k = k, lag = -4, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, k = k, lag = -4, idx = idx, na_pad = TRUE),
  runner(x2, k = k, lag = -4, idx = idx, f = mean2, na_pad = TRUE))


expect_equal(
  mean_run(x2, k = 4, lag = lag, idx = idx, na_pad = FALSE),
  runner(x2, k = 4, lag = lag, idx = idx, f = mean2, na_pad = FALSE))

expect_equal(
  mean_run(x2, k = 4, lag = lag, idx = idx, na_pad = TRUE),
  runner(x2, k = 4, lag = lag, idx = idx, f = mean2, na_pad = TRUE))