tests/testthat/test-checks.R

In o2ools: Tools for 'outbreaker2'

test_that("is_outbreaker_chains works correctly", {
  df_chains <- data.frame(a = 1)
  class(df_chains) <- c("outbreaker_chains", "data.frame")
  df_no_chains <- data.frame(a = 1)
  not_df <- list(a = 1)

  expect_true(is_outbreaker_chains(df_chains))
  expect_error(is_outbreaker_chains(df_no_chains), "'out' must be an object of class 'outbreaker_chains'.")
  expect_error(is_outbreaker_chains(not_df), "'out' must be a data frame.")
})

test_that("is_linelist works correctly", {
  df_linelist <- data.frame(id = 1:3, val = letters[1:3])
  df_no_id <- data.frame(val = letters[1:3])
  not_df <- list(id = 1:3)

  expect_true(is_linelist(df_linelist))
  expect_error(is_linelist(df_no_id), "'linelist' must contain an 'id' column.")
  expect_error(is_linelist(not_df), "'linelist' must be a data frame.")
})