tests/testthat/test-word_cor_network.R

In adityamangal410/deepSentimentR: Package to Analyze Sentiment Datasets

context("Test Word Correlation Network")

library(deepSentimentR)
library(dplyr)
library(lubridate)
data("sentiment140_test")

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(sentiment140_test)[1], dim(word_cor_network(sentiment140_test)$raw)[1])
})

test_that("all filters work", {
  expect_gte(dim(word_cor_network(sentiment140_test,
                                  start_date_time = as_datetime("2009-04-01"),
                                  end_date_time = as_datetime("2009-06-14"))$raw)[1], 1)
})


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

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

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

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

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

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

test_that("should give valid results", {
  result <- word_cor_network(sentiment140_test)
  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$word_cors))
})