tests/testthat/test-sort.R

In icd: Comorbidity Calculations and Tools for ICD-9 and ICD-10 Codes

context("sorting, ordering ICD-9")

test_that("sorting preserves order of names", {
  a <- c("Bad disease" = "100", "Another bad disease" = "200")
  b <- sort.icd9(a)
  expect_identical(a[["Bad disease"]], b[["Bad disease"]])
  expect_identical(a[["Another bad disease"]], b[["Another bad disease"]])
})

test_that("sorting of icd9 object", {
  j <- icd9(c("Bad disease" = "500", "Another bad disease" = "400"))
  k <- sort(j)
  expect_identical(j[2], k[1])
  expect_identical(j[1], k[2])
  expect_true(as.icd9cm("V9199") < as.icd9cm("E000"))
  expect_true(as.icd9cm("E000") > as.icd9cm("V9199"))
})

test_that("order some icd-9 codes", {
  expect_equal(
    order.icd9(c("1002", "1001")),
    c(2L, 1L)
  )
})


test_that("NAs are ordered at end", {
  expect_identical(
    order.icd9(c(NA_character_, "100")),
    c(2L, 1L)
  )
})

context("sorting, ordering icd10cm")

test_that("basic compare works", {
  expect_false(icd9_compare_rcpp(NA_character_, NA_character_))
  expect_false(icd9_compare_rcpp("010", "010"))
  expect_false(icd9_compare_rcpp(NA_character_, "010"))
  expect_true(icd9_compare_rcpp("010", NA_character_))
})

test_that("sorting of icd10 object", {
  a <- c("Bad disease" = "I119", "Another bad disease" = "I110")
  j <- icd10cm(a)
  k <- sort(j)
  expect_identical(j[2], k[1])
  expect_identical(j[1], k[2])
  expect_identical(icd:::sort.icd10(a), unclass(k))
  expect_identical(icd:::sort.icd10cm(a), unclass(k))
})

test_that("compare icd10", {
  expect_false(icd10cm_compare_rcpp("A100", "A100"))
  expect_true(icd10cm_compare_rcpp("A100", "A200"))
  expect_false(icd10cm_compare_rcpp("A190", "A100"))
  expect_true(icd10cm_compare_rcpp("A100", NA_character_))
  expect_false(icd10cm_compare_rcpp(NA_character_, "A100"))
})

test_that("sort icd10", {
  i10 <- as.icd10cm(c("Z00", "A99", "J4C"))
  expect_equal(
    order.icd10cm(i10),
    c(2, 3, 1)
  )
  expect_equal(
    sort(i10),
    as.icd10cm(c("A99", "J4C", "Z00"))
  )
  expect_equal(sort(as.icd10cm("Z04")), as.icd10cm("Z04"))
})

test_that("duplicates are accounted for", {
  expect_identical(
    order.icd10cm(c("A10", "B20", "A10")),
    c(1L, 3L, 2L)
  )
})

test_that("codes of different lengths", {
  expect_false(icd10cm_compare_rcpp("C4A", "C430"))
  expect_true(icd10cm_compare_rcpp("C430", "C4A"))
  expect_false(icd10cm_compare_rcpp("C4A0", "C430"))
  expect_false(icd10cm_compare_rcpp("C4A0", "C43"))
})

test_that("greater than and less than need to be implemented for is.unsorted", {
  expect_true(as.icd10cm("C43") < as.icd10cm("C4A"))
  expect_true(as.icd10cm("C44") > as.icd10cm("C4A"))
  expect_true(as.icd10cm("C43") <= as.icd10cm("C4A"))
  expect_true(as.icd10cm("C44") >= as.icd10cm("C4A"))
  # equality
  expect_true(as.icd10cm("C43") == as.icd10cm("C43"))
  expect_true(as.icd10cm("C4A") == as.icd10cm("C4A"))
  expect_true(as.icd10cm("C44") == as.icd10cm("C44"))
  expect_true(as.icd10cm("C43") >= as.icd10cm("C43"))
  expect_true(as.icd10cm("C4A") >= as.icd10cm("C4A"))
  expect_true(as.icd10cm("C44") >= as.icd10cm("C44"))
  expect_true(as.icd10cm("C43") <= as.icd10cm("C43"))
  expect_true(as.icd10cm("C4A") <= as.icd10cm("C4A"))
  expect_true(as.icd10cm("C44") <= as.icd10cm("C44"))
})

test_that("compare multiple", {
  x <- as.icd10cm(rep(c("C430", "C4A0", "C440"), times = 3))
  y <- as.icd10cm(rep(c("C430", "C4A0", "C440"), each = 3))
  expect_equal(x < y, c(
    FALSE, FALSE, FALSE,
    TRUE, FALSE, FALSE,
    TRUE, TRUE, FALSE
  ))
  expect_equal(x <= y, c(
    TRUE, FALSE, FALSE,
    TRUE, TRUE, FALSE,
    TRUE, TRUE, TRUE
  ))
  expect_equal(x > y, c(
    FALSE, TRUE, TRUE,
    FALSE, FALSE, TRUE,
    FALSE, FALSE, FALSE
  ))
  expect_equal(x >= y, c(
    TRUE, TRUE, TRUE,
    FALSE, TRUE, TRUE,
    FALSE, FALSE, TRUE
  ))
})

test_that("is_unsorted", {
  x <- as.icd10cm(c("C44", "C4A", "C43"))
  expect_true(is_unsorted(x))
  expect_false(is_unsorted(sort(x)))
  expect_false(is_unsorted(x[order(x)]))
})