test that all levels are preserved with 1 row predictions

Author: EmilHvitfeldt

tests/testthat/test-hier_clust-stats.R

@@ -27,6 +27,21 @@ test_that("predicting", {
   )
 })
 
+test_that("all levels are preserved with 1 row predictions", {
+  set.seed(1234)
+  spec <- hier_clust(num_clusters = 3) %>%
+    set_engine("stats")
+
+  res <- fit(spec, ~., mtcars)
+
+  preds <- predict(res, mtcars[1, ])
+
+  expect_identical(
+    levels(preds$.pred_cluster),
+    paste0("Cluster_", 1:3)
+  )
+})
+
 test_that("extract_centroids() works", {
   set.seed(1234)
   spec <- hier_clust(num_clusters = 3) %>%

tests/testthat/test-k_means-clustMixType.R

@@ -40,6 +40,21 @@ test_that("predicting", {
   )
 })
 
+test_that("all levels are preserved with 1 row predictions", {
+  set.seed(1234)
+  spec <- k_means(num_clusters = 3) %>%
+    set_engine("clustMixType")
+
+  res <- fit(spec, ~., iris)
+
+  preds <- predict(res, iris[1, ])
+
+  expect_identical(
+    levels(preds$.pred_cluster),
+    paste0("Cluster_", 1:3)
+  )
+})
+
 test_that("extract_centroids() works", {
   skip_if_not_installed("clustMixType")
 

tests/testthat/test-k_means-clusterR.R

@@ -32,6 +32,21 @@ test_that("predicting", {
   )
 })
 
+test_that("all levels are preserved with 1 row predictions", {
+  set.seed(1234)
+  spec <- k_means(num_clusters = 3) %>%
+    set_engine("ClusterR")
+
+  res <- fit(spec, ~., mtcars)
+
+  preds <- predict(res, mtcars[1, ])
+
+  expect_identical(
+    levels(preds$.pred_cluster),
+    paste0("Cluster_", 1:3)
+  )
+})
+
 test_that("extract_centroids() works", {
   skip_if_not_installed("ClusterR")
 

tests/testthat/test-k_means-klaR.R

@@ -51,6 +51,28 @@ test_that("predicting", {
   )
 })
 
+test_that("all levels are preserved with 1 row predictions", {
+  skip_if_not_installed("klaR")
+  skip_if_not_installed("modeldata")
+
+  data("ames", package = "modeldata")
+
+  ames_cat <- dplyr::select(ames, dplyr::where(is.factor))
+
+  set.seed(1234)
+  spec <- k_means(num_clusters = 3) %>%
+    set_engine("klaR")
+
+  res <- fit(spec, ~., ames_cat)
+
+  preds <- predict(res, ames_cat[1, ])
+
+  expect_identical(
+    levels(preds$.pred_cluster),
+    paste0("Cluster_", 1:3)
+  )
+})
+
 test_that("predicting ties argument works", {
   skip_if_not_installed("klaR")
 

tests/testthat/test-k_means-stats.R

@@ -27,6 +27,21 @@ test_that("predicting", {
   )
 })
 
+test_that("all levels are preserved with 1 row predictions", {
+  set.seed(1234)
+  spec <- k_means(num_clusters = 3) %>%
+    set_engine("stats")
+
+  res <- fit(spec, ~., mtcars)
+
+  preds <- predict(res, mtcars[1, ])
+
+  expect_identical(
+    levels(preds$.pred_cluster),
+    paste0("Cluster_", 1:3)
+  )
+})
+
 test_that("extract_centroids() works", {
   set.seed(1234)
   spec <- k_means(num_clusters = 3) %>%