tests/testthat/test_poly_kclusters.R

In mobilizingcs/mobilizr: Functions for the Mobilize Introduction to Data Science course.

library(mobilizr)
context("poly and kclusters")

# Create models
poly_m <- lm(weight ~ poly(height, 2), data = cdc)

# Test polt
test_that("poly works", {
  expect_equal(unname(coef(poly_m)), c(91.77498, -112.35122, 57.94401), tolerance = 1e5)
})



# Test kclusters
set.seed(123)
test_that("kclusters works", {
  expect_equal(kclusters(wt_lbs ~ ht_inches, data = futbol, k = 2), c(1, 1, 1, 2, 1, 1, 1, 2, 1, 2, 2, 2, 1, 1, 2, 1, 1, 1, 1 ,1, 1, 1, 1, 1, 2, 2, 1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 2, 1, 1, 2, 1, 1, 2, 1, 2, 2, 2, 2, 1, 1, 2, 1, 2, 1, 1))
})

rm(poly_m)