tests/testthat/test-children.R

In timbr: Forest/Tree Data Frames

test_that("children", {
  library(dplyr)

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

  expect_equal_forest(
    fr,
    fr |>
      summarise() |>
      children()
  )

  fr1 <- vec_expand_grid(key1 = letters, key2 = rev(letters)) |>
    mutate(value = row_number()) |>
    forest_by(key1, key2) |>
    summarise(value = sum(value))
  fr2 <- vec_expand_grid(key3 = rev(letters), key4 = letters) |>
    mutate(value = row_number()) |>
    forest_by(key3, key4) |>
    summarise(value = sum(value))
  fr <- rbind(fr1, fr2)

  expect_snapshot(
    fr |>
      children(),
    error = TRUE
  )

  fr2_2 <- fr |>
    children("key3") |>
    summarise(value = sum(value))

  expect_equal_forest(fr2, fr2_2)
})