tests/testthat/test_PredictionClust.R

In mlr3cluster: Cluster Extension for 'mlr3'

context("PredictionClust")

test_that("Construction", {
  task = tsk("usarrests")
  p = PredictionClust$new(row_ids = task$row_ids, partition = rep.int(1L, nrow(task$data())))
  expect_prediction(p)
  expect_prediction_clust(p)

  expect_prediction(c(p, p))
})

test_that("Internally constructed Prediction", {
  task = tsk("usarrests")
  lrn = mlr_learners$get("clust.featureless")
  lrn$param_set$values = list(num_clusters = 1L)
  p = lrn$train(task)$predict(task)
  expect_prediction(p)
  expect_prediction_clust(p)
})

test_that("filter works", {
  task = tsk("usarrests")
  lrn = mlr_learners$get("clust.featureless")
  lrn$param_set$values = list(num_clusters = 1L)
  p = lrn$train(task)$predict(task)
  pdata = p$data

  pdata = filter_prediction_data(pdata, row_ids = 1:3)
  expect_set_equal(pdata$row_ids, 1:3)
  expect_integer(pdata$partition, len = 3)
})