tests/testthat/test-ensemble.R

In text2sdg: Detecting UN Sustainable Development Goals in Text

# Test that detect_sdg returns an error when the input is not a character vector or a tCorpus object
test_that("detect_sdg returns an error when the input is not a character vector or a tCorpus object", {
  test_text <- 1:2
  expect_error(detect_sdg(test_text), "Argument text must be either class character or corpustools::tCorpus.")
})

# Test that detect_sdg returns an error when the synthetic argument is not one of "none", "third", "equal", or "triple"
test_that("detect_sdg returns an error when the synthetic argument is not one of 'none', 'third', 'equal', or 'triple'", {
  test_text <- c("Test text for SDG", "Test text for SDGs")
  expect_error(detect_sdg(test_text, synthetic = "invalid"), 'Argument synthetic must be one or more of "none","third","equal", or "triple".')
})

# Test that detect_sdg works when no SDGs are detected and that it returns a tibble with the correct columns
test_that("detect_sdg returns a tibble", {
  skip_on_cran()
  test_text <- c("a")
  expected_result <- tibble::tibble(
    document = factor(),
    sdg = character(),
    system = character(),
    hit = integer()
  )
  result <- detect_sdg(test_text)

  expect_equal(result, expected_result)
})

# Test that detect_sdg throws expected error with empty string as input
test_that("run detect_sdg with empty string", {
  test_text <- c("")
  expect_error(detect_sdg(test_text), "Argument text must not be an empty string.")
})


# Test that filtering SDGs works
test_that("filtering SDGs works", {
  skip_on_cran()
  test_text <- projects[c(19, 50, 83)]

  result <- detect_sdg(test_text, synthetic = c("none"), sdgs = c(3, 5))

  expect_equal(result %>%
    dplyr::distinct(sdg) %>%
    dplyr::pull(sdg), c("SDG-03", "SDG-05"))
})