tests/testthat/test-RF_train.R

In DCAN-Labs/RFRF: Functional Random Forest in R

#This script is a unit test to ensure that RF_Train produces expected outputs

test_that("RF_train", {

  library("RFRF")
  #declare our expected input variables
  number_of_training_participants <- 10
  covariance_matrix <- diag(2)
  outcome_column <- 1
  means <- c(0,0)
  formula = y~x
  mtry <- 1
  nodesize <- 1
  model_type = "rfsrc"

  #create simulated data for the RF_train
  training_dataset <- simulate_data(number_of_training_participants,covariance_matrix,outcome_column,means)

  #train the RF model
  random_forest <- RF_train(training_dataset,formula,mtry,nodesize,model_type)

  #check the outputs
  expect_type(random_forest$call, "language")
  expect_type(random_forest$mtry, "double")
  expect_true(random_forest$mtry >= 0)
  expect_type(random_forest$nodesize, "double")
  expect_true(random_forest$nodesize >= 0)
  expect_type(random_forest$yvar, "double")
  expect_true(is.vector(random_forest$yvar))
  expect_type(random_forest$yvar.names, "character")
  expect_true(is.data.frame(random_forest$xvar))
  expect_type(random_forest$xvar.names, "character")
  expect_true(random_forest$forest$forest)
})