@@ -16,8 +16,8 @@ test_that("step_training_window works with default n_recent", {
16
16
17
17
expect_equal(nrow(p ), 100L )
18
18
expect_equal(ncol(p ), 4L )
19
-
20
- expect_named(p , c(" time_value " , " geo_value " , " x" , " y" ))
19
+ expect_s3_class( p , " epi_df " )
20
+ expect_named(p , c(" geo_value " , " time_value " , " x" , " y" )) # order in epiprocess::new_epi_df
21
21
expect_equal(
22
22
p $ time_value ,
23
23
rep(seq(as.Date(" 2020-02-20" ), as.Date(" 2020-04-09" ), by = 1 ), times = 2 )
@@ -33,8 +33,8 @@ test_that("step_training_window works with specified n_recent", {
33
33
34
34
expect_equal(nrow(p2 ), 10L )
35
35
expect_equal(ncol(p2 ), 4L )
36
-
37
- expect_named(p2 , c(" time_value " , " geo_value " , " x" , " y" ))
36
+ expect_s3_class( p2 , " epi_df " )
37
+ expect_named(p2 , c(" geo_value " , " time_value " , " x" , " y" )) # order in epiprocess::new_epi_df
38
38
expect_equal(
39
39
p2 $ time_value ,
40
40
rep(seq(as.Date(" 2020-04-05" ), as.Date(" 2020-04-09" ), by = 1 ), times = 2 )
@@ -53,9 +53,9 @@ test_that("step_training_window does not proceed with specified new_data", {
53
53
54
54
expect_equal(nrow(p3 ), 10L )
55
55
expect_equal(ncol(p3 ), 4L )
56
-
56
+ expect_s3_class( p3 , " epi_df " )
57
57
# cols will be predictors, outcomes, time_value, geo_value
58
- expect_named(p3 , c(" x " , " y " , " time_value " , " geo_value " ))
58
+ expect_named(p3 , c(" geo_value " , " time_value " , " x " , " y " )) # order in epiprocess::new_epi_df
59
59
expect_equal(
60
60
p3 $ time_value ,
61
61
rep(seq(as.Date(" 2020-01-01" ), as.Date(" 2020-01-10" ), by = 1 ), times = 1 )
@@ -83,8 +83,8 @@ test_that("step_training_window works with multiple keys", {
83
83
84
84
expect_equal(nrow(p4 ), 12L )
85
85
expect_equal(ncol(p4 ), 5L )
86
-
87
- expect_named(p4 , c(" time_value " , " geo_value " , " additional_key" , " x" , " y" ))
86
+ expect_s3_class( p4 , " epi_df " )
87
+ expect_named(p4 , c(" geo_value " , " time_value " , " additional_key" , " x" , " y" ))
88
88
expect_equal(
89
89
p4 $ time_value ,
90
90
rep(c(
0 commit comments