tests/testthat/test-political.r

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

context("Political Sentiment Analysis")

test_that("Political API returns list of probabilities for four parties", {
  political_set <- c("Libertarian", "Liberal", "Conservative", "Green")
  test_string <- "Guns don't kill people, people kill people."
  affilation <- political(test_string, version=1)

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

test_that("Batch Political API returns list of lists of probabilities for four parties", {
  political_set <- c("Libertarian", "Liberal", "Conservative", "Green")
  test_string_arr <- c("Guns don't kill people, people kill people.", "Let's all be friends.")
  affiliation <- political(test_string_arr, version=1)
  expect_is(affiliation, "list")
  expect_is(affiliation[[1]], "list")
  expect_true(all(names(affiliation[[1]]) %in% political_set))
  expect_equal(sum(unlist(affiliation[[1]])), 1)
})

test_that("Political v2 API returns list of probabilities for four parties", {
  political_set <- c("Libertarian", "Liberal", "Conservative", "Green")
  test_string <- "Guns don't kill people, people kill people."
  affilation <- political(test_string)

  expect_is(affilation, "list")
  expect_true(all(names(affilation) %in% political_set))
  expect_equal(sum(unlist(affilation)), 1, tolerance=1e-6)
})

test_that("Batch Political v2 API returns list of lists of probabilities for four parties", {
  political_set <- c("Libertarian", "Liberal", "Conservative", "Green")
  test_string_arr <- c("Guns don't kill people, people kill people.", "Let's all be friends.")
  affiliation <- political(test_string_arr)
  expect_is(affiliation, "list")
  expect_is(affiliation[[1]], "list")
  expect_true(all(names(affiliation[[1]]) %in% political_set))
  expect_equal(sum(unlist(affiliation[[1]])), 1, tolerance=1e-6)
})

test_that("Throws error on empty text", {
  expect_error(political())
  expect_error(political(""))
  expect_error(political(" "))
  expect_error(political(version=1))
  expect_error(political("", version=1))
  expect_error(political(" ", version=1))
})