context("Test sim_ctd")

## test output format ##
test_that("sim_ctd working as expected", {
    ## skip on CRAN
    skip_on_cran()

    ## get data
    data(fake_outbreak)
    x <- fake_outbreak
    N <- length(x$ances)
    tTree <- data.frame(i = x$ances, j = 1:N)

    ## error check
    expect_error(sim_ctd(tTree = tTree, eps = 1.1, lambda = 0),
                 "eps and lambda must be probabilities")
    
    ## for eps = 1 / lambda = 0, there should be one
    ## contact per transmission pair
    ctd <- sim_ctd(tTree = tTree, eps = 1, lambda = 0)
    expect_equal(nrow(ctd), sum(!is.na(x$ances)))

    ## for eps = 1 / lambda = 1, ensure all possible
    ## contacts are reported
    ctd <- sim_ctd(tTree = tTree, eps = 1, lambda = 1)
    expect_equal(nrow(ctd), N*(N-1)/2)

})