context("Test mfpca")

test_that("all mfpca functions work on the DTI example", {
  skip_on_cran()
  
  data(DTI)
  DTI <- subset(DTI, Nscans < 6)  ## example where all subjects have 6 or fewer visits
  id <- DTI$ID
  Y <- DTI$cca
  
  mfpca.sc.DTI <- mfpca.sc(Y = Y, id = id, twoway = TRUE)
  mfpca.face.DTI <- mfpca.face(Y = Y, id = id, twoway = TRUE)
  
  expect_equal(dim(mfpca.face.DTI$Xhat)[1], dim(mfpca.sc.DTI$Yhat)[1])
  expect_equal(dim(mfpca.face.DTI$Xhat.subject)[1], dim(mfpca.sc.DTI$Yhat.subject)[1])
  expect_equal(dim(mfpca.face.DTI$scores$level1)[1], dim(mfpca.sc.DTI$scores$level1)[1])
  expect_equal(dim(mfpca.face.DTI$scores$level2)[1], dim(mfpca.sc.DTI$scores$level2)[1])
})

test_that("mfpca.face options work", {
  skip_on_cran()
  
  data(DTI)
  DTI <- subset(DTI, Nscans < 6)  ## example where all subjects have 6 or fewer visits
  id <- DTI$ID
  Y <- DTI$cca
  mfpca.base <- mfpca.face(Y = Y, id = id) 
  
  # visit argument
  mfpca.visit <- mfpca.face(Y = Y, id = id, visit = DTI$visit) 
  expect_equal(mfpca.base$npc$level1, mfpca.visit$npc$level1)
  
  # weight argument
  mfpca.weight <- mfpca.face(Y = Y, id = id, weight = "subj") 
  expect_equal(dim(mfpca.base$scores$level1)[1], dim(mfpca.weight$scores$level1)[1])
  
  # pve argument
  mfpca.pve <- mfpca.face(Y = Y, id = id, pve = 0.95) 
  expect_equal(dim(mfpca.base$scores$level1)[1], dim(mfpca.pve$scores$level1)[1])
  
  # npc argument
  mfpca.npc <- mfpca.face(Y = Y, id = id, npc = 5) 
  expect_equal(mfpca.npc$npc$level1, 5)
  expect_equal(mfpca.npc$npc$level2, 5)
})