Nothing
context("test-check_augment_function")
base_aug <- function(data = NULL, newdata = NULL) {
df <- if (!is.null(newdata)) newdata else data
if (has_rownames(df))
df$.rownames <- rownames(df)
df$.fitted <- 1:nrow(df)
df$.resid <- 1:nrow(df) + 1
df
}
data_only_aug <- function(model, data) {
as_tibble(base_aug(data))
}
consistent_aug <- function(model, data = NULL, newdata = NULL) {
if (is.null(data) && is.null(newdata))
stop("Must specify either `data` or `newdata` argument.", call. = FALSE)
as_tibble(base_aug(data, newdata))
}
test_that("strict = FALSE", {
expect_error(
check_augment_function(
aug = data_only_aug,
model = NULL,
data = NULL,
newdata = NULL,
strict = FALSE
),
"Must pass `data` argument as augment method accepts data argument."
)
expect_error(
check_augment_function(
aug = consistent_aug,
model = NULL,
data = iris,
newdata = NULL,
strict = FALSE
),
"Must pass `newdata` argument as augment method accepts newdata argument."
)
expect_silent(
check_augment_function(
aug = consistent_aug,
model = NULL,
data = iris,
newdata = iris,
strict = FALSE
)
)
})
test_that("strict = TRUE", {
expect_silent(
check_augment_function(
aug = consistent_aug,
model = NULL,
data = iris,
newdata = iris,
strict = TRUE
)
)
})
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.