tests/testthat/test-find_segments_to_plot.R

In cbaumbach/manyregs: manyregs --- Flexible and concise regression analysis

context("find_segments_to_plot")

source("setup.R")

model <- fitted_models[[1]]
model_summary <- summary(model)
expected_beta <- data.frame(
    x0 = c(1, 2), x1 = c(1, 2),
    y0 = c(0, with(model_summary, model_summary[variable == "x1" & level == "1", "lcl"])),
    y1 = c(0, with(model_summary, model_summary[variable == "x1" & level == "1", "ucl"])),
    midpoints = c(0, with(model_summary, model_summary[variable == "x1" & level == "1", "beta"])),
    stringsAsFactors = FALSE)
attr(expected_beta, "reference_line") <- 0

test_that("beta", {
    expected <- expected_beta
    actual <- find_segments_to_plot(model, type = "beta")
    expect_equal(actual, expected)
})

test_that("OR", {
    expected <- expected_beta
    expected$y0 <- exp(expected$y0)
    expected$y1 <- exp(expected$y1)
    expected$midpoints <- exp(expected$midpoints)
    attr(expected, "reference_line") <- 1
    actual <- find_segments_to_plot(model, type = "OR")
    expect_equal(actual, expected)
})