tests/testthat/test-death_age.R

In jesse-smith/covidReport: SCHD Package for Reporting COVID-19 Data

test_that("`death_plot_age()` matches doppelganger", {
  tbl_mock <- mockery::mock(tibble::tribble(
        ~ grp, ~ n, ~ rate, ~ percent,
       "0-17", 100,    0.2,      0.10,
      "18-24", 100,    0.1,      0.10,
      "25-34", 100,    0.1,      0.10,
      "35-44", 100,    0.1,      0.10,
      "45-54", 100,    0.1,      0.10,
      "55-64", 100,    0.1,      0.10,
      "65-74", 100,    0.1,      0.10,
      "75-84", 100,    0.1,      0.10,
        "85+", 90,    0.09,      0.09,
    "Missing", 10,    0.01,      0.01
  ))

  mockery::stub(
    death_plot_age,
    "death_calc_age",
    tbl_mock
  )

  m_dt <- mockery::mock(as.Date("2021-04-01"), cycle = TRUE)
  mockery::stub(
    death_plot_age,
    "date_inv",
    m_dt
  )

  plt <- death_plot_age(tibble::tibble(), date = "2021-04-01")

  mockery::expect_called(m_dt, n = 1L)
  suppressWarnings(
    vdiffr::expect_doppelganger(
      title = "death rate by age",
      fig = plt,
      path = "death-plot-age"
    )
  )
})

test_that("`void(death_table_age())` matches snapshot", {
  tbl_mock <- mockery::mock(tibble::tribble(
        ~ grp, ~ n, ~ rate, ~ percent,
       "0-17", 100,    0.2,      0.10,
      "18-24", 100,    0.1,      0.10,
      "25-34", 100,    0.1,      0.10,
      "35-44", 100,    0.1,      0.10,
      "45-54", 100,    0.1,      0.10,
      "55-64", 100,    0.1,      0.10,
      "65-74", 100,    0.1,      0.10,
      "75-84", 100,    0.1,      0.10,
        "85+", 90,    0.09,      0.09,
    "Missing", 10,    0.01,      0.01
  ))

  mockery::stub(
    death_table_age,
    "death_calc_age",
    tbl_mock
  )

  data <- tibble::tibble()

  tbl_void <- flextable::void(
    death_table_age(data, date = "2021-04-01"),
    part = "all"
  )

  expect_snapshot(tbl_void)
})

test_that("`death_calc_age()` matches snapshot", {

  data <- tibble::tibble(
    patient_dob = rep(
      as.Date("2021-04-01") - seq(1, 36e3, length.out = 100),
      times = 1000L
    ),
    die_from_illness_ind = rep(
      c("Y", rep("N", 3L), rep("U", 3L), rep(NA, 3L)),
      1e4L
    ),
    inv_death_dt = seq(
      as.Date("2020-03-26"),
      as.Date("2021-03-26"),
      length.out = 99999L
    ) %>% purrr::prepend(lubridate::NA_Date_),
    patient_deceased_dt = c(rep(as.Date("2020-03-10"), 99e3L), rep("2021-03-26", 1e3L)),
    inv_start_dt = patient_deceased_dt - 1L
  ) %>%
    dplyr::mutate(
      dplyr::across(
        where(lubridate::is.Date) & !c("patient_dob"),
        ~ format(.x, "%Y-%m-%dT%H:%M:%SZ")
      )
    ) %>%
    dplyr::slice_sample(prop = 1)

  mockery::stub(
    filter_deaths,
    "date_inv",
    lubridate::as_date
  )

  data_mock <- mockery::mock(filter_deaths(data))

  mockery::stub(
    death_calc_age,
    "filter_deaths",
    data_mock
  )

  tbl_death <- death_calc_age(data, date = "2021-04-01")

  expect_snapshot(tbl_death)
})