tests/testthat/test-data-population.R

In mrc-ide/naomi: Naomi Model for Subnational HIV Estimates

test_that("test can convert between time ids", {
  expect_equal(convert_quarter_id(2012, 1), 449)
  expect_equal(convert_quarter_id(2012, 4), 452)
  expect_equal(convert_calendar_quarter(2012, 1), "CY2012Q1")
  expect_equal(convert_calendar_quarter(2012, 4), "CY2012Q4")
  expect_equal(calendar_quarter_to_quarter_id("CY2012Q1"), 449)
  expect_equal(calendar_quarter_to_quarter_id("CY2012Q4"), 452)
  expect_equal(calendar_quarter_to_year("CY2012Q4"), 2012)
  expect_equal(calendar_quarter_to_year("CY2012Q1"), 2012)
})

test_that("can get quarter year labels", {
  expect_equal(quarter_year_labels(492), "December 2022")
  expect_equal(quarter_year_labels(c(492, 493, 494, 495)),
               c("December 2022", "March 2023", "June 2023", "September 2023"))
})

test_that("quarter_id_to_calendar_quarter", {
  expect_equal(quarter_id_to_calendar_quarter(449), "CY2012Q1")
  expect_equal(quarter_id_to_calendar_quarter(452), "CY2012Q4")
  expect_equal(quarter_id_to_calendar_quarter(c(449, 452)),
               c("CY2012Q1", "CY2012Q4"))
})