tests/testthat/test_LearnerClustHierarchical.R

In mlr-org/mlr3viz: Visualizations for 'mlr3'

skip_if_not_installed("mlr3")
skip_if_not_installed("mlr3cluster")
skip_if_not_installed("ggdendro")
set.seed(42)

test_that("autoplot.LearnerClustHierarchical", {
  skip_on_cran()
  require_namespaces("mlr3cluster")

  learner = mlr3::lrn("clust.agnes")$train(mlr3::tsk("usarrests"))
  p = autoplot(learner)
  expect_true(is.ggplot(p))
  expect_doppelganger("learner_clust.agnes", p)

  learner = mlr3::lrn("clust.hclust")$train(mlr3::tsk("usarrests"))
  p = autoplot(learner, type = "scree")
  expect_true(is.ggplot(p))
  expect_doppelganger("learner_clust.hclust", p)
})