tests/testthat/test-dplyr.R

In UchidaMizuki/timbr: Forest/Tree Data Frames

test_that("dplyr", {
  library(dplyr)

  df <- vec_expand_grid(key1 = letters[1:2],
                        key2 = letters[1:2],
                        key3 = letters[1:2]) %>%
    mutate(value = row_number())
  fr <- df %>%
    forest_by(key1, key2, key3)
  expect_s3_class(fr, "timbr_forest")

  df_sum <- df %>%
    group_by(key1, key2) %>%
    summarise(value = sum(value),
              .groups = "drop")

  fr_sum <- fr %>%
    summarise(value = sum(value))

  expect_equal(ungroup(as_tibble(fr_sum)), df_sum)
  expect_equal_forest(children(fr_sum), fr)

  fr_sum1 <- fr_sum %>%
    mutate(value1 = value + 1)

  loc <- fr_sum1$roots$.
  expect_equal(vec_slice(get_nodes(fr_sum1)$value1, loc),
               vec_slice(get_nodes(fr_sum)$value + 1, loc))
  expect_true(all(is.na(vec_slice(get_nodes(fr_sum1)$value1,
                                  vec_as_location(-loc, vec_size(get_nodes(fr_sum1)))))))
  expect_equal_forest(fr_sum,
                      fr_sum1 %>%
                        select(!value1))

  df <- fr %>%
    as_tibble() %>%
    mutate(value = value * 2)

  expect_equal_forest(fr_sum %>%
                        rows_update(df) %>%
                        children(),
                      fr %>%
                        mutate(value = value * 2))
  expect_equal_forest(fr_sum %>%
                        rows_patch(df) %>%
                        children(),
                      fr)
})

test_that("rows_update", {
  set.seed(1234)

  library(dplyr)

  fr <- vec_expand_grid(key1 = letters[1:3],
                        key2 = letters[1:3]) %>%
    mutate(value = row_number()) %>%
    forest_by(key1, key2) %>%
    summarise(value = sum(value))

  df <- vec_expand_grid(key1 = c("c", "a"),
                        key2 = c("b", "a", "c")) %>%
    mutate(value = row_number())

  expect_no_error(fr %>%
                    rows_update(df,
                                by = c("key1", "key2")) %>%
                    children())
  expect_no_error(fr %>%
                    rows_patch(df,
                                by = c("key1", "key2")) %>%
                    children())
  expect_error(fr %>%
                 rows_update(df,
                             by = c("key2", "key1")) %>%
                 children())
  expect_error(fr %>%
                 rows_patch(df,
                             by = c("key2", "key1")) %>%
                 children())

  df <- vec_expand_grid(key1 = c("c", "a"),
                        key2 = c("b", "a", "x")) %>%
    mutate(value = row_number())
  expect_error(fr %>%
                 rows_update(df,
                             by = c("key1", "key2")) %>%
                 children())
  expect_error(fr %>%
                 rows_patch(df,
                             by = c("key1", "key2")) %>%
                 children())

  fr <- vec_expand_grid(key1 = letters[1:3],
                        key2 = letters[1:3],
                        key3 = letters[1:3]) %>%
    mutate(value = row_number()) %>%
    forest_by(key1, key2, key3) %>%
    summarise(value = sum(value))
  df <- vec_expand_grid(key2 = c("c", "a"),
                        key3 = c("a", "b", "c")) %>%
    mutate(value = sample(1:9, n()))
  fr <- fr %>%
    rows_update(df,
                by = c("key2", "key3")) %>%
    climb(key2, key3) %>%
    as_tibble() %>%
    rename(value_object = value) %>%
    inner_join(df %>%
                 rename(value_expected = value),
               by = join_by(key2, key3))

  expect_equal(fr$value_object, fr$value_expected)
})