tests/testthat/test-optWeight.R

In benkeser/r2weight: Summarizing associations with multivariate outcomes

library(r2weight)

context("Testing optWeight function")

test_that("optWeight works as expected.", {
	# sample size
	n <- 100

	# set the seed
	set.seed(12345)

	# simulate nine covariate predictors
	x1 <- runif(n,0,4)
	x2 <- runif(n,0,4)
	x3 <- runif(n,0,4)

	# put all predictors in single data.frame
	X <- data.frame(x1=x1, x2=x2, x3=x3)

	# simulate three outcomes
	y1 <- x1 + 2*x2 + 4*x3 + rnorm(n, 0, 5)
	y2 <- x1 + 2*x2 + 4*x3 + rnorm(n, 0, 5)
	y3 <- x1 + 2*x2 + 4*x3 + rnorm(n, 0, 5)
	# put all outcomes in single data.frame
	Y <- data.frame(y1 = y1, y2 = y2, y3 = y3)

	# call optWeight using simple Super Learner library
	out1 <- optWeight(Y = Y, X = X, SL.library = c("SL.mean","SL.glm"))

	expect_true(length(out1$SL.weights) == 3)
	expect_true(sum(is.na(out1$SL.weights)) == 0)
	expect_true(length(out1$univariateR2) == 3)
})