tests/testthat/test-my_knn_cv.R

In thulinjt/project3package: Project 3 Package

library(dplyr)

my_train <- my_penguins %>%
  dplyr::select(body_mass_g,
                flipper_length_mm,
                bill_length_mm,
                bill_depth_mm)
my_cl <- my_penguins$species

test_that("my_knn_cv produces correct output type", {
  expect_type(my_knn_cv(train = my_train,
                        cl = my_cl,
                        k_nn = 3,
                        k_cv = 5),
              "list")
  expect_equal(length(my_knn_cv(train = my_train,
                                cl = my_cl,
                                k_nn = 3,
                                k_cv = 5)),
               2)
})