test_that("cur_group() works", {
df <- tibble(g = 1, x = 1)
gf <- group_by(df, g)
expect_equal(
df %>% summarise(key = list(cur_group())) %>% pull(key),
list(tibble(.rows = 1L))
)
expect_equal(
gf %>% summarise(key = list(cur_group())) %>% pull(key),
list(tibble(g = 1))
)
})
test_that("cur_group_idx() gives unique id", {
df <- tibble(x = c("b", "a", "b"))
gf <- group_by(df, x)
expect_equal(
summarise(gf, id = cur_group_id()),
tibble(x = c("a", "b"), id = 1:2)
)
expect_equal(
mutate(gf, id = cur_group_id()),
group_by(tibble(x = df$x, id = c(2, 1, 2)), x)
)
})
test_that("cur_data() gives current data without groups, cur_data_all() includes groups", {
df <- tibble(x = c("b", "a", "b"), y = 1:3)
gf <- group_by(df, x)
expect_equal(
df %>% summarise(x = list(cur_data())) %>% pull(),
list(df)
)
expect_equal(
gf %>% summarise(x = list(cur_data())) %>% pull(),
list(tibble(y = 2L), tibble(y = c(1L, 3L)))
)
expect_equal(
gf %>% summarise(x = list(cur_data_all())) %>% pull(),
list(tibble(x = "a", y = 2L), tibble(x = "b", y = c(1L, 3L)))
)
})
test_that("cur_data()/cur_data_all() keeps list columns as lists in rowwise_df (#5901)", {
df <- tibble(x = list(tibble(a = 1), tibble(a = 2))) %>%
rowwise()
expect_true(
all(summarise(df, test = vec_is_list(cur_data()$x))$test)
)
expect_true(
all(summarise(df, test = vec_is_list(cur_data_all()$x))$test)
)
})
test_that("cur_group_rows() retrieves row position in original data", {
df <- tibble(x = c("b", "a", "b"), y = 1:3)
gf <- group_by(df, x)
expect_equal(
df %>% summarise(x = list(cur_group_rows())) %>% pull(),
list(1:3)
)
expect_equal(
gf %>% summarise(x = list(cur_group_rows())) %>% pull(),
list(2L, c(1L, 3L))
)
})
test_that("cur_data() and cur_data_all() work sequentially", {
df <- tibble(a = 1)
expect_equal(
mutate(df, x = ncol(cur_data()), y = ncol(cur_data())),
tibble(a = 1, x = 1, y = 2)
)
gf <- tibble(a = 1, b = 2) %>% group_by(a)
expect_equal(
mutate(gf, x = ncol(cur_data_all()), y = ncol(cur_data_all())),
group_by(tibble(a = 1, b = 2, x = 2, y = 3), a)
)
})
test_that("give useful error messages when not applicable", {
expect_snapshot({
(expect_error(n()))
(expect_error(cur_data()))
(expect_error(cur_data_all()))
(expect_error(cur_column()))
(expect_error(cur_group()))
(expect_error(cur_group_id()))
(expect_error(cur_group_rows()))
})
})
Generated by dwww version 1.15 on Sat May 18 12:11:18 CEST 2024.