tests/testthat/test-alert_serfling.R

In CDCgov/Rnssp: A Signature R package for the National Syndromic Surveillance Program (NSSP)

context("test-alert_serfling")

test_that("alert_serfling functions as expected!", {
  df <- data.frame(
    date = seq(as.Date("2014-01-05"), as.Date("2022-02-05"), "weeks"),
    count = rpois(length(seq(as.Date("2014-01-05"), as.Date("2022-02-05"), "weeks")), 25)
  )

  df2 <- data.frame(
    date = rep(seq(as.Date("2014-01-05"), as.Date("2022-02-05"), "weeks"), 2),
    regions = c(rep("reg1", 422), rep("reg2", 422)),
    count = floor(runif(422 * 2, min = 0, max = 101))
  )

  df3 <- data.frame(
    date = rep(seq(as.Date("2014-01-05"), as.Date("2022-02-05"), "weeks"), 2),
    regions = c(rep("reg1", 422), rep("reg2", 422)),
    count = floor(runif(422 * 2, min = -100, max = 100))
  )

  df_serfling <- alert_serfling(df, baseline_end = as.Date("2020-03-01"))
  df2_serfling <- alert_serfling(group_by(df2, regions),
                             baseline_end = as.Date("2020-03-01"))

  expect(is.data.frame(df_serfling), failure_message = "alert_nbinom test fails!")
  expect_equal(length(df_serfling), length(df) + 7)
  expect(is.data.frame(df2_serfling), failure_message = "alert_nbinom test fails!")
  expect_equal(length(df2_serfling), length(df2) + 7)

  expect_error(alert_serfling(df, baseline_end = as.Date("2014-01-31")))
  expect_error(alert_serfling(df2, baseline_end = as.Date("2020-03-01")))
  expect_error(alert_serfling(df3, baseline_end = as.Date("2020-03-01")))
})