tests/testthat/test_boost_spline.R

In schalkdaniel/compboost: Efficient Component-Wise Boosting Implementation

context("Wrapper boostSpline works")

test_that("boostSpline function works", {
	expect_output({
	  mod = boostSplines(data = iris, target = "Sepal.Length", loss = LossQuadratic$new())
  })

	expect_length(mod$getBaselearnerNames(), ncol(iris)-1)
  expect_length(mod$getSelectedBaselearner(), 100)
  expect_length(mod$getInbagRisk(), 101)
  expect_length(mod$predict(), nrow(iris))

  expect_output(mod$train(150))

  expect_length(mod$getSelectedBaselearner(), 150)
  expect_length(mod$getInbagRisk(), 151)
  expect_length(mod$predict(), nrow(iris))
  expect_equal(mod$predict(), mod$predict(iris))
})