tests/testthat/test-04-LDA_TS.R

In LDATS: Latent Dirichlet Allocation Coupled with Time Series Analyses

context("Check LDA_TS functions")

data(rodents)
lda_data <- rodents$document_term_table
document_covariate_table <- rodents$document_covariate_table
  mod0 <- LDA_TS(rodents,
                 topics = 2, nseeds = 1, formulas = ~ 1, nchangepoints = 0,
                 timename = "newmoon",
                 control = list(nit = 10))
  mod1 <- LDA_TS(rodents,
                 topics = 2, nseeds = 1, formulas = ~ 1, nchangepoints = 1,
                 timename = "newmoon",
                 control = list(nit = 50))


test_that("LDA_TS on 0 changepoints", {
  expect_is(mod0, "LDA_TS")
  expect_equal(length(names(mod0)), 4)
  expect_is(mod0[[4]], "TS_fit")
})

test_that("LDA_TS on 1 changepoints", {

  expect_is(mod1, "LDA_TS")
  expect_equal(length(names(mod1)), 4)
  expect_is(mod1[[4]], "TS_fit")
})

test_that("check print on LDA_TS", {
  expect_output(print(mod1))
})

test_that("Check LDA_TS_controls_list", {
  expect_is(LDA_TS_control(), "list")
  expect_equal(length(LDA_TS_control()), 3)
})

test_that("Check conform_LDA_TS_data", {
  expect_is(conform_LDA_TS_data(rodents), "list")
  expect_is(conform_LDA_TS_data(rodents[[1]]), "list")
  expect_message(conform_LDA_TS_data(rodents[[1]]))
  expect_error(conform_LDA_TS_data(list(term1 = 1, term2 = 2)))

  expect_is(conform_LDA_TS_data(list(term = rodents[[1]])), "list")
  expect_message(conform_LDA_TS_data(list(term = rodents[[1]])))

  expect_error(conform_LDA_TS_data(list(term = rodents[[1]], covariate1 = 1,
                                        covariate2 = 2)))
  expect_error(conform_LDA_TS_data("ok"))
})

test_that("Check package_LDA_TS", {
  topics <- 2
  nseeds <- 1
  formulas <- ~ 1
  nchangepoints <- 1
  weights <- document_weights(lda_data)
  timename <-  "newmoon"
  control <- LDA_TS_control(nit = 50)
  LDAs <- LDA_set(lda_data, topics, nseeds, 
                  control$LDA_set_control)
  sel_LDA <- select_LDA(LDAs, control$LDA_set_control)
  TSs <- TS_on_LDA(sel_LDA, document_covariate_table, formulas, nchangepoints,
                   timename, weights, control$TS_control)
  sel_TSs <- select_TS(TSs, control$TS_control)
  
  expect_is(package_LDA_TS(LDAs, sel_LDA, TSs, sel_TSs), "LDA_TS")
  expect_equal(length(package_LDA_TS(LDAs, sel_LDA, TSs, sel_TSs)), 4)
  expect_error(package_LDA_TS())
  expect_error(package_LDA_TS("ok", sel_LDA, TSs, sel_TSs))
  expect_error(package_LDA_TS(LDAs, "ok", TSs, sel_TSs))
  expect_error(package_LDA_TS(LDAs, sel_LDA, "ok", sel_TSs))
  expect_error(package_LDA_TS(LDAs, sel_LDA, TSs, "ok"))
})