tests/testthat/test-mglm_spd_checkpoint.R

In mrparker909/MGLMRiem: Multivariate Generalized Linear Models on Riemannian Manifolds

test_that("checkpointing works", {
  
  # create test data
  num = 5
  X = 1:num
  Y = array(0, dim=c(2,2,num))
  
  set.seed(123)
  for(i in 1:num) {
    Y[,,i] = randspd_FAST(2)
  }
  
  # create checkpoint file
  res1=mglm_spd(X = as.matrix(t(X)), Y = Y, maxiter = 200, enableCheckpoint = T)
  expect_equal(F, res1$converged)
  
  # load checkpoint file
  load("checkpoint.rda")
  
  # delete checkpoint file
  file.remove("checkpoint.rda")
  
  # run from checkpoint file
  res2=mglm_spd_checkpoint(checkpoint, maxiter = 100, enableCheckpoint=F)
  expect_equal(T, res2$converged)
  
  
  # create checkpoint file
  res1=mglm_spd(X = as.matrix(t(X)), Y = Y, maxiter = 200, enableCheckpoint = T, checkpointPath = "./output/", Memory = 5)
  expect_equal(F, res1$converged)
  
  # load checkpoint file
  load("./output/checkpoint.rda")
  
  # delete checkpoint file
  file.remove("./output/checkpoint.rda")
  
  # run from checkpoint file
  res2=mglm_spd_checkpoint(checkpoint, maxiter = 50, enableCheckpoint=T, checkpointPath = "./output/", Memory = 5)
  expect_equal(T, res2$converged)
  
  
})