tests/testthat/test-interpolate.R

In fabletools: Core Tools for Packages in the 'fable' Framework

context("test-interpolate.R")

test_that("Test interpolation", {
  skip_if_not_installed("fable")
  NA_pos <- c(0, sample(seq_len(NROW(us_deaths)), 10), NROW(us_deaths))
  us_deaths$value[NA_pos] <- NA
  mbl_miss <- us_deaths %>%
    model(fable::TSLM(value ~ trend() + season()))
  interpolated <- interpolate(mbl_miss, us_deaths)
  expect_true(all(!is.na(interpolated$value)))
})