context("TaskSurv")
test_that("Task duplicates rows", {
task = tsk("lung")
expect_task_surv(task)
expect_set_equal(extract_vars(task$formula())$rhs, ".")
})
test_that("right censoring", {
expect_silent({
task = TaskSurv$new("right", backend = survival::rats, time = "time", event = "status")
})
expect_equal(task$censtype, "right")
expect_equal(task$formula(), as.formula(Surv(time, status, type = "right") ~ .))
})
test_that("left censoring", {
expect_silent({
task = TaskSurv$new("left", backend = survival::rats, time = "time", event = "status", type = "left")
})
expect_equal(task$censtype, "left")
expect_equal(task$formula(), as.formula(Surv(time, status, type = "left") ~ .))
})
test_that("interval2 censoring", {
expect_silent({
task = TaskSurv$new(
id = "interval2_censored", backend = survival::bladder2[, -c(1, 7)],
time = "start", time2 = "stop", type = "interval2")
})
expect_equal(task$censtype, "interval2")
expect_equal(ncol(task$truth()), 3)
expect_equal(task$formula(), as.formula(Surv(time = start, time2 = stop, type = "interval2") ~ .))
})
test_that("interval censoring", {
expect_silent({
task = TaskSurv$new(
id = "interval_censored", backend = survival::bladder1,
time = "start", time2 = "stop", event = "status", type = "interval")
})
expect_equal(task$censtype, "interval")
expect_equal(ncol(task$truth()), 3)
expect_equal(task$formula(), as.formula(Surv(start, stop, status, type = "interval") ~ .))
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.