tests/testthat/test-read_functions.R

In Technomics/costmisc: Misc costverse functions

test_that("read_folder loads files correctly", {

  check_data <- list()
  for (i in 1:4) {
    check_data[[i]] <- mtcars[seq((i - 1) * 8 + 1, (i * 8)), ]
    rownames(check_data[[i]]) <- NULL
  }

  check_data1 <- check_data
  names(check_data1) <- paste0("mtcars_", 1:4, "_csv")

  check_data2 <- check_data
  names(check_data2) <- paste0("/mtcars_", 1:4, ".csv")

  check_data3 <- check_data1 %>%
    listindex_to_col("new_id")

  the_dir <- system.file("examples/mtcars", package = "costmisc")

  x1 <- read_folder(the_dir, read.csv)
  x2 <- read_folder(the_dir, read.csv, .clean_file_names = FALSE)
  x3 <- read_folder(the_dir, read.csv, .id = "new_id")

  expect_equal(x1, check_data1)
  expect_equal(x2, check_data2)
  expect_equal(x3, check_data3)

})

test_that("col_rep returns the correct col_type specification", {

  test_str1 <- "cindlfDt?-a"
  test_str2 <- "c3n2dl3-2c"

  x <- col_rep(test_str1)
  x_readxl <- col_rep(test_str1, "readxl")
  x_sql <- col_rep(test_str1, "sql")

  x2 <- col_rep(test_str2)

  x_readxl_check <- c("text", "numeric", "numeric", "numeric", "logical", "text", "date", "date", "guess", "skip", NA)
  x_sql_check <- c("VARCHAR", "INTEGER", "DOUBLE", "DOUBLE", "BIT", "VARCHAR", "DATETIME", "DATETIME", NA, NA, "COUNTER")

  x2_check <- "cccnndlll--c"

  expect_equal(test_str1, x)
  expect_equal(x_readxl, x_readxl_check)
  expect_equal(x_sql, x_sql_check)
  expect_equal(x2, x2_check)

})

test_that("functions for reading excel tables work", {

  example_file <- system.file("examples/excel_examples.xlsx", package = "costmisc")

  wb <- openxlsx::loadWorkbook(example_file)

  # check that list of tables align
  table_list <- get_excel_tables(wb)
  table_list2 <- get_excel_tables(wb, "tables")

  table_check <- tibble::tibble(sheet = rep("tables", 2),
                                table = c("tbl_mtcars", "tbl_iris"),
                                range = c("I2:T34", "B2:F152"))

  expect_equal(table_list, table_check)
  expect_equal(table_list2, table_check)

  expect_error(get_excel_tables(wb, "not real"), "Sheet 'not real' does not exist.")

  # check that table reads in correctly
  df_mtcars <- read_excel_table(wb, "tbl_mtcars")
  df_mtcars2 <- read_excel_table(wb, "tbl_mtcars", table_list)

  expect_equal(df_mtcars, df_mtcars2)

  rownames(df_mtcars) <- df_mtcars$Model
  df_mtcars <- df_mtcars[,-1]

  expect_equal(df_mtcars, mtcars)

  expect_error(read_excel_table(wb, "tbl_fake"), "Table \"tbl_fake\" not uniquely found")
})

test_that("when openxlsx cannot be loaded an error triggers", {

  with_mock(
    requireNamespace = function(ns, quietly) FALSE,
    expect_error(get_excel_tables(0, 0), "Package \"openxlsx\" needed for this function to work. Please install it."),
    expect_error(read_excel_table(0, 0), "Package \"openxlsx\" needed for this function to work. Please install it.")
  )

})