test_that("Test heatmap", {
  small_plot <- plot(fairness_heatmap(fobject))[[3]]
  big_plot <- plot(fairness_heatmap(fobject_big), flip_axis = TRUE)[[3]]

  # heatmap converts shape while number of explainers exceed number of metrics
  expect_equal(c(small_plot$labels$x, small_plot$labels$y), c(big_plot$labels$y, big_plot$labels$x))

  top_dendogram <- plot(fairness_heatmap(fobject))[[1]]

  expect_equal(top_dendogram$labels$title, "Heatmap")
  expect_equal(top_dendogram$labels$subtitle, "With dendograms")

Try the fairmodels package in your browser

Any scripts or data that you put into this service are public.

fairmodels documentation built on Aug. 24, 2022, 1:05 a.m.