library(testthat)
library(AutoEDA)

#==========================================================
# print.ClusterResult()
#==========================================================

test_that("print.ClusterResult works", {

  cl <- cluster_analysis(iris)

  expect_no_error(
    print(cl)
  )

})

test_that("print.ClusterResult returns object invisibly", {

  cl <- cluster_analysis(iris)

  x <- print(cl)

  expect_identical(
    x,
    cl
  )

})

test_that("print.ClusterResult works with two clusters", {

  cl <- cluster_analysis(
    iris,
    centers = 2
  )

  expect_no_error(
    print(cl)
  )

})

test_that("print.ClusterResult reports correct number of clusters", {

  cl <- cluster_analysis(
    iris,
    centers = 3
  )

  expect_equal(
    length(unique(cl$cluster)),
    3
  )

})

#==========================================================
# Invalid input
#==========================================================

test_that("print.ClusterResult rejects invalid object", {

  expect_error(

    AutoEDA:::print.ClusterResult(
      iris
    )

  )

})

test_that("print.ClusterResult rejects NULL", {

  expect_error(

    AutoEDA:::print.ClusterResult(
      NULL
    )

  )

})