Merge pull request #21 from kenmawer/km-issue_27

Km issue 27 tests

Author: kenmawer

musings/example-recipe.R

@@ -40,7 +40,7 @@ r <- epi_recipe(x) %>% # if we add this as a class, maybe we get better
   step_epi_ahead(death_rate, ahead = 7) %>%
   step_epi_lag(death_rate, lag = c(0, 7, 14)) %>%
   step_epi_lag(case_rate, lag = c(0, 7, 14)) %>%
-  step_narm()
+  step_naomit2()
 
 
 slm <- linear_reg()

tests/testthat/test-step_naomit2.R

@@ -0,0 +1,33 @@
+library(dplyr)
+library(epiprocess)
+library(parsnip)
+library(workflows)
+
+# Random generated dataset
+set.seed(100)
+x <- tibble(geo_value = rep("nowhere",200),
+            time_value = as.Date("2021-01-01") + 0:199,
+            case_rate = rpois(100,20) + 1:200,
+            death_rate = rpois(100,10) + 1:200) %>%
+  as_epi_df()
+
+r <- epi_recipe(x) %>%
+  step_epi_ahead(death_rate, ahead = 7) %>%
+  step_epi_lag(death_rate, lag = c(0,7,14))
+
+extract <- function(recipe) {
+  recipe
+}
+
+z1 <- step_naomit2(r)
+z2 <- r %>%
+  step_naomit(all_predictors()) %>%
+  step_naomit(all_outcomes(), skip = TRUE)
+
+# Tests
+test_that("Check that both functions behave the same way", {
+  expect_identical(z1$steps[[3]][-1][-5],z2$steps[[3]][-1][-5])
+  expect_identical(z1$steps[[4]][-1][-5],z2$steps[[4]][-1][-5])
+  expect_identical(class(z1$steps[[3]]),class(z2$steps[[3]]))
+  expect_identical(class(z1$steps[[4]]),class(z2$steps[[4]]))
+})