tests/testthat/test_info_functions.R

In stagedtrees: Staged Event Trees

context("info functions")

test_that("test that sevt_nvar return number of var", {
  ev <-
    sevt.list(x = list(
      A = c(1, 2),
      B = c("x", "y"),
      C = c("3", "4")
    ))
  expect_equal(stagedtrees:::sevt_nvar(ev), 3)

  ev <- sevt.list(x = list(
    A = c(1, 2)
  ))
  expect_equal(stagedtrees:::sevt_nvar(ev), 1)
})


test_that("get_stage", {
  ev <-
    sevt(list(
      A = c(1, 2),
      B = c("x", "y"),
      C = c("3", "4")
    ))
  expect_equal(get_stage(ev, c("1", "x")), expected = "1")
  expect_equal(get_stage(ev, c("1", "y")), expected = "1")
  expect_equal(get_stage(ev, c("2")), expected = "1")
})


test_that("get_path", {
  ev <-
    sevt(list(
      A = c("1", "2"),
      B = c("x", "y"),
      C = c("3", "4")
    ))
  expect_equivalent(get_stage(ev, get_path(ev, "B", "1")[1, ]), "1")
  expect_equivalent(get_stage(ev, get_path(ev, "C", "1")[1, ]), "1")
  expect_equivalent(get_stage(ev, get_path(ev, "C", "1")[2, ]), "1")
})