tests/testthat/test-cast-shadow.R

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

test_that("cast_shadow returns a tibble",{
  expect_s3_class(cast_shadow(airquality, Ozone), "tbl_df")
})

test_that("cast_shadow adds the right number of columns",{
  expect_equal(2, ncol(cast_shadow(airquality, Ozone)))
  expect_equal(4, ncol(cast_shadow(airquality, Ozone, Solar.R)))
  expect_equal(6, ncol(cast_shadow(airquality, Ozone, Solar.R, Temp)))
})

test_that("cast_shadow adds a column with suffix '_NA'",{
  expect_equal(names(cast_shadow(airquality, Ozone)),
               c("Ozone","Ozone_NA"))
  expect_equal(names(cast_shadow(airquality, Ozone, Solar.R)),
               c("Ozone","Solar.R", "Ozone_NA", "Solar.R_NA"))
  expect_equal(names(cast_shadow(airquality, Ozone, Solar.R, Temp)),
               c("Ozone","Solar.R","Temp", "Ozone_NA", "Solar.R_NA", "Temp_NA"))
})