tests/testthat/test_str_eq_binomial.r

In simDAG: Simulate Data from a DAG and Associated Node Information

test_that("unspecified node", {
  test_node <- node("A", type="binomial", parents=c("C", "D"))
  expect_equal(str_eq_binomial(test_node), "A ~ Bernoulli(logit())")
})

test_that("one parent", {
  test_node <- node("A", type="binomial", parents="C", betas=1.2,
                    intercept=-12)
  expect_equal(str_eq_binomial(test_node), "A ~ Bernoulli(logit(-12 + 1.2*C))")
})

test_that("one parent with return_prob", {
  test_node <- node("A", type="binomial", parents="C", betas=1.2,
                    intercept=-12, return_prob=TRUE)
  expect_equal(str_eq_binomial(test_node), "A ~ logit(-12 + 1.2*C)")
})

test_that("multiple parents", {
  test_node <- node("A", type="binomial", parents=c("C", "D", "E"),
                    betas=c(1.2, 0.3, 7), intercept=-12)
  expect_equal(str_eq_binomial(test_node),
               "A ~ Bernoulli(logit(-12 + 1.2*C + 0.3*D + 7*E))")
})