tests/testthat/test-gets-old-FSKA.R

In mssm: Multivariate State Space Models

context("Test versus old results for kernel methods")

test_that("'FSKA' gives the same", {
  set.seed(90638579)
  n <- 5000L
  p <- 4L
  X <- matrix(rnorm(n * p), nrow = p)
  ws <- exp(rnorm(n))
  ws <- ws / sum(ws)

  exact <- naive(X = X, ws = ws, Y = X, n_threads = 2L)
  expect_known_value(exact, "FSKA-old-res-small-eps-exact.RDS")

  o1 <- FSKA(X = X, ws = ws, Y = X, N_min = 10L, eps = 1e-3,
                    n_threads = 1L)
  # all.equal(o1, exact)
  # o1 <- o1 / sum(o1)
  # exact <- exact / sum(exact)
  # all.equal(o1, exact)
  expect_known_value(o1, "FSKA-old-res-small-eps.RDS")

  o2 <- FSKA(X = X, ws = ws, Y = X, N_min = 10L, eps = 1e-3,
                     n_threads = 1L)
  expect_known_value(o2, "FSKA-old-res-small-eps.RDS")
  expect_equal(o1, o2)

  o3 <- FSKA(X = X, ws = ws, Y = X, N_min = 10L, eps = 1e-12,
                    n_threads = 1L)
  expect_equal(o3, exact)
})

test_that("we get the same KD-tree", {
  set.seed(90638579)
  n <- 5000L
  p <- 4L
  X <- matrix(rnorm(n * p), nrow = p)

  t1 <- test_KD_note(X, 10L)
  t2 <- test_KD_note(X, 10L)
  expect_equal(t1, t2)

  # we cannot directly test `t1`s indices' order as they may differ within
  # each node as the C++ partial sorting algorithm yields all but one element
  # in an unspecified. Thus, we have to sort them within each leaf
  leaf <- unlist(mapply(rep, x = seq_along(t1$n_elems), times = t1$n_elems))
  stopifnot(length(leaf) == n)
  t1$indices <- unlist(tapply(t1$indices, leaf, sort))

  expect_known_value(t1, "KD-tree.RDS")
})