saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-05-23 22:13:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add unit tests for disabling uncertainty estimation in R

Browse source
This commit is contained in:
Ben Letham 2020-02-04 13:05:28 -08:00
parent 6e51130f28
commit 8d0f23f8be
2 changed files with 22 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
R/tests/testthat/test_diagnostics.R
View file

@ -90,6 +90,18 @@ test_that("cross_validation_default_value_check", {
expect_equal(sum(dplyr::select(df.cv1 - df.cv2, y, yhat)), 0)
})
test_that("cross_validation_uncertainty_disabled", {
skip_if_not(Sys.getenv('R_ARCH') != '/i386')
for (uncertainty in c(0, FALSE)) {
m <- prophet(uncertainty.samples = uncertainty)
m <- fit.prophet(m = m, df = DATA, algorithm = "Newton")
df.cv <- cross_validation(
m, horizon = 4, units = "days", period = 4, initial = 115)
expected.cols <- c('y', 'ds', 'yhat', 'cutoff')
expect_equal(expected.cols, colnames(df.cv))
}
})
test_that("performance_metrics", {
skip_if_not(Sys.getenv('R_ARCH') != '/i386')
m <- prophet(DATA)

10
R/tests/testthat/test_prophet.R
View file

@ -73,6 +73,16 @@ test_that("fit_predict_constant_history", {
expect_equal(tail(fcst$yhat, 1), 0)
})
test_that("fit_predict_uncertainty_disabled", {
skip_if_not(Sys.getenv('R_ARCH') != '/i386')
for (uncertainty in c(0, FALSE)) {
m <- prophet(train, uncertainty.samples = uncertainty)
fcst <- predict(m, future)
expected.cols <- c('ds', 'trend', 'additive_terms', 'weekly', 'multiplicative_terms', 'yhat')
expect_equal(expected.cols, colnames(fcst))
}
})
test_that("setup_dataframe", {
history <- train
m <- prophet(history, fit = FALSE)