tests/testthat/test-rbind_parquet.R

In parquetize: Convert Files to Parquet Format

test_that("Checks rbind_parquet creates correct output file", {
  temp_dir <- tempfile()

  dir.create(temp_dir, showWarnings = FALSE)

  file.create(fileext = file.path(temp_dir, "test_data1-4.parquet"))
  write_parquet(data.frame(
    x = c("a","b","c"),
    y = c(1L,2L,3L)
  ), file.path(temp_dir, "test_data1-4.parquet"))

  file.create(fileext = file.path(temp_dir, "test_data4-6.parquet"))
  write_parquet(data.frame(
    x = c("d","e","f"),
    y = c(4L,5L,6L)
  ), file.path(temp_dir, "test_data4-6.parquet"))

  test_data <- rbind_parquet(folder = temp_dir,
                             output_name = "test_data",
                             delete_initial_files = FALSE)

  expect_equal(
    unname(unlist(lapply(test_data, class))),
    c("character", "integer")
  )

  expect_equal(names(test_data), c("x",
                                   "y"))

})