feat(croston): print the demand and interval

Author: m-muecke

R/croston.R

@@ -144,6 +144,8 @@ croston_model <- function(
     alpha = alpha,
     type = type,
     init = init,
+    init_demand = init_demand,
+    init_interval = init_interval,
     y = y,
     fit_demand = res$fit_demand,
     fit_interval = res$fit_interval,
@@ -233,6 +235,14 @@ print.croston_model <- function(
   cat("Call:", deparse(x$call), "\n\n")
   cat("alpha:", format(x$alpha, digits = digits), "\n")
   cat("method:", x$type, "\n")
+  cat(
+    "initial:",
+    "demand =",
+    format(x$init_demand, digits = digits),
+    ", interval =",
+    format(x$init_interval, digits = digits),
+    "\n"
+  )
   invisible(x)
 }
 

tests/testthat/test-croston.R

@@ -55,6 +55,16 @@ test_that("croston_model with numeric init and optimized alpha", {
   expect_equal(fit$fit_interval[1], 2)
 })
 
+test_that("croston_model stores resolved initial values", {
+  fit <- croston_model(y, init = c(3, 2))
+  expect_equal(fit$init_demand, 3)
+  expect_equal(fit$init_interval, 2)
+
+  fit_opt <- croston_model(y, alpha = NULL, init = "mean")
+  expect_equal(fit_opt$init_demand, fit_opt$fit_demand[1])
+  expect_equal(fit_opt$init_interval, fit_opt$fit_interval[1])
+})
+
 test_that("croston_model errors on invalid init", {
   expect_error(croston_model(y, init = c(3)), "length 2")
   expect_error(croston_model(y, init = c(-1, 2)), "non-negative")