tests/testthat/test-select_forcis_columns.R

In FRBCesab/forcis: An R Client to Access the FORCIS Database

## Data for tests ----

req_cols <- get_required_columns()

df_net <- data.frame(matrix(1:(length(req_cols) + 4), nrow = 1))
colnames(df_net) <- c(req_cols, "n_pachyderma", "conglobatus", "g_rubescens",
                      "toto")

df_net$"data_type" <- "Net"

df_cpr <- data.frame(matrix(1:(length(req_cols) + 4), nrow = 1))
colnames(df_cpr) <- c(req_cols, "species", "count_bin_min", "count_bin_max",
                      "toto")

df_cpr$"data_type" <- "CPR North"


## select_forcis_columns() ----

test_that("Test select_forcis_columns() for error", {
  
  expect_error(select_forcis_columns(df_net, col = 2),
               "Argument 'cols' must be a character vector",
               fixed = TRUE)
  
  expect_error(select_forcis_columns(df_net, col = "titi"),
               "Some columns to select are absent from data",
               fixed = TRUE)
})

test_that("Test select_forcis_columns() for success", {
  
  expect_silent(df <- select_forcis_columns(df_net))
  
  expect_true(is.data.frame(df))
  expect_equal(ncol(df), ncol(df_net) - 1)
  expect_equal(nrow(df), 1L)
  
  expect_silent(df <- select_forcis_columns(df_net, cols = "toto"))
  
  expect_true(is.data.frame(df))
  expect_equal(ncol(df), ncol(df_net))
  expect_equal(nrow(df), 1L)
  expect_equal(colnames(df)[length(req_cols) + 1], "toto")
  
  expect_silent(df <- select_forcis_columns(df_cpr))
  
  expect_true(is.data.frame(df))
  expect_equal(ncol(df), ncol(df_cpr) - 1)
  expect_equal(nrow(df), 1L)
  
  expect_silent(df <- select_forcis_columns(df_cpr, cols = "toto"))
  
  expect_true(is.data.frame(df))
  expect_equal(ncol(df), ncol(df_cpr))
  expect_equal(nrow(df), 1L)
  expect_equal(colnames(df)[length(req_cols) + 1], "toto")
})