tests/testthat/test-bundle.R

In yutannihilation/tiedr: Experimental Package For Multi-Gather And Multi-Spread

context("test-bundle")

tb <- tibble::tibble

test_that("relocate_bundled_cols() works", {
  expect_equal(relocate_bundled_cols(c("x", "a1", "y", "a2"), a = c("a1", "a2")), c("x", "a", "y"))
  expect_equal(relocate_bundled_cols(c("x", "a1", "y", "a2"), !!!list(a = c("a1", "a2"))), c("x", "a", "y"))

  expect_equal(relocate_bundled_cols(c("x", "a1", "b1", "y", "b2", "a2"), a = c("a1", "a2"), b = c("b1", "b2")), c("x", "a", "b", "y"))
})

test_that("bundle() for simple cases works", {
  d <- data.frame(x = 1:3, a1 = 11:13, y = 1:3, a2 = 21:23)

  expected <- d[, c("x", "y"), drop = FALSE]
  expected$data <- d[, c("a1", "a2")]
  expected <- expected[, c("x", "data", "y")]

  # unnamed cases
  expect_equal(bundle(d, a1, a2), expected)
  expect_equal(bundle(d, c("a1", "a2")), expected)
  expect_equal(bundle(d, -x, -y), expected)
  expect_equal(bundle(d, starts_with("a")), expected)

  # named cases
  names(expected) <- c("x", "foo", "y")

  expect_equal(bundle(d, foo = c(a1, a2)), expected)
  expect_equal(bundle(d, foo = c(-x, -y)), expected)
  expect_equal(bundle(d, foo = starts_with("a")), expected)

  # unnamed, but specify .key
  expect_equal(bundle(d, a1, a2, .key = "foo"), expected)
})

test_that("relocate_unbundled_cols() works", {
  expect_equal(relocate_unbundled_cols(c("x", "a", "y"), a = c("a1", "a2")), c("x", "a1", "a2", "y"))
  expect_equal(relocate_unbundled_cols(c("x", "a", "y"), !!!list(a = c("a1", "a2"))), c("x", "a1", "a2", "y"))

  expect_equal(relocate_unbundled_cols(c("x", "b", "y", "a"), a = c("a1", "a2"), b = c("b1", "b2")), c("x", "b1", "b2", "y", "a1", "a2"))
})

test_that("unbundle() works", {
  d <- tb(id = 1:3, a = tb(value = 1:3, type = c("a", "b", "c")))
  expect_equal(unbundle(d, a), tb(id = 1:3, a_value = 1:3, a_type = c("a", "b", "c")))
  expect_equal(unbundle(d, a, sep = "."), tb(id = 1:3, a.value = 1:3, a.type = c("a", "b", "c")))
  expect_equal(unbundle(d, a, sep = NULL), tb(id = 1:3, value = 1:3, type = c("a", "b", "c")))

  d <- tb(id = 1:3,
          A = tb(a = tb(value = 1:3, type = c("a", "b", "c")), b = tb(value = 4:6, type = c("a", "b", "c"))),
          B = tb(a = tb(value = 1:3, type = c("x", "y", "z"))))
  d1 <- unbundle(d, A:B)
  expect_equal(d1$id, 1:3)
  expect_equal(d1$A_a, tb(value = 1:3, type = c("a", "b", "c")))
  expect_equal(d1$A_b, tb(value = 4:6, type = c("a", "b", "c")))
  expect_equal(d1$B_a, tb(value = 1:3, type = c("x", "y", "z")))
})