tests/testthat/test-generate_cloud.R

In UBC-MDS/textprepr: Performs Pre-Processing of Tweets

test_that("generate_cloud throws an error if argument
          tweets is not a character vector", {
            expect_error(generate_cloud(c(1, 2)))
            expect_error(generate_cloud(c(TRUE, FALSE)))
            expect_error(generate_cloud(c(1L, 2L)))
            })

test_that("generate_cloud throws an error if argument
          type is not one of words, hashtag or stopwords", {
            expect_error(generate_cloud("tweets", type = "invalid"))
            expect_error(generate_cloud("tweets", type = "WORDS"))
            expect_error(generate_cloud("tweets", type = 10))
          })

test_that("generate_cloud shows the wrong proportion of words", {
  tweets <- c("Make America Great Again! @DonaldTrump #America",
              "It's rocket-science tier investment~~ #LoveElonMusk",
              "America America America always GREAT",
              "make america great again! #AMERICA")

  expected_hashtag <- data.frame(
    word = c("#america", "#loveelonmusk"),
    freq = c(2L, 1L)
  )

  expected_words <- data.frame(
    word = c("again", "always", "america", "great", "investment",
             "it's", "make", "rocket-science", "tier"),
    freq = c(2L, 1L, 5L, 3L, 1L, 1L, 2L, 1L, 1L)
  )

  expected_stopwords <- data.frame(
    word = c("always", "america", "great", "investment",
             "make", "rocket-science", "tier"),
    freq = c(1L, 5L, 3L, 1L, 2L, 1L, 1L)
  )

  actual_hashtag <- generate_cloud(tweets, type = "hashtag")
  actual_words <- generate_cloud(tweets)
  actual_stopwords <- generate_cloud(tweets, type = "stopwords")

  expect_identical(actual_hashtag, expected_hashtag)
  expect_identical(actual_words, expected_words)
  expect_identical(actual_stopwords, expected_stopwords)
})