tests/testthat/test-query_freq_by_polarity.R

In adityamangal410/deepSentimentR: Package to Analyze Sentiment Datasets

context("Test Query for Frequency by Polarity")

library(deepSentimentR)
library(dplyr)
library(lubridate)

test_data <- data.frame("user" = c("test_user1", "test_user1", "test_user2"),
                        "date" = c(as_datetime("2009-04-01"), as_datetime("2009-04-05"), as_datetime("2009-04-10")),
                        "text" = c("Sample tweet 1 from user1", "Sample tweet 2 from user1", "Sample tweet 1 from user2"),
                        "polarity" = c(0, 4, 4),
                        "nouns" = c(4,4,4),
                        "adjectives" = c(1,1,1),
                        "prepositions" = c(1,1,1),
                        "articles" = c(2,2,2),
                        "pronouns" = c(3,2,2),
                        "verbs" = c(1,2,4),
                        "adverbs" = c(1,3,2),
                        "interjections" = c(0,0,0),
                        "id" = c(1,2,3),
                        "query" = c("test", "test", "test"))

test_that("no filters shows all records", {
  expect_equal(dim(test_data)[1], dim(freq_by_polarity(test_data)$raw)[1])
})

test_that("user filter works", {
  expect_equal(2, dim(freq_by_polarity(test_data, user_list = c("test_user1"))$raw)[1])
})

test_that("date filter works", {
  expect_equal(2, dim(freq_by_polarity(test_data,
                                       start_date_time = as_datetime("2009-03-31"),
                                       end_date_time = as_datetime("2009-04-06"))$raw)[1])
})

test_that("keyword filter works", {
  expect_equal(2, dim(freq_by_polarity(test_data, keyword_list = c("user1"))$raw)[1])
})

test_that("all filters work", {
  expect_equal(1, dim(freq_by_polarity(test_data,
                                       user_list = c("test_user1"),
                                       start_date_time = as_datetime("2009-03-31"),
                                       end_date_time = as_datetime("2009-04-06"),
                                       keyword_list = c("tweet 1"))$raw)[1])
})


test_that("should fail for non-date as date arguments", {
  expect_error(freq_by_polarity(test_data,
                                start_date_time = "2019",
                                end_date_time = "2020"))
})

test_that("only start date is given without end date", {
  expect_error(freq_by_polarity(test_data,
                                start_date_time = as_datetime("2009-03-31")))
})

test_that("keyword_list should be a valid list", {
  expect_error(freq_by_polarity(test_data,
                                keyword_list = NULL))
})

test_that("user_list should be a valid list", {
  expect_error(freq_by_polarity(test_data,
                                user_list= NULL))
})

test_that("empty keyword list should give full data frame", {
  result <- freq_by_polarity(test_data,
                             keyword_list=list())
  expect_equal(dim(result$raw)[1], dim(test_data)[1])
})

test_that("empty user list should give full data frame", {
  result <- freq_by_polarity(test_data,
                             user_list=list())
  expect_equal(dim(result$raw)[1], dim(test_data)[1])
})

test_that("should give valid results", {
  result <- freq_by_polarity(test_data)
  expect_true(! is.null(result$plot))
  expect_true(inherits(result$plot, "ggplot"))
  expect_true(is.data.frame(result$raw))
  expect_true(is.data.frame(result$frequency))
})