tests/testthat/test-tidyr_verbs.R

In R6Frame: R6 Wrapper for Data Frames

context("tidyr verbs")

org <- data.frame(Q1 = c("Example 1", "Example 2"), Score1 = c(9, 8), Score2 = c(80, 70), stringsAsFactors = FALSE)

# Complete ---------------------------------------------------------------------
test_that("complete works with R6 data.frame", {
  skip_if_not_installed("dplyr")
  skip_if_not_installed("tidyr")

  df <- R6Frame$new(org)
  lvl <- c("Example 1", "Example 2", "Example 3")
  df$data$Q1 <- factor(df$data$Q1, levels = lvl)
  df <- tidyr::complete(df, Q1)

  expect_s3_class(df, "R6Frame")
  expect_identical(as.character(df$data$Q1), lvl)

})

test_that("complete works with R6 data.table", {
  skip_if_not_installed("dplyr")
  skip_if_not_installed("tidyr")

  dt <- R6Frame$new(data.table::as.data.table(org))
  lvl <- c("Example 1", "Example 2", "Example 3")
  dt$data$Q1 <- factor(dt$data$Q1, levels = lvl)
  dt <- tidyr::complete(dt, Q1)

  expect_s3_class(dt, "R6Frame")
  expect_identical(as.character(dt$data$Q1), lvl)

})

test_that("complete works with R6 tbl_df", {
  skip_if_not_installed("dplyr")
  skip_if_not_installed("tidyr")

  tbl <- R6Frame$new(dplyr::as.tbl(org))
  lvl <- c("Example 1", "Example 2", "Example 3")
  tbl$data$Q1 <- factor(tbl$data$Q1, levels = lvl)
  tbl <- tidyr::complete(tbl, Q1)

  expect_s3_class(tbl, "R6Frame")
  expect_identical(as.character(tbl$data$Q1), lvl)

})

test_that("expand works with R6 tbl_df", {
  skip_if_not_installed("dplyr")
  skip_if_not_installed("tidyr")

  df <- data.frame(
    Q1 = rep("Example", 7L),
    year   = c(2010, 2010, 2010, 2010, 2012, 2012, 2012),
    qtr    = c(   1,    2,    3,    4,    1,    2,    3),
    return = rnorm(7)
  )

  tbl <- R6Frame$new(df)
  tbl <- tidyr::expand(tbl, Q1, year, qtr)

  expect_s3_class(tbl, "R6Frame")

})

test_that("fill works with R6 tbl_df", {
  skip_if_not_installed("dplyr")
  skip_if_not_installed("tidyr")

  df <- data.frame(Q1 = rep("Example", 12L), Month = 1:12, Year = c(2000, rep(NA, 11)))
  tbl <- R6Frame$new(df)
  tbl <- tidyr::fill(tbl, Year)

  expect_s3_class(tbl, "R6Frame")
})