test-plots.R

context("test-plots")

test_that("output from ds_screener is as expected", {
  p <- ds_screener(airquality)
  vdiffr::expect_doppelganger("ds_screener_1", p)
})

test_that("output from ds_plot_scatter is as expected", {
  p <- ds_plot_scatter(mtcarz, mpg, disp, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_scatter_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_scatter_2", p[[2]])
})

test_that("output from ds_plot_histogram is as expected", {
  p <- ds_plot_histogram(mtcarz, mpg, disp, hp, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_hist_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_hist_2", p[[2]])
  vdiffr::expect_doppelganger("ds_plot_hist_3", p[[3]])
})

test_that("output from ds_plot_density is as expected", {
  p <- ds_plot_density(mtcarz, mpg, disp, hp, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_density_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_density_2", p[[2]])
  vdiffr::expect_doppelganger("ds_plot_density_3", p[[3]])
})

test_that("output from ds_plot_bar is as expected", {
  p <- ds_plot_bar(mtcarz, cyl, gear, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_bar_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_bar_2", p[[2]])
})

test_that("output from ds_plot_box_single is as expected", {
  p <- ds_plot_box_single(mtcarz, mpg, disp, hp, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_box_single_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_box_single_2", p[[2]])
})

test_that("output from ds_plot_bar_stacked is as expected", {
  p <- ds_plot_bar_stacked(mtcarz, cyl, gear, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_bar_stacked_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_bar_stacked_2", p[[2]])
})

test_that("output from ds_plot_bar_grouped is as expected", {
  p <- ds_plot_bar_grouped(mtcarz, cyl, gear, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_bar_grouped_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_bar_grouped_2", p[[2]])
})

test_that("output from ds_plot_box_group is as expected", {
  p <- ds_plot_box_group(mtcarz, cyl, gear, mpg, print_plot = FALSE)
  vdiffr::expect_doppelganger("ds_plot_box_group_1", p[[1]])
  vdiffr::expect_doppelganger("ds_plot_box_group_2", p[[2]])
})

ndata <- dplyr::select(mtcarz, mpg, disp, hp, wt, qsec, drat)
fdata <- dplyr::select(mtcarz, cyl, gear, am, vs)

test_that("ds_plot_scatter throws errors as expected", {
  expect_error(ds_plot_scatter(fdata), 'Data has no continuous variables.')
  expect_error(ds_plot_scatter(fdata, cyl, gear), 'Data has no continuous variables.')
  expect_error(ds_plot_scatter(ndata, mpg), 'Scatter plot requires 2 continuous variables.')
})

test_that("ds_plot_histogram throws errors as expected", {
  expect_error(ds_plot_histogram(fdata), 'Data has no continuous variables.')
  expect_error(ds_plot_histogram(fdata, cyl, gear), 'Data has no continuous variables.')
})

test_that("ds_plot_density throws errors as expected", {
  expect_error(ds_plot_density(fdata), 'Data has no continuous variables.')
  expect_error(ds_plot_density(fdata, cyl, gear), 'Data has no continuous variables.')
})

test_that("ds_plot_bar throws errors as expected", {
  expect_error(ds_plot_bar(ndata), 'Data has no categorical variables.')
  expect_error(ds_plot_bar(mtcarz, mpg, disp), 'Data has no categorical variables.')
})

test_that("ds_plot_box_single throws errors as expected", {
  expect_error(ds_plot_box_single(fdata), 'Data has no continuous variables.')
  expect_error(ds_plot_box_single(fdata, cyl, gear), 'Data has no continuous variables.')
})

test_that("ds_plot_bar_stacked throws errors as expected", {
  expect_error(ds_plot_bar_stacked(ndata), 'Data has no categorical variables.')
  expect_error(ds_plot_bar_stacked(mtcarz, mpg, disp), 'Data has no categorical variables.')
  expect_error(ds_plot_bar_stacked(mtcarz, cyl), 'Stacked bar plot requires 2 categorical variables.')
})

test_that("ds_plot_bar_grouped throws errors as expected", {
  expect_error(ds_plot_bar_grouped(ndata), 'Data has no categorical variables.')
  expect_error(ds_plot_bar_grouped(mtcarz, mpg, disp), 'Data has no categorical variables.')
  expect_error(ds_plot_bar_grouped(mtcarz, cyl), 'Grouped bar plot requires 2 categorical variables.')
})

test_that("ds_plot_box_group throws errors as expected", {
  expect_error(ds_plot_box_group(fdata), 'Data has no continuous variables.')
  expect_error(ds_plot_box_group(ndata), 'Data has no categorical variables.')
  expect_error(ds_plot_box_group(fdata, cyl, gear), 'Data has no continuous variables.')
  expect_error(ds_plot_box_group(mtcarz, mpg, disp), 'Data has no categorical variables.')
})