tests/testthat/test-add_splines.R

In ddsjoberg/bstfun: Functions for the MSK Biostatistics Department

test_that("add_spline function gives error if spline variable names exist", {

  # Error if not specifying variable names and they already exist
  expect_error(
    gtsummary::trial %>%
      add_splines(variable = age) %>%
      add_splines(variable = age),
    "*"
  )

  # No error if specifying variable names and default variable names already exist
  expect_error(
    gtsummary::trial %>%
      add_splines(variable = age) %>%
      add_splines(variable = age, new_names = c("sptest1", "sptest2", "sptest3")),
    NA
  )

  # Error if specifying variable names and they already exist
  expect_error(
    gtsummary::trial %>%
      add_splines(variable = age, new_names = c("sptest1", "sptest2", "sptest3")) %>%
      add_splines(variable = age, new_names = c("sptest1", "sptest2", "sptest3")),
    "*"
  )

  # Error if new_names does not have same length as the number of new column
  expect_error(
    gtsummary::trial %>%
      add_splines(variable = age, new_names = c("sptest1", "sptest2")) %>%
      add_splines(variable = age, new_names = c("sptest1", "sptest2", "sptest3")),
        "`new_names=` must be the same length as the number of new columns (n = 3)", fixed = TRUE
)
})