@@ -36,19 +36,21 @@ test_that("epi_recipe formula works", {
36
36
r <- epi_recipe(y ~ x , tib )
37
37
ref_var_info <- tibble :: tribble(
38
38
~ variable , ~ type , ~ role , ~ source ,
39
- " x" , " numeric" , " predictor" , " original" ,
40
- " y" , " numeric" , " outcome" , " original" ,
39
+ " x" , c( " integer " , " numeric" ) , " predictor" , " original" ,
40
+ " y" , c( " integer " , " numeric" ) , " outcome" , " original" ,
41
41
" time_value" , " date" , " time_value" , " original" ,
42
- " geo_value" , " nominal" , " geo_value" , " original"
42
+ " geo_value" , c( " string " , " unordered " , " nominal" ) , " geo_value" , " original"
43
43
)
44
44
expect_identical(r $ var_info , ref_var_info )
45
45
expect_equal(nrow(r $ template ), 1L )
46
46
47
47
# with an epi_key as a predictor
48
48
r <- epi_recipe(y ~ x + geo_value , tib )
49
- ref_var_info <- ref_var_info %> % tibble :: add_row(
50
- variable = " geo_value" , type = " nominal" , role = " predictor" ,
51
- source = " original" , .after = 1 )
49
+ ref_var_info <- ref_var_info %> %
50
+ tibble :: add_row(
51
+ variable = " geo_value" , type = list (c(" string" , " unordered" , " nominal" )),
52
+ role = " predictor" ,
53
+ source = " original" , .after = 1 )
52
54
expect_identical(r $ var_info , ref_var_info )
53
55
expect_equal(nrow(r $ template ), 1L )
54
56
@@ -61,11 +63,13 @@ test_that("epi_recipe formula works", {
61
63
62
64
# with an additional key
63
65
r <- epi_recipe(y ~ x + geo_value , tib )
64
- ref_var_info <- ref_var_info %> % tibble :: add_row(
65
- variable = " z" , type = " nominal" , role = " key" ,
66
- source = " original" )
66
+ ref_var_info <- ref_var_info %> %
67
+ tibble :: add_row(
68
+ variable = " z" , type = list (c(" string" , " unordered" , " nominal" )),
69
+ role = " key" ,
70
+ source = " original" )
67
71
68
- # expect_identical(r$var_info, ref_var_info)
72
+ expect_identical(r $ var_info , ref_var_info )
69
73
70
74
})
71
75
@@ -81,20 +85,20 @@ test_that("epi_recipe epi_df works", {
81
85
ref_var_info <- tibble :: tribble(
82
86
~ variable , ~ type , ~ role , ~ source ,
83
87
" time_value" , " date" , " time_value" , " original" ,
84
- " geo_value" , " nominal" , " geo_value" , " original" ,
85
- " x" , " numeric" , " raw" , " original" ,
86
- " y" , " numeric" , " raw" , " original"
88
+ " geo_value" , c( " string " , " unordered " , " nominal" ) , " geo_value" , " original" ,
89
+ " x" , c( " integer " , " numeric" ) , " raw" , " original" ,
90
+ " y" , c( " integer " , " numeric" ) , " raw" , " original"
87
91
)
88
92
expect_identical(r $ var_info , ref_var_info )
89
93
expect_equal(nrow(r $ template ), 1L )
90
94
91
95
r <- epi_recipe(tib , formula = y ~ x )
92
96
ref_var_info <- tibble :: tribble(
93
97
~ variable , ~ type , ~ role , ~ source ,
94
- " x" , " numeric" , " predictor" , " original" ,
95
- " y" , " numeric" , " outcome" , " original" ,
98
+ " x" , c( " integer " , " numeric" ) , " predictor" , " original" ,
99
+ " y" , c( " integer " , " numeric" ) , " outcome" , " original" ,
96
100
" time_value" , " date" , " time_value" , " original" ,
97
- " geo_value" , " nominal" , " geo_value" , " original"
101
+ " geo_value" , c( " string " , " unordered " , " nominal" ) , " geo_value" , " original"
98
102
)
99
103
expect_identical(r $ var_info , ref_var_info )
100
104
expect_equal(nrow(r $ template ), 1L )
@@ -106,7 +110,7 @@ test_that("epi_recipe epi_df works", {
106
110
)
107
111
ref_var_info <- ref_var_info %> %
108
112
tibble :: add_row(
109
- variable = " time_value" , type = " date" , role = " funny_business" ,
113
+ variable = " time_value" , type = list ( " date" ) , role = " funny_business" ,
110
114
source = " original"
111
115
)
112
116
expect_identical(r $ var_info , ref_var_info )
0 commit comments