tests/testthat/test-grouping.R

In unnest: Unnest Hierarchical Data Structures

context("Grouping")

test_that("Grouped stacking of unnamed levels works", {

  expect_equal(unnestl(xx, dedupe = T, s(stack = T,
                             groups = list(one = s("a", s("b", s("c"))),
                                           two = s("a", s("b", s("c")))))),
               list(one = list(a.b.c.1.a = c(1, 1), a.b.c.1.b = c(1, 1),
                               a.b.c.2.a = c(2, 2), a.b.c.2.c = c(2, 2),
                               a.b.c.3.a = c(3, 3), a.b.c.3.b = c(3, 3)),
                    two = structure(list(), .Names = character(0))))

  expect_equal(unnestl(xx, dedupe = T,
                       s(stack = T,
                         groups = list(one = s("a", s("b", s("c", s(2)))),
                                       two = s("a", s("b", s("c")))))),
               list(one = list(a.b.c.2.a = c(2, 2), a.b.c.2.c = c(2, 2)),
                    two = list(a.b.c.1.a = c(1, 1), a.b.c.1.b = c(1, 1),
                               a.b.c.3.a = c(3, 3), a.b.c.3.b = c(3, 3))))

  expect_equal(unnest(x, s("a",
                           s("b",
                             s("e",
                               s("f", s(stack = T))),
                             s("c", s(stack = T))))),
               structure(list(a.b.c.a = c(1, 2, 3, 1, 2, 3),
                              a.b.c.b = c(1, NA, 3, 1, NA, 3),
                              a.b.c.c = c(NA, 2, NA, NA, 2, NA),
                              a.b.e.f = c(1L, 2L, 1L, 2L, 1L, 2L)),
                         class = "data.frame", row.names = c(NA, 6L)))


})


test_that("Grouped stacking is de-duped correctly", {

  # s(exclude = "c") is a leaf node, so "a.b.d" is considered matched here
  expect_equal(unnest(xx, dedupe = T,
                      s(stack = T,
                        groups = list(one = list(s("a", s("b", s("d"))),
                                                 s("a", s("id"))),
                                      two = s("a",
                                              s("id"),
                                              s("b", s(exclude = "c")))))),
               list(one = unnest(xx, s(stack = T,
                                       s("a", s("b", s("d"))),
                                       s("a", s("id")))),
                    two = unnest(xx, s(stack = T,
                                       s("a",
                                         s("id"),
                                         s("b", s(exclude = c("c"))))))))

  expect_equal(unnest(xx, dedupe = T,
                      s(stack = T, # as = ".ix.",
                        groups = list(one = list(s("a", s("b", s("d"))),
                                                 s("a", s("id"))),
                                      two = s("a"))))$two,
               unnest(xx, s(stack = T,
                            s("a",
                              s("b", s("e"), s("c"))))))

  ## stacked node is not the same as spread node
  expect_equal(unnest(xx, dedupe = T,
                      s(stack = T, # as = ".ix.",
                        groups = list(one = list(s("a", s("b", s("d", stack = T))),
                                                 s("a", s("id"))),
                                      two = s("a"))))$two,
               unnest(xx, s(stack = T, s("a", s(exclude = "id")))))

  expect_equal(unnest(xx, dedupe = T,
                      s(stack = T, # as = ".ix.",
                        groups = list(one = list(s("a", s("b", s("d"))),
                                                 s("a", s("id"))),
                                      two = list(s("a"),
                                                 s("a/id"),
                                                 s("a/b/d")))))$two,
               unnest(xx, s(stack = T,
                            s("a"))))

})

# ## sigfault
# unnest(xx, s(stack = T, dedupe = T, as = ".ix.",
#              groups = list(one = list(s("a", s("b", s("d"))),
#                                       s("a", s("id"))),
#                            two = s("a"))))