tests/testthat/test-ndlst-methods.R

In DomBennett/treeman: Phylogenetic Tree Manipulation Class and Methods

# LIBS
library(treeman)
library(testthat)

# RUNNING
context('Testing \'ndlst-methods\'')
# slot
test_that('.getPrinds() works', {
  tree <- randTree(10)
  prinds <- treeman:::.getPrinds(tree@ndlst)
  i <- sample(1:tree@nall, 1)
  prind <- which(names(tree@ndlst) == tree@ndlst[[i]][['prid']])
  expect_equal(prinds[i], prind)
})
test_that('.getSltSpns() works', {
  tree <- randTree(10)
  spns <- treeman:::.getSltSpns(tree@ndlst)
  i <- sample(1:tree@nall, 1)
  spn <- tree@ndlst[[i]][['spn']]
  expect_equal(spns[i], spn)
})
test_that('.getTinds() works', {
  tree <- randTree(10)
  tinds <- treeman:::.getTinds(tree@ndlst)
  tids <- names(tree@ndlst)[tinds]
  expect_true(all(tree['tips'] %in% tids))
  expect_true(all(tids %in% tree['tips']))
})
# single node
test_that('.getNdSstrFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  # test
  rnd_tip <- sample(tree@tips, 1)
  res <- treeman:::.getNdSstrFrmLst(tree@ndlst, rnd_tip)
  expect_true(res %in% ids)
})
test_that('.getNdPridsFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # tests
  rnd_tip <- sample(tree@tips, 1)
  res <- treeman:::.getNdPridsFrmLst(ndlst, prinds=prinds,
                                     id=rnd_tip)
  expect_true(ndlst[[rnd_tip]]['prid'] %in% res)
  expect_true(tree@root %in% res)
  rnd_nd <- sample(tree@nds, 1)
  res <- treeman:::.getNdPridsFrmLst(ndlst, prinds=prinds,
                                     id=rnd_nd)
  expect_true(ndlst[[rnd_nd]]['prid'] %in% res)
  expect_true(tree@root %in% res)
  res <- treeman:::.getNdPridsFrmLst(ndlst, prinds=prinds,
                                     tree@root)
  expect_true(tree@root == res)
})
test_that('.getNdPrdstFrmLst([basic]) works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # tests
  res <- treeman:::.getNdPrdstsFrmLst(ndlst, prinds=prinds,
                                     id=tree@root)
  expect_that(res, equals(0))
  rnd_nd <- sample(tree@nds[tree@nds != tree@root], 1)
  res <- treeman:::.getNdPrdstsFrmLst(ndlst, prinds=prinds,
                                     id=rnd_nd)
  tree_age <- getAge(tree)
  expect_that(res, is_less_than(tree_age))
})
test_that('.getNdPtidsFrmLst([basic]) works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # tests
  rnd_tip <- sample(tree@tips, 1)
  res <- treeman:::.getNdPtidsFrmLst(ndlst, prinds=prinds,
                                     id=rnd_tip)
  expect_that(length(res), equals(0))
  intrnls <- tree@nds[tree@nds != tree@root]
  res <- treeman:::.getNdPtidsFrmLst(ndlst, prinds=prinds,
                                     id=tree@root)
  expect_true(all(intrnls %in% res))
  rnd_nd <- sample(intrnls, 1)
  res <- treeman:::.getNdPtidsFrmLst(ndlst, prinds=prinds,
                                     id=rnd_nd)
  expect_false(all(intrnls %in% res))
})
test_that('.getNdKidsFrmLst([basic]) works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  tinds <- tree@tinds
  # tests
  res <- treeman:::.getNdKidsFrmLst(ndlst, prinds=prinds,
                                    tinds=tinds, id=tree@root)
  expect_true(all(tree@tips %in% res))
  rnd_nd <- sample(tree@nds[tree@nds != tree@root], 1)
  res <- treeman:::.getNdKidsFrmLst(ndlst, prinds=prinds,
                                    tinds=tinds, id=rnd_nd)
  expect_false(all(tree@tips %in% res))
  rnd_tip <- sample(tree@tips, 1)
  res <- treeman:::.getNdKidsFrmLst(ndlst, prinds=prinds,
                                    tinds=tinds, id=rnd_tip)
  expect_that(length(res), equals(0))
})
test_that('.getNdPDFrmLst([basic]) works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # tests
  res <- treeman:::.getNdPDFrmLst(ndlst, prinds=prinds,
                                  id=tree@root)
  expect_that(res, equals(tree@pd))
  rnd_nd <- sample(tree@nds[tree@nds != tree@root], 1)
  res <- treeman:::.getNdPDFrmLst(ndlst, prinds=prinds,
                                  id=rnd_nd)
  expect_that(res, is_less_than(tree@pd))
})
# multiple nodes
test_that('.getNdsPtidsFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # test
  ptids <- treeman:::.getNdsPtidsFrmLst(tree@ndlst, ids=ids, prinds=prinds,
                                        parallel=FALSE, progress="none")
  expect_true(all(sapply(ptids[names(ptids) %in% tree['tips']],
                         length) == 0))
})
test_that('.getNdsPridsFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # test
  prids <- treeman:::.getNdsPridsFrmLst(tree@ndlst, ids=ids, prinds=prinds,
                                        parallel=FALSE, progress="none")
  expect_true(prids[[tree['root']]] == tree['root'])
})
test_that('.getNdsPDFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # test
  pds <- treeman:::.getNdsPDFrmLst(tree@ndlst, ids=ids, prinds,
                                   parallel=FALSE, progress="none")
  expect_true(sum(pds[names(pds) %in% tree['tips']]) == 0)
})
test_that('.getNdsKidsFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  tinds <- tree@tinds
  # test
  kids <- treeman:::.getNdsKidsFrmLst(tree@ndlst, ids=ids, prinds=prinds,
                                      tinds=tinds, parallel=FALSE, progress="none")
  expect_true(all(sapply(kids[names(kids) %in% tree['tips']],
                         length) == 0))
})
test_that('.getNdsPrdstsFrmLst() works', {
  # init
  tree <- randTree(100)
  ndlst <- tree@ndlst
  ids <- names(ndlst)
  prinds <- tree@prinds
  # test
  prdsts <- treeman:::.getNdsPrdstsFrmLst(tree@ndlst, ids=ids, prinds=prinds,
                                          parallel=FALSE, progress="none")
  tree_age <- getAge(tree)
  expect_true(all(prdsts <= tree_age))
})
# tree
test_that('.getTreeAge() works', {
  # init
  tree <- randTree(100)
  prinds <- tree@prinds
  tids <- tree@tips
  # test
  res <- treeman:::.getTreeAgeFrmLst(tree@ndlst, tids=tids,
                                     prinds=prinds, parallel=FALSE)
  expect_lt(res, tree['pd'])
  tree_age <- getAge(tree)
  expect_equal(res, tree_age)
})