tests/testthat/test-as-shadow.R

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

test_that("as_shadow returns a data.frame or tibble",{
  expect_s3_class(as_shadow(airquality), "data.frame")
  expect_s3_class(as_shadow(airquality), "tbl")
})

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

test_that("as_shadow returns correct values",{
  expect_equal(as.character(test_df$x_NA), c("!NA", "!NA"))
  expect_equal(as.character(test_df$y_NA), c("NA", "NA"))
  expect_equal(as.character(test_df$z_NA), c("!NA", "NA"))
})

test_that("as_shadow returns factors",{
  expect_s3_class(test_df$x_NA, "factor")
  expect_s3_class(test_df$y_NA, "factor")
  expect_s3_class(test_df$z_NA, "factor")
})

test_that("as_shadow returns shade",{
  expect_s3_class(test_df$x_NA, "shade")
  expect_s3_class(test_df$y_NA, "shade")
  expect_s3_class(test_df$z_NA, "shade")
})

test_that("as_shadow returns correct levels",{
  expect_equal(levels(test_df$x_NA), c("!NA", "NA"))
})

test_that("as_shadow errors when given non dataframe or 0 entry",{
  expect_snapshot(
    error = TRUE,
    as_shadow(0)
    )
  expect_snapshot(
    error = TRUE,
    as_shadow("a")
    )
  expect_snapshot(
    error = TRUE,
    as_shadow(matrix(airquality))
    )
  expect_snapshot(
    error = TRUE,
    as_shadow(NULL)
    )
})

test_that("The dimensions are the same after using as_shadow",{
  expect_equal(dim(as_shadow(airquality)),dim(airquality))
})

library(purrr)

aq_shadow <- as_shadow(airquality)
classes <- map(aq_shadow,class)
first_classes <- map(classes, pluck, 1)
second_classes <- map(classes, pluck, 2)

test_that("as_shadow returns shadow first",{
  expect_equal(sum(first_classes == "shade"), ncol(airquality))
})

test_that("as_shadow returns factor second",{
  expect_equal(sum(second_classes == "factor"), ncol(airquality))
})

test_that("as_shadow returns columns with additional suffix _NA",{
  expect_equal(names(as_shadow(airquality)), paste0(names(airquality),"_NA"))
})