inst/tests/test-weight_matrix.r

In bayou: Bayesian Fitting of Ornstein-Uhlenbeck Models to Phylogenies

#context("weight matrix can be calculated")
testthat::test_that("weight matrix can be calculated", {
  library(bayou)
  data(chelonia, package="geiger")
  tree <- chelonia$phy
  dat <- chelonia$dat
  cache <- bayou:::.prepare.ou.univariate(tree, dat)
  pars <- list(alpha=0.01, sig2=1, k=3, theta=c(3,4,5), ntheta=3, sb=c(411, 400, 47), loc=c(25, 17, 33), t2=c(2,3,3))
  TotExp <- exp(-cache$height*pars$alpha)
  stree <- pars2simmap(pars, tree)
  sW <- simmapW(stree$tree, pars)
  testthat::expect_equal(apply(sW, 1, sum), rep(1,226))
  bW <- bayou:::C_weightmatrix(cache, pars)$W
  testthat::expect_equal(bW, sW)
  testthat::expect_equal(apply(bW,2,sum), c(79.58365, 117.46516, 28.95119), tolerance=0.0001)
})