tests/testthat/test-impute-below-at.R

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

miss_vec <- rnorm(100)

# add 20 missing values
miss_vec[sample(1:100, 20)] <- NA

miss_df <- data.frame(miss_vec)

test_that("impute_below_at leaves no NA values", {
  expect_false(
    anyNA(impute_below_at(miss_df, "miss_vec")[
      which(is.na(miss_df$miss_vec)),
    ])
  )
})

test_that("impute_below_at works with vars", {
  expect_false(
    anyNA(impute_below_at(miss_df, dplyr::vars(miss_vec))[
      which(is.na(miss_df$miss_vec)),
    ])
  )
})

miss_vec <- rnorm(100)

# add 20 missing values
miss_vec[sample(1:100, 20)] <- NA

miss_df <- data.frame(miss_vec)

test_that("impute_below_if leaves no NA values", {
  expect_false(
    anyNA(impute_below_if(airquality, .predicate = is.numeric))
  )
})