tests/testthat/test-compute.r2.R

In mobForest: Model Based Random Forest Analysis

# Create Data
response <- matrix(c(rep(0, 100), rep(10, 100)))
predictions <-
  matrix(nrow = 200, ncol = 3,
         data = c(rep(1, 100), rep(8, 100), rep(1, 100), rep(8, 100),
                  rep(1, 100), rep(8, 100)))


# Run Tests
test_that("response and prediction have same number of rows", {
  expect_error(compute.r2(response[1:10, ], predictions),
               "incorrect number of dimensions")
})

test_that("compute.r2 calculates the R2 value using the average prediction", {
  expect_equal(compute.r2(response, predictions), .9)
})