tests/testthat/testthat.R

In OlivierGranacher/modelStan: Helper Functions for Stan Models

library(testthat)
library(modelStan)
library(data.table)


# Test of function standardise
  context("Test of standardize")
  x1 <- c(1, 1)
  x2 <- c(-1, 1)
  x3 <- c(1, 2, NA)
  expect_equal(standardize(x1), c(0, 0))
  expect_warning(standardize(x1))
  expect_warning(standardize(x2, log = T))
  expect_equal(standardize(x3), (x3 - mean(x3, na.rm = T))/sd(x3, na.rm = T))

# Test of standardizeDT
  context("Test of standardizeDT")
  dt <- data.table::data.table(x = 1:10, y = letters[1:2])
  modelStan::standardizeDT(dt)
  expect_equal(names(dt)[3], expected = "x_std")
  expect_equal(dt[1, x_std], -1.4863011)
  dt <- data.table::data.table(x = 1:10, y = letters[1:2])
  modelStan::standardizeDT(dt, log = T)
  expect_equal(names(dt)[3], expected = "x_log_std")
  expect_equal(dt[1, x_log_std], -2.0605622)
  dt <- data.table::data.table(x = 1:10, y = letters[1:2])
  modelStan::standardizeDT(dt, cols = "x")
  expect_equal(names(dt)[3], expected = "x_std")
  expect_equal(dt[1, x_std], -1.4863011)

  # Test of simPot
  context("Test of simPotData")
  expect_equal(dim(simPotData(N = 3,
                              np = c(5, 5, 5))), c(45, 6))

  # Test of armaTimesSeries
  context("Test of armaTimeSeries")
  expect_equal(length(armaTimeSeries(N = 10, p = 2, q = 2)), 10)