tests/testthat/test_input_checks_node_time_to_event.r

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

test_that("prob_fun args not specified", {

  prob_car_crash <- function(data, sim_time, base_p) {
    return(base_p + sim_time * 0.0001)
  }
  assign("prob_car_crash", value=prob_car_crash, envir=.GlobalEnv)

  dag <- empty_dag() +
    node_td("car_crash", type="time_to_event", prob_fun=prob_car_crash,
            event_duration=1)

  expect_error(sim_discrete_time(dag, n_sim=10, max_t=3))
})

test_that("prob_fun args not specified, but it has a default value", {

  prob_car_crash <- function(data, sim_time, base_p=0.1) {
    return(base_p + sim_time * 0.0001)
  }
  assign("prob_car_crash", value=prob_car_crash, envir=.GlobalEnv)

  dag <- empty_dag() +
    node_td("car_crash", type="time_to_event", prob_fun=prob_car_crash,
            event_duration=1)

  sim <- sim_discrete_time(dag, n_sim=10, max_t=3)
  expect_true(inherits(sim, "simDT"))
})