tests/testthat/test_kmeans.R

In binst: Data Preprocessing, Binning for Classification and Regression

library(binst)

test_that("kmeans breaks are close enough", {
  breaks_kmeans <- create_breaks(1:10)
  expect_length(breaks_kmeans, 2)
  expect_equal(breaks_kmeans > 3, c(TRUE, TRUE))
  expect_equal(breaks_kmeans < 8, c(TRUE, TRUE))
})

test_that("kmeans control parameters are correctly passed", {
  breaks_kmeans <- create_breaks(1:10, control=list(centers=4))
  expect_length(breaks_kmeans, 3)
  breaks_kmeans <- create_breaks(1:10, centers=4)
  expect_length(breaks_kmeans, 3)
})

test_that("kmeans control parameters with alternative algorithm can be used", {
  breaks_kmeans <- create_breaks(1:10, control=list(algorithm='Forgy'))
  expect_length(breaks_kmeans, 2)
})