tests/testthat/test-write.R

In chunked: Chunkwise Text-File Processing for 'dplyr'

context("write")

source("util.R")
tbl_iris <- get_tbl_iris()

test_that("write_csv_chunkwise works", {
  tmp <- tempfile()
  write_csv_chunkwise(tbl_iris, tmp, row.names = FALSE)
  iris2 <- read.csv(tmp, stringsAsFactors = TRUE)
  expect_equal(iris2, iris)
})

test_that("write_csv2_chunkwise works", {
  tmp <- tempfile()
  write_csv2_chunkwise(tbl_iris, tmp, row.names = FALSE)
  iris2 <- read.csv2(tmp, stringsAsFactors = TRUE)
  expect_equal(iris2, iris)

  write_chunkwise(tbl_iris, tmp, format="csv2", row.names = FALSE)
  iris2 <- read.csv2(tmp, stringsAsFactors = TRUE)
  expect_equal(iris2, iris)
})

test_that("write_chunkwise to db works", {
  db <- get_empty_db()
  iris2 <- write_chunkwise(tbl_iris, db, "iris")
  expect_equal(as.data.frame(iris2)[1:4], as.data.frame(tbl_iris)[1:4])
  tmp <- tempfile()
  write_chunkwise(iris2, tmp, row.names = FALSE)
  iris3 <- read.csv(tmp, stringsAsFactors = TRUE)
  expect_equal(iris3, iris)
  DBI::dbDisconnect(db$con)
})