tests/testthat/test-repeat_baseline_values.R

In lvmisc: Veras Miscellaneous

test_that("error handling works", {
  df <- data.frame(x = 1, y = 2, z = 3)

  expect_error(
    repeat_baseline_values(df, score, x, y, 1),
    "Column `score` not found in `df`.",
    class = "error_column_not_found"
  )
  expect_error(
    repeat_baseline_values(df, x, id, y, 1),
    "Column `id` not found in `df`.",
    class = "error_column_not_found"
  )
  expect_error(
    repeat_baseline_values(df, x, y, time, 1),
    "Column `time` not found in `df`.",
    class = "error_column_not_found"
  )
  expect_error(
    repeat_baseline_values(df, x, y, z, 1, repeat_NA = "yes"),
    "`repeat_NA` must be logical; not character.",
    class = "error_argument_type"
  )
})

test_that("repeat_baseline_values() works", {
  df <- data.frame(
    id = rep(1:5, each = 4),
    time = rep(1:4, 5),
    score = rep(c(10:12, NA), 5)
  )
  out <- repeat_baseline_values(df, score, id, time, 1, repeat_NA = FALSE)

  expect_equal(out, rep(10, 20))
})

test_that("repeat_NA argument works", {
  df <- data.frame(
    id = rep(1:5, each = 4),
    time = rep(1:4, 5),
    score = rep(c(10:12, NA), 5)
  )
  out <- repeat_baseline_values(df, score, id, time, 1)

  expect_equal(out, rep(c(rep(10, 3), NA), 5))
})