tests/testthat/test-pla.R

In prinvars: Principal Variables

obj <- pla(
  x=USArrests,
  cor=FALSE,
)

test_that("pla", {
  blocks <- list()
  blocks[[1]] <- new("Block",
    features=c("Murder"),
    explained_variance=0.006,
    ev_influenced=c(4))
  blocks[[2]] <- new("Block",
    features=c("Assault"),
    explained_variance=0.966,
    ev_influenced=c(1))
  blocks[[3]] <- new("Block",
    features=c("UrbanPop"),
    explained_variance=0.028,
    ev_influenced=c(2))
  blocks[[4]] <- new("Block",
    features=c("Rape"),
    explained_variance=0.0008,
    ev_influenced=c(3))
  expect_equal(obj$blocks, blocks, tolerance=1e-2)
})

test_that("pla.keep_blocks", {
  result <- pla.keep_blocks(object=obj, blocks=c(1, 3))
  expected <- scale(USArrests[c(1, 3)], center=TRUE, scale=FALSE)
  attr(expected, "scaled:center") <- NULL

  expect_equal(result$x, expected)
})

test_that("pla.drop_blocks", {
  result <- pla.drop_blocks(object=obj, blocks=c(1, 3))
  expected <- scale(USArrests[c(2, 4)], center=TRUE, scale=FALSE)
  attr(expected, "scaled:center") <- NULL

  expect_equal(result$x, expected)
})