tests/testthat/test-leaves.R

In UchidaMizuki/timbr: Forest/Tree Data Frames

test_that("leaves", {
  library(dplyr)

  fr1 <- vec_expand_grid(key1 = rev(letters[1:5]),
                         key2 = letters[1:5]) %>%
    mutate(value = row_number()) %>%
    forest_by(key1, key2)

  expect_equal_forest(leaves(fr1), fr1)

  fr2 <- vec_expand_grid(key1 = letters[1:5],
                         key2 = rev(letters[1:5]),
                         key3 = letters[1:5]) %>%
    mutate(value = row_number()) %>%
    forest_by(key1, key2, key3)

  fr2_summarised <- fr2 |>
    summarise()

  fr3 <- rbind(fr1, fr2_summarised) |>
    summarise() |>
    leaves()

  expect_equal(fr3 |>
                 climb(key2) |>
                 as_tibble() |>
                 ungroup(),
               fr1 |>
                 as_tibble() |>
                 ungroup() |>
                 select(!key1))

  expect_equal(fr3 |>
                 climb(key3) |>
                 as_tibble() |>
                 ungroup(),
               fr2 |>
                 as_tibble() |>
                 ungroup() |>
                 select(!c(key1, key2)))
})