tests/testthat/test_ProcessStackInteractions.R

In raubreywhite/RAWmisc: R Convenience functions

context("Interactions ProcessStack")


test_that("simple interactions", {
  set.seed(4)
  x <- floor(1:10000 / 3500)
  interaction <- rep(c(0, 1), 5000)
  y <- 1 * x + 1 * x * interaction + rnorm(10000)

  data <- data.frame(x = as.factor(x), y, interaction)
  assign("data", data, envir = globalenv())

  stack <- RAWmisc::CreateStackSkeleton(n = length(1))
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "x:interaction"
  stack$confounders <- list(c("x", "interaction"))
  stack$data <- "data"
  stack$graphExposureScaleMultiply <- 2
  stack$graphExposureScaleAdd <- 5
  stack$graphReference <- 0
  stack$graphExposureLocationsLabels <- list(c(0.1, 0.5, 1))
  stack$graphExposureLocations <- list(c(0.1, 0.5, 1))
  stack$graphTitleMain <- "title"
  stack$graphTitleX <- "test"
  if (interactive()) stack$graphFileName <- "/git/test.png"

  a <- ProcessStack(stack = stack, i = 1)

  expect_equal(round(a$a_b * 10) / 10, c(1, 2))
})


test_that("simple interactions", {
  set.seed(4)
  x <- floor(1:10000 / 3500)
  interaction <- rep(c(0, 1), 5000)
  y <- 1 * x + 1 * x * interaction + rnorm(10000)

  data <- data.frame(x = x, y, interaction)
  assign("data", data, envir = globalenv())

  stack <- RAWmisc::CreateStackSkeleton(n = length(1))
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "x:interaction"
  stack$confounders <- list(c("x", "interaction"))
  stack$nameBase <- "x"
  stack$nameInteractions <- "x:interaction"
  stack$data <- "data"
  stack$graphExposureScaleMultiply <- 2
  stack$graphExposureScaleAdd <- 5
  stack$graphReference <- 0
  stack$graphExposureLocationsLabels <- list(c(0.1, 0.5, 1))
  stack$graphExposureLocations <- list(c(0.1, 0.5, 1))
  stack$graphTitleMain <- "title"
  stack$graphTitleX <- "test"
  if (interactive()) stack$graphFileName <- "/git/test.png"

  a <- ProcessStack(stack = stack, i = 1)

  expect_equal(round(a[exposure == "COMBINATION: x + x:interaction"]$a_b * 10) / 10, 2)
})

test_that("simple interactions with 3 interaction categories", {
  set.seed(4)
  x <- floor(1:9000 / 3500)
  interaction <- rep(c(0, 1, 2), 3000)
  y <- 1 * x + 1 * x * interaction + rnorm(9000)

  data <- data.frame(x = x, y, interaction = factor(interaction))
  assign("data", data, envir = globalenv())

  stack <- RAWmisc::CreateStackSkeleton(n = length(1))
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "x:interaction"
  stack$confounders <- list(c("x", "interaction"))
  stack$nameBase <- "x"
  stack$nameInteractions <- list(c("x:interaction1", "x:interaction2"))
  stack$data <- "data"
  stack$graphExposureScaleMultiply <- 2
  stack$graphExposureScaleAdd <- 5
  stack$graphReference <- 0
  stack$graphExposureLocationsLabels <- list(c(0.1, 0.5, 1))
  stack$graphExposureLocations <- list(c(0.1, 0.5, 1))
  stack$graphTitleMain <- "title"
  stack$graphTitleX <- "test"
  if (interactive()) stack$graphFileName <- "/git/test.png"

  a <- ProcessStack(stack = stack, i = 1)

  expect_equal(round(a[stringr::str_detect(exposure, "COMBINATION:")]$a_b * 10) / 10, c(1, 2, 3))
})

test_that("interactions with spline", {
  set.seed(4)
  x <- floor(1:10000 / 3500)
  interaction <- rep(c(0, 1), 5000)
  y <- 1 * x + 1 * x * interaction + rnorm(10000)

  data <- data.frame(x, y, interaction)
  assign("data", data, envir = globalenv())

  stack <- RAWmisc::CreateStackSkeleton(n = length(1))
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "splines::ns(x,df=2):interaction"
  stack$confounders <- list(c("splines::ns(x,df=2)", "interaction"))
  stack$data <- "data"
  stack$graphExposureScaleMultiply <- 2
  stack$graphExposureScaleAdd <- 5
  stack$graphReference <- 0
  stack$graphExposureLocationsLabels <- list(c(0.1, 0.5, 1))
  stack$graphExposureLocations <- list(c(0.1, 0.5, 1))
  stack$graphTitleMain <- "title"
  stack$graphTitleX <- "test"


  a <- ProcessStack(stack = stack, i = 1)

  expect_equal(a$a_b, NA)
})

test_that("spline interact and get out results", {
  set.seed(4)
  x <- 1:10000
  interaction <- rep(c(0, 1), 5000)
  y <- 1 * x + 1 * x * interaction + rnorm(10000)

  data <- data.frame(x, y, interaction)
  fit0 <- lm(y ~ splines::ns(x, df = 4) + interaction, data = data)
  fit1 <- lm(y ~ splines::ns(x, df = 4) + interaction + splines::ns(x, df = 4):interaction, data = data)

  stack <- RAWmisc::CreateStackSkeleton(n = 1)
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "splines::ns(x,df=4):interaction"
  stack$nameInteractions <- "interaction"
  stack$data <- "data"

  a <- ExtractFitsSplinesInteractions(
    fit0 = fit0,
    fit1 = fit1,
    fit1aic = fit1,
    stack = stack,
    i = 1,
    data = data,
    form = "y~splines::ns(x,df=4)+interaction+splines::ns(x,df=4):interaction"
  )

  expect_equal(round(a$b * 10) / 10, c(NA, 1, 2))
})

test_that("spline interact and get out results 2 ", {
  set.seed(4)
  x <- floor(1:10000 / 3500)
  interaction <- rep(c(0, 1), 5000)
  y <- 1 * x + 1 * x * interaction + rnorm(10000)

  data <- data.frame(x, y, interaction)
  assign("data", data, envir = globalenv())

  stack <- RAWmisc::CreateStackSkeleton(n = length(1))
  stack$regressionType <- "linear"
  stack$outcome <- "y"
  stack$exposure <- "splines::ns(x,df=2):interaction"
  stack$confounders <- list(c("splines::ns(x,df=2)", "interaction"))
  stack$nameInteractions <- "interaction"
  stack$data <- "data"
  stack$graphExposureScaleMultiply <- 2
  stack$graphExposureScaleAdd <- 5
  stack$graphReference <- 0
  stack$graphExposureLocationsLabels <- list(c(0.1, 0.5, 1))
  stack$graphExposureLocations <- list(c(0.1, 0.5, 1))
  stack$graphTitleMain <- "title"
  stack$graphTitleX <- "test"


  a <- ProcessStack(stack = stack, i = 1)

  expect_equal(round(a$a_b * 10) / 10, c(NA, 1, 2))
})