tests/testthat/test_tfidf.R

In jmotif: Time Series Analysis Toolkit Based on Symbolic Aggregate Discretization, i.e. SAX

bag1 <- data.frame(
  "words" = c("this", "is", "a", "sample"),
  "counts" = c(1, 1, 2, 1),
  stringsAsFactors = FALSE
)

bag2 <- data.frame(
  "words" = c("this", "is", "another", "example"),
  "counts" = c(1, 1, 2, 3),
  stringsAsFactors = FALSE
)

ll <- list("1" = bag1, "2" = bag2)

tfidf <- bags_to_tfidf(ll)

test_that("test tfidf #1", {

  example <- log(1 + 3) * log(2 / 1)
  expect_equal(example, tfidf[tfidf$words == "example", 3])

  a <- log(1 + 2) * log(2 / 1)
  expect_equal(a, tfidf[tfidf$words == "a", 2])

})