Skip to content

Commit 216aee3

Browse files
committed
don't use CENTROIDS in test
1 parent 12fa3c9 commit 216aee3

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

tests/testthat/test-k_means_diagnostics.R

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,19 @@
11
test_that("kmeans sse metrics work", {
2+
set.seed(1234)
23
kmeans_fit_stats <- k_means(num_clusters = mtcars[1:3, ]) %>%
34
set_engine("stats", algorithm = "MacQueen") %>%
45
fit(~., mtcars)
56

7+
# We don't use CENTROIDS argument becuase it breaks testing for different
8+
# versions of ClusterR #186
69
kmeans_fit_ClusterR <- k_means(num_clusters = 3) %>%
7-
set_engine("ClusterR", CENTROIDS = as.matrix(mtcars[1:3, ])) %>%
10+
set_engine("ClusterR") %>%
811
fit(~., mtcars)
912

1013
km_orig <- kmeans(mtcars, centers = mtcars[1:3, ], algorithm = "MacQueen")
1114
km_orig_2 <- ClusterR::KMeans_rcpp(
1215
data = mtcars,
13-
clusters = 3,
14-
CENTROIDS = as.matrix(mtcars[1:3, ])
16+
clusters = 3
1517
)
1618

1719
expect_equal(sse_within(kmeans_fit_stats)$wss,
@@ -92,7 +94,7 @@ test_that("kmeans sihouette metrics work", {
9294
fit(~., mtcars)
9395

9496
kmeans_fit_ClusterR <- k_means(num_clusters = 3) %>%
95-
set_engine("ClusterR", CENTROIDS = as.matrix(mtcars[1:3, ])) %>%
97+
set_engine("ClusterR") %>%
9698
fit(~., mtcars)
9799

98100
new_data <- mtcars[1:4, ]
@@ -124,7 +126,7 @@ test_that("kmeans sihouette metrics work with new data", {
124126
fit(~., mtcars)
125127

126128
kmeans_fit_ClusterR <- k_means(num_clusters = 3) %>%
127-
set_engine("ClusterR", CENTROIDS = as.matrix(mtcars[1:3, ])) %>%
129+
set_engine("ClusterR") %>%
128130
fit(~., mtcars)
129131

130132
new_data <- mtcars[1:4, ]

0 commit comments

Comments
 (0)