tests/testthat/test-season.R

In forecast: Forecasting Functions for Time Series and Linear Models

# A unit test for na.interp() and tsclean()
test_that("tests for monthdays", {
  expect_error(monthdays(rnorm(10)))
  expect_error(monthdays(rnorm(10)))
  expect_equal(
    c(monthdays(ts(rep(100, 12), frequency = 12))),
    c(31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31)
  )
  expect_equal(c(monthdays(ts(rep(1, 4), frequency = 4))), c(90, 91, 92, 92))
  # Test leap years
  expect_equal(monthdays(ts(rep(1, 48), frequency = 12))[38], 29)
  expect_equal(monthdays(ts(rep(1, 16), frequency = 4))[13], 91)
})

test_that("tests for seasonaldummy", {
  expect_error(seasonaldummy(1))
  testseries <- ts(rep(1:7, 5), frequency = 7)
  dummymat <- seasonaldummy(testseries)
  expect_length(testseries, nrow(dummymat))
  expect_shape(dummymat, ncol = 6)
  expect_lt(
    max(abs(seasonaldummy(wineind)[1:11, ] - diag(11))),
    .Machine$double.eps
  )
})

test_that("tests for seasonaldummyf", {
  expect_error(seasonaldummy(1))
  expect_warning(dummymat <- seasonaldummyf(wineind, 4), "deprecated")
  expect_shape(dummymat, dim = c(4, 11))
})

test_that("tests for fourier", {
  expect_error(fourier(1))
  testseries <- ts(rep(1:7, 5), frequency = 7)
  fouriermat <- fourier(testseries, 3)
  expect_length(testseries, nrow(fouriermat))
  expect_shape(fouriermat, ncol = 6)
  expect_all_true(grepl("-7", colnames(fouriermat)))
})

test_that("tests for fourierf", {
  expect_warning(fouriermat <- fourierf(wineind, 4, 10), "deprecated")
  expect_shape(fouriermat, dim = c(10, 8))
})

test_that("tests for stlm", {
  expect_warning(stlm(ts(rep(5, 24), frequency = 4), etsmodel = "ZZZ"))
})

test_that("tests for forecast.stlm", {
  expect_error(forecast.stlm(
    stlm(wineind),
    newxreg = matrix(rep(1, 24), ncol = 2)
  ))
  stlmfit1 <- stlm(woolyrnq, method = "ets")
  stlmfit2 <- stlm(woolyrnq, method = "arima", approximation = FALSE)
  fcfit1 <- forecast(stlmfit1)
  fcfit2 <- forecast(stlmfit1, fan = TRUE)
  expect_identical(fcfit2$level, seq(from = 51, to = 99, by = 3))
  fcstlmfit3 <- forecast(stlmfit2)
  expect_true(all(
    round(forecast(stlm(ts(rep(100, 120), frequency = 12)))$mean, 10) == 100
  ))
  expect_true(all(
    round(
      forecast(stlm(ts(rep(100, 120), frequency = 12), lambda = 1))$mean,
      10
    ) ==
      100
  ))
})

test_that("tests for stlf", {
  expect_equal(forecast(stlm(wineind))$mean, stlf(wineind)$mean)
  expect_equal(
    forecast(stlm(wineind, lambda = .5))$mean,
    stlf(wineind, lambda = .5)$mean
  )
  fit1 <- stlf(wineind, lambda = .2, biasadj = FALSE)
  fit2 <- stlf(wineind, lambda = .2, biasadj = TRUE)
  expect_false(identical(fit1$mean, fit2$mean))
  # Constant series should not error
  series <- ts(rep(950, 20), frequency = 4)
  constantForecast <- expect_no_error(stlf(series))
  # Small eps
  expect_all_true(
    c(abs(constantForecast$mean - mean(series)) < .Machine$double.eps^0.5)
  )
  y <- ts(rep(1:7, 3), frequency = 7)
  expect_equal(c(stlf(y)$mean), rep(1:7, 2))
})

test_that("tests for ma", {
  testseries <- ts(1:20, frequency = 4)
  expect_equal(frequency(ma(testseries, order = 4)), frequency(testseries))
  maseries <- ma(testseries, order = 3)
  expect_identical(which(is.na(maseries)), c(1L, 20L))
  expect_all_true(abs(maseries[2:19] - 2:19) < .Machine$double.eps^0.5)
  maseries <- ma(testseries, order = 2, centre = FALSE)
  maseries <- ma(testseries, order = 2, centre = TRUE)
  expect_identical(which(is.na(maseries)), c(1L, 20L))
  expect_all_true(abs(maseries[2:19] - 2:19) < .Machine$double.eps^0.5)
})