tests/testthat/test-fetch_csv.R

In Huh/collar: Utilities for Accessing and Manipulating GPS Data

test_that("Check assertions", {
  # Error should occur when NULL is passed
  expect_condition(
    fetch_csv(NULL),
    "file_path argument is empty, please provide a valid path to a file"
  )

  expect_condition(
    fetch_delim(NULL),
    "file_path argument is empty, please provide a valid path to a file"
  )

  # Try to read file that does not exist
  my_file <- paste0(Sys.time(), runif(10))
  expect_condition(
    fetch_csv("myfile"),
    "One or more files described in file_path do not exist"
  )
  expect_condition(
    fetch_delim("myfile"),
    "One or more files described in file_path do not exist"
  )

  # Successfully read and return csv
  tmp_dir <- tempdir()
  df <- data.frame(Val = c(2, 3))
  write.csv(df, file = file.path(tmp_dir, "mycsv.csv"), row.names = F)
  expect_s3_class(
    fetch_csv(file.path(tmp_dir, "mycsv.csv")),
    "data.frame"
  )
  expect_s3_class(
    fetch_delim(file.path(tmp_dir, "mycsv.csv")),
    "data.frame"
  )
})

test_that("Read single csv", {
  # Create and pass a file to fetch_csv
  tmp_dir <- tempdir()

  df <- data.frame(Val = c(2, 3))

  write.csv(df, file = file.path(tmp_dir, "mycsv.csv"), row.names = F)

  expect_s3_class(
    fetch_csv(file.path(tmp_dir, "mycsv.csv")),
    "data.frame"
  )
  expect_s3_class(
    fetch_delim(file.path(tmp_dir, "mycsv.csv")),
    "data.frame"
  )

})

test_that("Read multiple csv files", {
  # Create and pass multiple files to fetch_csv
  tmp_dir <- tempdir()

  df <- data.frame(Val = c(2, 3))

  write.csv(df, file = file.path(tmp_dir, "mycsv1.csv"), row.names = F)
  write.csv(df, file = file.path(tmp_dir, "mycsv2.csv"), row.names = F)

  fls <- paste(tmp_dir, c("mycsv1.csv", "mycsv2.csv"), sep = "/")

  expect_s3_class(
    fetch_csv(fls),
    "data.frame"
  )
  expect_s3_class(
    fetch_delim(fls),
    "data.frame"
  )

  expect_equal(
    fetch_csv(fls)$val,
    rep(df$Val, 2)
  )
  expect_equal(
    fetch_delim(fls)$val,
    rep(df$Val, 2)
  )
})

test_that("Check features of read_csv work", {
  tmp_dir <- tempdir()

  df <- data.frame(Val = c(" ab ", "cd", "Missing"))

  write.csv(df, file = file.path(tmp_dir, "readr.csv"), row.names = F)

  rd <- fetch_csv(
    file.path(tmp_dir, "readr.csv"),
    trim_ws = TRUE,
    na = "Missing"
  )
  rd2 <- fetch_delim(
    file.path(tmp_dir, "readr.csv"),
    trim_ws = TRUE,
    na = "Missing"
  )

  expect_equal(
    rd$val[1],
    "ab"
  )
  expect_equal(
    rd2$val[1],
    "ab"
  )
  expect_true(
    is.na(rd$val[3])
  )
  expect_true(
    is.na(rd2$val[3])
  )

})