tests/testthat/test-dcr_checks.R

In datacleanr: Interactive and Reproducible Data Cleaning

test_that("Initial dcr checks recognize wrong class", {
  testdf <- iris
  class(testdf) <- "something"

  expect_error(dcr_checks(testdf),
    regexp = "Please provide a data.frame, tibble, or data.table.",
    fixed = TRUE
  )
})


test_that("Initial dcr checks recognize non-existing file", {
  expect_error(dcr_checks("Abc.Rds"),
    regexp = "File does not exist.",
    fixed = TRUE
  )
})

test_that("Initial dcr checks recognize wrong file ext", {
  expect_error(dcr_checks("Abc.Rdd"),
    regexp = "Please provide a *.Rds file.",
    fixed = TRUE
  )
})

test_that("Initial dcr checks recognizes wrong class in rds from filepath", {
  tmpfile <- tempfile(fileext = ".Rds")

  testdf <- iris
  class(testdf) <- "something"
  saveRDS(testdf, file = tmpfile)

  expect_error(dcr_checks(tmpfile),
    regexp = "Please provide a data.frame, tibble, or data.table in your file path.",
    fixed = TRUE
  )
  unlink(tmpfile)
})

test_that("Initial dcr checks works on iris from filepath", {
  tmpfile <- tempfile(fileext = ".Rds")
  saveRDS(iris, file = tmpfile)

  expect_identical(
    dcr_checks(tmpfile),
    list(
      dataset = iris,
      file_path = tmpfile
    )
  )
  expect_identical(
    dcr_checks(fs::path(tmpfile)),
    list(
      dataset = iris,
      file_path = fs::path(tmpfile)
    )
  )
  unlink(tmpfile)
})


test_that("Initial dcr checks works on iris", {
  expect_identical(
    dcr_checks(iris),
    list(
      dataset = iris,
      file_path = NULL
    )
  )
})


test_that("Initial dcr checks fails with bad .dcrflag column", {
  expect_error(dcr_checks(iris %>% dplyr::mutate(.dcrflag = "A")),
    regexp = "Detected column .dcrflag - Please ensure it is of class logical (i.e. TRUE/FALSE).",
    fixed = TRUE
  )
})


test_that("Initial dcr checks passes with TRUE/FALSE .dcrflag", {
  expect_identical(
    dcr_checks(iris %>%
      dplyr::mutate(
        .dcrflag = rep(c(TRUE, FALSE), 75)
      )),
    list(
      dataset = iris %>%
        dplyr::mutate(
          .dcrflag = rep(c(TRUE, FALSE), 75)
        ),
      file_path = NULL
    )
  )
})





# new ---------------------------------------------------------------------

test_that("Initial dcr checks fails when .annotation column exists", {
  expect_error(dcr_checks(iris %>% dplyr::mutate(.annotation = "A")),
    regexp = "A .annotation column already exists. Please rename it (e.g., to .dcrflag) to prevent any clashes and try again.",
    fixed = TRUE
  )
})

test_that("Dataset >= 1.5e6 notifies user", {
  expect_message(dcr_checks(data.frame(x = seq_len(1.5e6))),
    regexp = "Consider breaking it up into smaller chunks with a split-check-combine approach.\n ",
    fixed = TRUE
  )
})



# test_that("User warned about large data set", {
#
#
#   testdf <- data.frame(x = 1:15001)
#
#   expect_warning(dcr_checks(testdf),
#                  regexp = "Data set has over")
# })
#