tests/testthat/test-01-kernels.R

In haziqj/iprior: Regression Modelling using I-Priors

context("Kernel matrices")

test_that("Linear kernel", {

  x <- 1:3
  y <- matrix(1:6, ncol = 2)
  y.cen <- scale(y, scale = FALSE)
  expect_true(is.kern_linear(kern_canonical(x)))
  expect_true(is.kern_linear(kern_canonical(y)))
  expect_error(kern_canonical(x, y))
  expect_error(kern_canonical(x, y, centre = FALSE))

  res1 <- kern_canonical(y, y)
  res2 <- kern_canonical(y.cen, y.cen, centre = FALSE)
  res3 <- kern_canonical(y.cen, y.cen)
  res4 <- kern_canonical(y)
  expect_true(identical(res1, res2, res3, res4))

})

test_that("Pearson kernel", {

  x <- factor(1:3)
  y <- factor(1:2)
  z <- factor(4)
  expect_true(is.kern_pearson(kern_pearson(x)))
  expect_true(is.kern_pearson(kern_pearson(x, y)))
  expect_error(kernel(x, z))
  expect_equivalent(kern_pearson(x), kern_pearson(x, x))

})

test_that("Pearson must take in factors",{

  expect_warning(kern_pearson(1:100))

})

test_that("fBm kernel", {

  x <- 1:3
  y <- matrix(1:6, ncol = 2)
  expect_true(is.kern_fbm(kern_fbm(x)))
  expect_true(is.kern_fbm(kern_fbm(y)))
  expect_error(kern_fbm(x, y))

  # fBm-1 equals canonical
  res1 <- kern_canonical(y, y)
  res2 <- kern_fbm(y, y, gamma = 1)
  res3 <- kern_fbm(y, gamma = 1)
  expect_equivalent(res1, res2)
  expect_equivalent(res2, res3)
  expect_equivalent(res1, res3)

  # fBm-1 equals canonical
  res1 <- kern_canonical(y, y, centre = FALSE)
  res2 <- kern_fbm(y, y, gamma = 1, centre = FALSE)
  res3 <- kern_fbm(y, gamma = 1, centre = FALSE)
  expect_equivalent(res1, res2)
  expect_equivalent(res2, res3)
  expect_equivalent(res1, res3)

})

test_that("SE kernel", {

  x <- 1:3
  y <- matrix(1:6, ncol = 2)
  expect_true(is.kern_se(kern_se(x, l = 2)))
  expect_true(is.kern_se(kern_se(y)))
  expect_error(kern_se(x, y))

  res1 <- kern_se(y, y)
  res2 <- kern_se(y)
  expect_equivalent(res1, res2)

})

test_that("Polynomial kernel", {

  x <- 1:3
  y <- matrix(1:6, ncol = 2)
  expect_true(is.kern_poly(kern_poly(x, d = 3)))
  expect_true(is.kern_poly(kern_poly(y, c = 3)))
  expect_error(kern_poly(x, y))
  expect_error(kern_poly(x, d = -1))
  expect_error(kern_poly(x, c = -1))

  res1 <- kern_poly(y, y, lam.poly = 3)
  res2 <- kern_poly(y, lam.poly = 3)
  expect_equivalent(res1, res2)

})

test_that("Non-integer value for polynomial degree", {

  expect_error(kern_poly(1:3, d = pi))

})