tests/testthat/test-dplyr-S3methods.R

In adibender/pam: Functions and Utilities for fitting Piece-wise Exponential (Additive) models

context("Dplyr methods for specific classes")

test_that("ped class is preserved", {
	data("veteran", package = "survival")
	veteran <- dplyr::slice(veteran, 1:2)
	ped <- split_data(Surv(time, status)~ trt + age, cut=c(0, 100, 400), 
		data = veteran, id = "id")

	expect_is(filter(ped, id == 1), "ped")
	expect_is(slice(ped, 1), "ped")
	expect_is(arrange(ped, desc(id)), "ped")
	expect_is(select(ped, id), "ped")
	expect_is(rename(ped, ID=id), "ped")
	expect_is(mutate(ped, id = id + 1), "ped")
	expect_is(transmute(ped, id = id + 1), "ped")
	expect_is(sample_n(ped, 1), "ped")
	expect_is(sample_frac(ped, 0.5), "ped")
})