test_that("heat_plot works", { set.seed(1) model_fit <- compute_mallows( setup_rank_data(potato_visual), compute_options = set_compute_options(nmc = 20, burnin = 3) ) expect_s3_class(heat_plot(model_fit, type = "MAP"), "ggplot") obj <- heat_plot(model_fit) expect_equal(sum(obj$data$probability), 20) model_fit <- compute_mallows( setup_rank_data(potato_visual), compute_options = set_compute_options(nmc = 20, burnin = 3, rho_thinning = 3) ) obj <- heat_plot(model_fit) expect_equal(sum(obj$data$probability), 20) set.seed(1) model_fit <- compute_mallows( setup_rank_data(potato_visual), compute_options = set_compute_options(nmc = 20) ) expect_error(heat_plot(model_fit), "Please specify the burnin.") burnin(model_fit) <- 1 expect_s3_class(heat_plot(model_fit, type = "MAP"), "ggplot") model_fit <- compute_mallows( setup_rank_data(cluster_data), model_options = set_model_options(n_clusters = 3), compute_options = set_compute_options(nmc = 20, burnin = 10) ) expect_error( heat_plot(model_fit), "heat_plot only works for a single cluster" ) })