tests/testthat/test-str_extractors.R

In atorus-research/Tplyr: A Traceability Focused Grammar of Clinical Data Summary

string <- c(" 0  (0.0%)", " 8  (9.3%)", "78 (90.7%)")

test_that("String extractor errors generate properly", {
  expect_error(
    str_extract_fmt_group(c(1), 1),
    "Paramter `string`"
  )

  expect_error(
    str_extract_fmt_group(string, "hi"),
    "Paramter `format_group`"
  )

  expect_error(
    str_extract_num(c(1), 1),
    "Paramter `string`"
  )

  expect_error(
    str_extract_num(string, "hi"),
    "Paramter `format_group`"
  )
})

test_that("Format groups can be extracted", {
  expect_equal(
    str_extract_fmt_group(string, 1),
    c(' 0', ' 8', '78')
  )

  expect_equal(
    str_extract_fmt_group(string, 2),
    c("(0.0%)", "(9.3%)", "(90.7%)")
  )

  expect_equal(
    str_extract_fmt_group(string, 3),
    rep(NA_character_, 3)
  )
})

test_that("Numbers from format groups can be extracted", {
  expect_equal(
    str_extract_num(string, 1),
    c(0, 8, 78)
  )

  expect_equal(
    str_extract_num(string, 2),
    c(0.0, 9.3, 90.7)
  )

  expect_equal(
    str_extract_num(string, 3),
    rep(NA_real_, 3)
  )
})