checks on arguments

Author: abichat

R/aggregate_hclust.R

@@ -112,6 +112,12 @@ step_aggregate_hclust_new <- function(terms, role, trained,
 prep.step_aggregate_hclust <- function(x, training, info = NULL, ...) {
   col_names <- recipes_eval_select(x$terms, training, info)
   check_type(training[, col_names], quant = TRUE)
+  check_in(x$dist_metric, name_x = "dist_metric",
+           values = c("euclidean", "maximum", "manhattan",
+                      "canberra", "binary", "minkowski"))
+  check_in(x$linkage_method, name_x = "linkage_method",
+           values = c("ward.D", "ward.D2", "single", "complete",
+                      "average", "mcquitty", "median", "centroid"))
 
   ct <-
     training[, col_names] %>%

tests/testthat/test-aggregate_hclust.R

@@ -91,3 +91,27 @@ test_that("step_aggregate_hclust works", {
   expect_equal(colnames(baked2),
                c(colnames(cheese_abundance), new_names))
 })
+
+
+test_that("step_aggregate_hclust throws errors", {
+
+  expect_error(
+    recipe(~ ., data = cheese_abundance) %>%
+      step_aggregate_hclust(all_numeric_predictors(),
+                            n_clusters = nc, fun_agg = sum,
+                            dist_metric = "hello") %>%
+      prep(),
+    '`dist_metric` must be one of "euclidean",'
+  )
+
+  expect_error(
+    recipe(~ ., data = cheese_abundance) %>%
+      step_aggregate_hclust(all_numeric_predictors(),
+                            n_clusters = nc, fun_agg = sum,
+                            linkage_method = "hello") %>%
+      prep(),
+    '`linkage_method` must be one of "ward.D",'
+  )
+
+
+})