1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
df <- tibble::tribble(
~ id, ~group, ~letter, ~n1, ~ n2,
1, 1, NA, NA, 1,
2, 1, "a", 2, NA,
3, 1, NA, NA, NA,
4, 1, "a", NA, 4,
5, 2, "a", 5, NA,
6, 2, NA, NA, 6,
)
df_db <- copy_to(src_memdb(), df, name = "df", overwrite = TRUE)
df <- tibble::tribble(
~ id, ~group, ~n1,
1, 1, NA,
2, 1, 2,
3, 1, NA,
4, 1, NA,
5, 2, 5,
6, 2, NA,
)
df_lazy_ns <- tbl_lazy(df, con = simulate_sqlite())
df_lazy_std <- tbl_lazy(df)
test_that("fill works", {
expect_equal(
df_db %>%
window_order(id) %>%
group_by(group) %>%
tidyr::fill(n1, n2) %>%
collect(),
tibble::tribble(
~ id, ~group, ~letter, ~n1, ~ n2,
1, 1, NA, NA, 1,
2, 1, "a", 2, 1,
3, 1, NA, 2, 1,
4, 1, "a", 2, 4,
5, 2, "a", 5, NA,
6, 2, NA, 5, 6,
) %>%
group_by(group)
)
})
test_that("up-direction works", {
expect_snapshot(
df_lazy_ns %>%
window_order(id) %>%
tidyr::fill(n1, .direction = "up")
)
expect_snapshot(
df_lazy_std %>%
window_order(id) %>%
tidyr::fill(n1, .direction = "up")
)
})
test_that("up-direction works", {
expect_snapshot(
df_lazy_std %>%
window_order(id) %>%
tidyr::fill(n1, .direction = "updown")
)
expect_snapshot(
df_lazy_std %>%
window_order(id) %>%
tidyr::fill(n1, .direction = "downup")
)
})
test_that("up-direction works with descending", {
expect_snapshot(
df_lazy_ns %>%
window_order(desc(id)) %>%
tidyr::fill(n1, .direction = "up")
)
expect_snapshot(
df_lazy_std %>%
window_order(desc(id)) %>%
tidyr::fill(n1, .direction = "up")
)
})
test_that("groups are respected", {
expect_snapshot(
group_by(df_lazy_ns, group) %>%
window_order(id) %>%
tidyr::fill(n1)
)
expect_snapshot(
group_by(df_lazy_std, group) %>%
window_order(id) %>%
tidyr::fill(n1)
)
})
test_that("fill errors on unsorted data", {
expect_snapshot({
(expect_error(df_db %>% tidyr::fill(n1)))
})
})
test_that("fill() produces nice error messages", {
expect_snapshot(error = TRUE, {
lazy_frame(x = 1) %>% tidyr::fill(non_existent)
})
})
|
