tests/testthat/test-zap_missing.R

In haven: Import and Export 'SPSS', 'Stata' and 'SAS' Files

test_that("strips na tags", {
  x1 <- labelled(tagged_na("a", "b"), c(a = tagged_na("a"), b = 1))
  x2 <- zap_missing(x1)

  expect_equal(na_tag(x2), c(NA_character_, NA))
  expect_equal(attr(x2, "labels"), c(b = 1))
})

test_that("converts user-defined missings", {
  x1 <- labelled_spss(c(1, 2, 99), c(missing = 99), na_values = 99)
  x2 <- zap_missing(x1)
  expect_s3_class(x2, "haven_labelled")
  expect_equal(as.integer(x2), c(1, 2, NA))

  x3 <- labelled_spss(1:10, na_values = c(2, 4), na_range = c(8, 10))
  x4 <- zap_missing(x3)
  expect_s3_class(x4, "haven_labelled")
  expect_equal(as.integer(x4), c(1, NA, 3, NA, 5, 6, 7, NA, NA, NA))
})

test_that("converts data frame", {
  x1 <- labelled(tagged_na("a", "b"), c(a = tagged_na("a"), b = 1))

  df1 <- tibble::tibble(x1 = 1, x2 = 2:1)
  df2 <- zap_missing(df1)

  expect_equal(na_tag(df1$x1), c(NA_character_, NA))
  expect_equal(df1$x2, df2$x2)
})