tests/testthat/test_inspect_na_grouped.R

In alastairrushworth/inspectdf: Inspection, Comparison and Visualisation of Data Frames

library(vdiffr)
context("inspect_na with grouped dataframes")

# load in some example data
data("starwars", package = "dplyr")
data("band_instruments", package = "dplyr")
data("storms", package = "dplyr")
data(mtcars, airquality)

test_that("inspect_na & grouped dataframe = df (mtcars)", {
  x <- try(mtcars %>% dplyr::group_by(am) %>% inspect_na(), silent = TRUE)
  expect_is(x, "data.frame")
  expect_equal(attr(x, "type")$input_type, "grouped")
  expect_equal(colnames(x)[1], "am")
})

test_that("inspect_na & grouped dataframe = df (storms)", {
  x <- try(storms %>% dplyr::group_by(status) %>% inspect_na(), silent = TRUE)
  expect_is(x, "data.frame")
  expect_equal(attr(x, "type")$input_type, "grouped")
  expect_equal(colnames(x)[1], "status")
})

test_that("inspect_na for tech", {
  x_year <- try(tech %>% dplyr::group_by(year) %>% inspect_na(), silent = TRUE)
  x_quarter <- try(tech %>% dplyr::group_by(quarter) %>% inspect_na(), silent = TRUE)
  expect_is(x_year, "data.frame")
  expect_is(x_quarter, "data.frame")
  expect_equal(attr(x_year, "type")$input_type, "grouped")
  expect_equal(attr(x_quarter, "type")$input_type, "grouped")
  expect_equal(colnames(x_quarter)[1], "quarter")
  expect_equal(colnames(x_year)[1], "year")
})

# test_that("Grouped df correlation plots work", {
#   set.seed(10)
#   tech$apple[sample.int(nrow(tech), sample.int(500))] <- NA
#   tech$microsoft[sample.int(nrow(tech), sample.int(500))] <- NA
#   tech$ibm[sample.int(nrow(tech), sample.int(500))] <- NA
#   expect_doppelganger("Inspect-na-grouped-tech-year", tech %>% dplyr::group_by(year) %>% inspect_na() %>% show_plot)
#   expect_doppelganger("Inspect-na-grouped-tech-quarter", tech %>% dplyr::group_by(quarter) %>% inspect_na() %>% show_plot)
# })