inst/tinytest/test_runner_vec.R

In gogonzo/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))
at <- sample(1:100, 10)
at_date <- sample(idx, 10)
sum2 <- function(x, na_rm = TRUE) {
  if (all(is.na(x)) || length(x) == 0) {
    return(NA)
  } else {
    sum(x, na.rm = na_rm)
  }
}

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

expect_equal(
  sum_run(x2, lag = 3, na_pad = TRUE),
  runner(x2, lag = 3, f = sum2, na_pad = TRUE)
)



expect_equal(
  sum_run(x2, lag = 3)[at],
  sum_run(x2, lag = 3, at = seq_along(x2)[at])
)

expect_equal(
  sum_run(x2, lag = 3, na_pad = TRUE)[at],
  sum_run(x2, lag = 3, at = seq_along(x2)[at], na_pad = TRUE)
)

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

expect_equal(
  sum_run(x2, lag = -3, na_pad = TRUE),
  runner(x2, lag = -3, f = sum2, na_pad = TRUE)
)


expect_equal(
  sum_run(x2, lag = -3)[at],
  sum_run(x2, lag = -3, at = seq_along(x2)[at])
)

expect_equal(
  sum_run(x2, lag = -3, na_pad = TRUE)[at],
  sum_run(x2, lag = -3, at = seq_along(x2)[at], na_pad = TRUE)
)

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

expect_equal(
  sum_run(x2, lag = 100, na_pad = TRUE),
  as.numeric(
    runner(x2, lag = 100, f = sum2, na_pad = TRUE)
  )
)

expect_equal(
  sum_run(x2, lag = -100),
  runner(x2, lag = -100, f = sum2)
)

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



expect_equal(
  sum_run(x2, lag = 100)[at],
  sum_run(x2, lag = 100, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, lag = 100, na_pad = TRUE)[at],
  sum_run(x2, lag = 100, at = seq_along(x2), na_pad = TRUE)[at])

expect_equal(
  sum_run(x2, lag = -100)[at],
  sum_run(x2, lag = -100, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, lag = -100, na_pad = TRUE)[at],
  sum_run(x2, lag = -100, at = seq_along(x2)[at], na_pad = TRUE))

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

expect_equal(
  sum_run(x2, k = 3, na_pad = TRUE),
  runner(x2, k = 3, f = sum2, na_pad = TRUE))


expect_equal(
  sum_run(x2, k = 3)[at],
  sum_run(x2, k = 3, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 3, na_pad = TRUE)[at],
  sum_run(x2, k = 3, at = seq_along(x2)[at], na_pad = TRUE))

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

expect_equal(
  sum_run(x2, k = 1, na_pad = TRUE),
  runner(x2, k = 1, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 99),
  runner(x2, k = 99, f = sum2))

expect_equal(
  sum_run(x2, k = 99, na_pad = TRUE),
  runner(x2, k = 99, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 100),
  runner(x2, k = 100, f = sum2))

expect_equal(
  sum_run(x2, k = 100, na_pad = TRUE),
  runner(x2, k = 100, f = sum2, na_pad = TRUE))



expect_equal(
  sum_run(x2, k = 1)[at],
  sum_run(x2, k = 1, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 1, na_pad = TRUE)[at],
  sum_run(x2, k = 1, at = seq_along(x2)[at], na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 99)[at],
  sum_run(x2, k = 99, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 99, na_pad = TRUE)[at],
  sum_run(x2, k = 99, at = seq_along(x2)[at], na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 100)[at],
  sum_run(x2, k = 100, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 100, na_pad = TRUE)[at],
  sum_run(x2, k = 100, at = seq_along(x2)[at], na_pad = TRUE))

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

expect_equal(
  sum_run(x2, k = 5, lag = 3, na_pad = TRUE),
  runner(x2, k = 5, lag = 3, f = sum2, na_pad = TRUE))

#here!
expect_equal(
  sum_run(x2, k = 5, lag = 3, na_rm = FALSE),
  runner(x2, k = 5, lag = 3, f = sum2, na_rm = FALSE)
)

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



expect_equal(
  sum_run(x2, k = 5, lag = 3)[at],
  sum_run(x2, k = 5, lag = 3, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 5, lag = 3, na_pad = TRUE)[at],
  sum_run(x2, k = 5, lag = 3, at = seq_along(x2)[at], na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 5, lag = 3, na_rm = FALSE)[at],
  sum_run(x2, k = 5, lag = 3, at = seq_along(x2)[at], na_rm = FALSE))

expect_equal(
  sum_run(x2, k = 5, lag = 3, na_pad = TRUE, na_rm = FALSE)[at],
  sum_run(x2, k = 5, lag = 3, at = seq_along(x2)[at],
          na_pad = TRUE, na_rm = FALSE)
)

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

expect_equal(
  sum_run(x2, k = 5, lag = -3, na_pad = TRUE),
  runner(x2, k = 5, lag = -3, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 5, lag = -3, na_rm = FALSE),
  runner(x2, k = 5, lag = -3, f = sum2, na_rm = FALSE)
)

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



expect_equal(
  sum_run(x2, k = 5, lag = -3)[at],
  sum_run(x2, k = 5, lag = -3, at = seq_along(x2)[at])
)

expect_equal(
  sum_run(x2, k = 5, lag = -3, na_pad = TRUE)[at],
  sum_run(x2, k = 5, lag = -3, at = seq_along(x2)[at], na_pad = TRUE)
)

expect_equal(
  sum_run(x2, k = 5, lag = -3, na_rm = FALSE)[at],
  sum_run(x2, k = 5, lag = -3, at = seq_along(x2)[at], na_rm = FALSE)
)

expect_equal(
  sum_run(x2, k = 5, lag = -3, na_pad = TRUE, na_rm = FALSE)[at],
  sum_run(x2, k = 5, lag = -3, at = seq_along(x2)[at],
          na_pad = TRUE, na_rm = FALSE)
)

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

expect_equal(
  sum_run(x2, k = 5, lag = -7, na_pad = TRUE),
  runner(x2, k = 5, lag = -7, f = sum2, na_pad = TRUE))



expect_equal(
  sum_run(x2, k = 5, lag = -7)[at],
  sum_run(x2, k = 5, lag = -7, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 5, lag = -7, na_pad = TRUE)[at],
  sum_run(x2, k = 5, lag = -7, at = seq_along(x2)[at], na_pad = TRUE))

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

expect_equal(
  sum_run(x2, k = 1, lag = -1, na_pad = TRUE),
  runner(x2, k = 1, lag = -1, f = sum2, na_pad = TRUE))



expect_equal(
  sum_run(x2, k = 1, lag = -1)[at],
  sum_run(x2, k = 1, lag = -1, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 1, lag = -1, na_pad = TRUE)[at],
  sum_run(x2, k = 1, lag = -1, at = seq_along(x2)[at], na_pad = TRUE))

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

expect_equal(
  sum_run(x2, k = 1, lag = 1, na_pad = TRUE),
  runner(x2, k = 1, lag = 1, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 1, lag = 1)[at],
  sum_run(x2, k = 1, lag = 1, at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 1, lag = 1, na_pad = TRUE)[at],
  sum_run(x2, k = 1, lag = 1, at = seq_along(x2)[at], na_pad = TRUE))

# various ------
expect_equal(
  sum_run(x2, k = k, lag = 1),
  runner(x2, k = k, lag = 1, f = sum2))

expect_equal(
  sum_run(x2, k = k, lag = 1, na_pad = TRUE),
  runner(x2, k = k, lag = 1, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = lag),
  runner(x2, lag = lag, f = sum2))

expect_equal(
  sum_run(x2, k = 3, lag = lag),
  runner(x2, k = 3, lag = lag, f = sum2))

expect_equal(
  sum_run(x2, k = 3, lag = lag, na_pad = TRUE),
  runner(x2, k = 3, lag = lag, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = lag),
  runner(x2, k = k, lag = lag, f = sum2))

expect_equal(
  sum_run(x2, k = k, lag = lag, na_pad = TRUE),
  runner(x2, k = k, lag = lag, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = 1)[at],
  sum_run(x2, k = k[at], lag = 1, at = seq_along(x2)[at]))


expect_equal(
  sum_run(x2, k = k, lag = 1, na_pad = TRUE)[at],
  sum_run(x2, k = k[at], lag = 1, at = seq_along(x2)[at], na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = lag)[at],
  sum_run(x2, lag = lag[at], at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 3, lag = lag)[at],
  sum_run(x2, k = 3, lag = lag[at], at = seq_along(x2)[at]))

expect_equal(
  sum_run(x2, k = 3, lag = lag, na_pad = TRUE)[at],
  sum_run(x2, k = 3, lag = lag[at], at = seq_along(x2)[at], na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = lag)[at],
  sum_run(x2, k = k[at], lag = lag[at], at = seq_along(x2)[at])
)

expect_equal(
  sum_run(x2, k = k, lag = lag, na_pad = TRUE)[at],
  sum_run(x2, k = k[at], lag = lag[at], at = seq_along(x2)[at], na_pad = TRUE)
)

# date window ------
expect_equal(
  sum_run(x2, idx = idx, na_pad = FALSE),
  runner(x2, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, idx = idx, na_pad = TRUE),
  runner(x2, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, lag = 3, idx = idx, na_pad = FALSE),
  runner(x2, lag = 3, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, lag = 3, idx = idx, na_pad = TRUE),
  runner(x2, lag = 3, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, lag = -3, idx = idx, na_pad = FALSE),
  runner(x2, lag = -3, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, lag = -3, idx = idx, na_pad = TRUE),
  runner(x2, lag = -3, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, k = 3, idx = idx, na_pad = FALSE),
  runner(x2, k = 3, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, k = 3, idx = idx, na_pad = TRUE),
  runner(x2, k = 3, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, lag = -1, idx = idx, na_pad = FALSE),
  runner(x2, lag = -1, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, lag = -1, idx = idx, na_pad = TRUE),
  runner(x2, lag = -1, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, lag = 100, idx = idx, na_pad = FALSE),
  runner(x2, lag = 100, idx = idx, f = sum2, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, lag = 100, idx = idx, na_pad = TRUE),
  runner(x2, lag = 100, idx = idx, f = sum2, na_pad = TRUE)
)

expect_equal(
  sum_run(x2, lag = -100, idx = idx, na_pad = FALSE),
  runner(x2, lag = -100, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = -100, idx = idx, na_pad = TRUE),
  runner(x2, lag = -100, idx = idx, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = lag, idx = idx, na_pad = FALSE),
  runner(x2, lag = lag, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = lag, idx = idx, na_pad = TRUE),
  runner(x2, lag = lag, idx = idx, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 3, lag = 4, idx = idx, na_pad = FALSE),
  runner(x2, k = 3, lag = 4, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 3, lag = 4, idx = idx, na_pad = TRUE),
  runner(x2, k = 3, lag = 4, idx = idx, f = sum2, na_pad = TRUE))

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

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

expect_equal(
  sum_run(x2, k = k, lag = 0, idx = idx, na_pad = FALSE),
  runner(x2, k = k, lag = 0, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = k, lag = 0, idx = idx, na_pad = TRUE),
  runner(x2, k = k, lag = 0, idx = idx, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = -4, idx = idx, na_pad = FALSE),
  runner(x2, k = k, lag = -4, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = k, lag = -4, idx = idx, na_pad = TRUE),
  runner(x2, k = k, lag = -4, idx = idx, f = sum2, na_pad = TRUE))


expect_equal(
  sum_run(x2, k = 4, lag = lag, idx = idx, na_pad = FALSE),
  runner(x2, k = 4, lag = lag, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 4, lag = lag, idx = idx, na_pad = TRUE),
  runner(x2, k = 4, lag = lag, idx = idx, f = sum2, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = lag, idx = idx, na_pad = FALSE),
  runner(x2, k = k, lag = lag, idx = idx, f = sum2, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = k, lag = lag, idx = idx, na_pad = TRUE),
  runner(x2, k = k, lag = lag, idx = idx, f = sum2, na_pad = TRUE))

# at date ------
ids <- match(at_date, idx)
expect_equal(
  sum_run(x2, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = 3, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, lag = 3, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = 3, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = 3, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = -3, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, lag = -3, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = -3, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = -3, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 3, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = 3, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 3, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = 3, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = -1, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = -1, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = -1, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = -1, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = 100, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, lag = 100, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = 100, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = 100, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = -100, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, lag = -100, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = -100, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = -100, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, lag = lag, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, lag = lag[ids], idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, lag = lag, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, lag = lag[ids], idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 3, lag = 4, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = 3, lag = 4, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 3, lag = 4, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = 3, lag = 4, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 3, lag = -4, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = 3, lag = -4, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 3, lag = -4, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = 3, lag = -4, idx = idx, at = at_date, na_pad = TRUE))


expect_equal(
  sum_run(x2, k = k, lag = -4, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = k[ids], lag = -4, idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = k, lag = -4, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = k[ids], lag = -4, idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = 4, lag = lag, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = 4, lag = lag[ids], idx = idx, at = at_date, na_pad = FALSE))

expect_equal(
  sum_run(x2, k = 4, lag = lag, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = 4, lag = lag[ids], idx = idx, at = at_date, na_pad = TRUE))

expect_equal(
  sum_run(x2, k = k, lag = lag, idx = idx, na_pad = FALSE)[ids],
  sum_run(x2, k = k[ids], lag = lag[ids], idx = idx,
          at = at_date, na_pad = FALSE)
)

expect_equal(
  sum_run(x2, k = k, lag = lag, idx = idx, na_pad = TRUE)[ids],
  sum_run(x2, k = k[ids], lag = lag[ids], idx = idx,
          at = at_date, na_pad = TRUE)
)