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
|
context("compute_stack")
test_that("compute_stack works as expected", {
dat <- data.frame(
g1 = c("a", "a", "a", "b", "b", "c", "c", "c"),
g2 = c(1, 2, 3, 1, 3, 1, 2, 3),
val = 1:8
)
# Basic stacking: stack y value at each x
stacked <- dat %>% compute_stack(~val, ~g1) %>% as.data.frame()
expected <- data.frame(
stack_lwr_ = c(0,1,3,0,4,0,6,13),
stack_upr_ = c(1,3,6,4,9,6,13,21)
)
expect_equal(stacked[c("stack_lwr_", "stack_upr_")], expected)
# With grouping
stacked <- dat %>% group_by(g2) %>% compute_stack(~val, ~g1) %>%
as.data.frame()
expected <- expected[order(dat$g2), ] %>% as.data.frame()
expected <- expected[order(dat$g2), ] %>% as.data.frame()
expect_equal(stacked[c("stack_lwr_", "stack_upr_")], expected)
})
test_that("Zero-row inputs", {
res <- mtcars[0,] %>% compute_stack(~wt, ~cyl)
expect_equal(nrow(res), 0)
expect_true(setequal(
names(res),
c(names(mtcars), "group__", "stack_upr_", "stack_lwr_"))
)
# Grouped
res <- mtcars %>% group_by(cyl) %>% dplyr::filter(FALSE) %>% compute_stack(~wt, ~cyl)
expect_equal(nrow(res), 0)
expect_true(setequal(
names(res),
c(names(mtcars), "group__", "stack_upr_", "stack_lwr_"))
)
})
|
