tests/testthat/test-language.r

In IndicoDataSolutions/IndicoIo-R: A simple R Wrapper for the indico set of APIs - http://indico.io

context("Language Detection")

language_set <- c(
  "English",
  "Spanish",
  "Tagalog",
  "Esperanto",
  "French",
  "Chinese",
  "French",
  "Bulgarian",
  "Latin",
  "Slovak",
  "Hebrew",
  "Russian",
  "German",
  "Japanese",
  "Korean",
  "Portuguese",
  "Italian",
  "Polish",
  "Turkish",
  "Dutch",
  "Arabic",
  "Persian (Farsi)",
  "Czech",
  "Swedish",
  "Indonesian",
  "Vietnamese",
  "Romanian",
  "Greek",
  "Danish",
  "Hungarian",
  "Thai",
  "Finnish",
  "Norwegian",
  "Lithuanian"
)

test_that("Language API returns list of probabilities for the list of languages", {
  test_string <- "Clearly an English sentence."
  languages <- language(test_string)

  expect_is(languages, "list")
  expect_true(all(names(languages) %in% language_set))
  expect_equal(sum(unlist(languages)), 1)
})

test_that("Throws error on empty text", {
  expect_error(language())
  expect_error(language(""))
  expect_error(language(" "))
})

test_that("Batch Language API returns list of list of probabilities for the list of languages", {
  test_string_arr <- c("Clearly an English sentence.", "Clearly not an English sentence.")
  languages <- language(test_string_arr)

  expect_is(languages, "list")
  expect_is(languages[[1]], "list")
  expect_true(all(names(languages[[1]]) %in% language_set))
  expect_equal(sum(unlist(languages[[1]])), 1)
})