tests/testthat/test-checks.R

In keyATM: Keyword Assisted Topic Models

test_that("get_doc_index", {
  # Single
  W_raw <- list(d1 = c("a", "b", "c"), d2 = c())
  expect_warning(
    tmp <- keyATM:::get_doc_index(W_raw, check = TRUE),
    "Index to check: 2"
  )
  expect_warning(
    tmp <- keyATM:::get_doc_index(W_raw, check = FALSE),
    "Index to check: 2"
  )

  # Multiple
  W_raw <- list(d1 = c("a", "b", "c"), d2 = c(), d3 = c())
  expect_warning(
    tmp <- keyATM:::get_doc_index(W_raw, check = TRUE),
    "Indexes to check: 2 and 3"
  )
  expect_warning(
    tmp <- keyATM:::get_doc_index(W_raw, check = FALSE),
    "Indexes to check: 2 and 3"
  )
})

test_that("check_keuywords", {
  keywords <- list(t1 = c("a", "b", "c"), t2 = c("d", "e"))

  unique_words <- c("a", "b", "c", "d")
  expect_warning(
    tmp <- keyATM:::check_keywords(unique_words, keywords, prune = TRUE),
    "A keyword is pruned"
  )
  expect_error(
    tmp <- keyATM:::check_keywords(unique_words, keywords, prune = FALSE),
    "A keyword is not found"
  )

  unique_words <- c("a", "b", "d")
  expect_warning(
    tmp <- keyATM:::check_keywords(unique_words, keywords, prune = TRUE),
    "Keywords are pruned"
  )
  expect_error(
    tmp <- keyATM:::check_keywords(unique_words, keywords, prune = FALSE),
    "Keywords are not found"
  )
})