tests/testthat/test-add-n-miss.R

In njtierney/narnia: Data Structures, Summaries, and Visualisations for Missing Data

test_df <- data.frame(x = c(NA, 2, 3), y = c(1, NA, 3), z = c(1, 2, 3))

df_n_miss <- data.frame(n_miss_all = c(1L, 1L, 0L))

test_df_answer <- dplyr::bind_cols(test_df, df_n_miss)


test_that("add_n_miss adds a column", {
  expect_equal(ncol(test_df) + 1, ncol(add_n_miss(test_df)))
})

test_that("add_n_miss adds a column named 'n_miss_all'", {
  expect_equal(names(add_n_miss(test_df)), c(names(test_df), "n_miss_all"))
})

test_that("add_n_miss adds a column named 'n_miss_vars' when a variable is selected", {
  expect_equal(names(add_n_miss(test_df, y)), c(names(test_df), "n_miss_vars"))
})

test_that("add_n_miss returns the correct number", {
  expect_equal(add_n_miss(test_df), test_df_answer)
})