tests/testthat/test-plot.r

In jamesdunham/dgirt: Dynamic Estimation of Group-Level Opinion

context("plots")

test_that("dgirt_plot and plot can be called", {
  expect_silent(plot(toy_dgirtfit))
  expect_silent(dgirt_plot(toy_dgirtfit))
})


test_that("plots can be amended", {
  expect_silent(dgirt_plot(toy_dgirtfit) %+% ylab("foo"))
})

test_that("plot calls dgirt_plot", {
  p <- plot(toy_dgirtfit)
  dp <- dgirt_plot(toy_dgirtfit)
  expect_equal(p, dp)
})

test_that("plot_rhats handles parameters", {
  expect_error(plot_rhats(toy_dgirtfit, pars = "xi", facet_vars = "state"),
               "not indexed by")
  expect_silent(plot_rhats(toy_dgirtfit, pars = "xi", color_var = "year"))
})

test_that("pars must be a length-one character", {
  expect_error(plot_rhats(toy_dgirtfit, pars = NULL), "not a string")
  expect_error(plot_rhats(toy_dgirtfit, pars = c("xi", "sd_item")),
               "not a string")
})

test_that("argument validation works", {
  estimates = as.data.frame(toy_dgirtfit)
  expect_silent(dgirt_plot(estimates, group_name = NULL,
      time_name = 'year', geo_name = 'state'))
  expect_error(dgirt_plot(estimates, group_name = NULL,
      time_name = c('year', 'race3'), geo_name = 'state'), "not a string")
  expect_silent(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_fun = 'mean'))
  expect_error(dgirt_plot(estimates,  group_name = NULL,
    time_name = 'year', geo_name = 'state', y_fun = mean), "not a string")
  expect_error(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_fun = 'foo'),
  "could not find function")
  expect_error(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_max = NULL), "should be too")
  expect_error(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_min = NULL), "should be too")
  expect_error(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_fun = NULL), "should be too")
  expect_silent(dgirt_plot(estimates, group_name = NULL,
    time_name = 'year', geo_name = 'state', y_min = NULL, y_max = NULL))
})