tests/testthat/test_train_model.R

In rebordao/modellatoR: Automatic deployment of a statistical model; from data to reporting.

library(modellatoR)
context("Functional Training")

# Creates a modellatoR's project, with its folder structure
modellatoR::create_project(working_dir = tempdir(),
                           project_name = "test",
                           project_minimal = T)

# Setups project
data(boston)
modellatoR::setup_project(data = boston, method_id = "rf", out_var = "medv")

# Loads params
load(file.path(tempdir(), "test", "config", "params.RData"))

# Trains model
cutoff <- round(.8 * nrow(boston))
model <- modellatoR::train_model(
  trainset = subset(boston, 1:nrow(boston) < cutoff),
  testset = subset(boston, 1:nrow(boston) >= cutoff),
  params = params)

test_that("training doesn't output a valid model", {
  expect_is(model, "randomForest")
})

# Removes folder structure
unlink(file.path(tempdir(), "test"), recursive = T)