tests/testthat/test-list.r

In plyr: Tools for Splitting, Applying and Combining Data

context("Lists")

test_that("data frame variables converted to list names", {
  x <- dlply(esoph, .(alcgp), function(df) mean(df$ncases))
  expect_that(names(x), equals(levels(esoph$alcgp)))

  x <- dlply(esoph, .(alcgp, agegp), function(df) mean(df$ncases))
  y <- ddply(esoph, .(alcgp, agegp), function(df) mean(df$ncases))
  labs <- paste(y$alcgp, y$agegp, sep = ".")
  expect_that(names(x), equals(labs))

})

test_that("list names are preserved", {
  a <- 1:10
  expect_that(names(llply(a)), equals(NULL))

  names(a) <- letters[1:10]
  expect_that(names(llply(a)), equals(letters[1:10]))

})

# Test for #142
test_that(".n column can be renamed or dropped", {
  f <- function() data.frame(r=runif(1))

  out1 <- rdply(4, f)
  out2 <- rdply(4, f, .id='x')

  expect_equal(names(out1), c('.n', 'r'))
  expect_equal(names(out2), c('x', 'r'))

  # more testing
  expect_identical(out1$.n, seq_len(4))
  expect_identical(out2$x, seq_len(4))

  out <- rdply(4, f, .id=NULL)
  expect_equal(names(out), c('r'))
  expect_false(out$r[4] == 4)

  out <- rdply(4, f, .id='r') # names conflict
  expect_equal(names(out), c('r'))

  out <- rdply(4, f, .id='.n') # defaults
  expect_equal(names(out), c('.n', 'r'))
  expect_identical(out$.n, seq_len(4))
})