tests/testthat/test-leaves.R

In 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))
  )
})