tests/testthat/test-create_age_groups.R

In Health-SocialCare-Scotland/phimethods: Standard Methods for Use in Public Health Scotland

test_that("Return correct age groups", {
  expect_identical(
    create_age_groups(c(4, 51, 21, 89),
      0, 80, 10,
      as_factor = FALSE
    ),
    c("0-9", "50-59", "20-29", "80+")
  )

  expect_identical(
    create_age_groups(c(8, 94, 44, 55, 14),
      0, 90, 5,
      as_factor = TRUE
    ),
    factor(c("5-9", "90+", "40-44", "55-59", "10-14"),
      levels = c(
        "0-4", "5-9", "10-14", "15-19", "20-24", "25-29", "30-34",
        "35-39", "40-44", "45-49", "50-54", "55-59", "60-64",
        "65-69", "70-74", "75-79", "80-84", "85-89", "90+"
      ),
      ordered = TRUE
    )
  )

  expect_identical(
    create_age_groups(c(81, 86, 33, 11),
      4, 84, 3,
      as_factor = FALSE
    ),
    c("79-81", "82+", "31-33", "10-12")
  )

  expect_identical(
    create_age_groups(c(0, 99, 1000, 5, 5),
      5, 90, 5,
      as_factor = FALSE
    ),
    c(NA, "90+", "90+", "5-9", "5-9")
  )
})

test_that("Default value for age groups", {
  expect_identical(
    create_age_groups(10, as_factor = TRUE),
    factor(c("10-14"),
      levels = c(
        "0-4", "5-9", "10-14", "15-19", "20-24", "25-29", "30-34",
        "35-39", "40-44", "45-49", "50-54", "55-59", "60-64",
        "65-69", "70-74", "75-79", "80-84", "85-89", "90+"
      ),
      ordered = TRUE
    )
  )
})


test_that("Handling of non-numeric values for x", {
  # If x is not numeric cut will error
  expect_error(
    create_age_groups(c("1", "57", "apple", "12"), as_factor = FALSE)
  )

  # This is true even if all elements are numbers stored as character
  expect_error(
    create_age_groups(c("26", "9", "78", "81"), as_factor = FALSE)
  )
})