test_that("transform_response works with factors", { expected_species <- iris %>% pull(Species) %>% fct_collapse("0" = c("virginica", "versicolor"), "1" = "setosa") %>% fct_relevel("0", "1") transformed_species <- transform_response(iris$Species) expect_equal(transformed_species, expected_species) }) test_that("transform_response works with integers", { expected_species <- iris %>% pull(Species) %>% fct_collapse("0" = c("virginica", "versicolor"), "1" = "setosa") %>% fct_relevel("0", "1") transformed_species <- dplyr::case_when( iris$Species == "setosa" ~ 1L, iris$Species == "virginica" ~ 2L, iris$Species == "versicolor" ~ 3L ) %>% transform_response() expect_equal(transformed_species, expected_species) }) test_that("transform_response works with characters", { expected_species <- iris %>% pull(Species) %>% fct_collapse("0" = c("virginica", "versicolor"), "1" = "setosa") %>% fct_relevel("0", "1") transformed_species <- transform_response(as.character(iris$Species)) expect_equal(transformed_species, expected_species) }) test_that("tranform_response works with fct and .condition is selected", { test_iris <- create_iris_df() expected_species <- test_iris$Species_bin_fct_virg transformed_species <- transform_response(test_iris$Species, "virginica") expect_equal(transformed_species, expected_species) }) test_that("tranform_response works with int and .condition is selected", { test_iris <- create_iris_df() expected_species <- test_iris$Species_bin_fct_virg transformed_species <- transform_response(test_iris$Species_int, 3) expect_equal(transformed_species, expected_species) }) test_that("tranform_response works with chr and .condition is selected", { test_iris <- create_iris_df() expected_species <- test_iris$Species_bin_fct_virg transformed_species <- transform_response(test_iris$Species_chr, "virginica") expect_equal(transformed_species, expected_species) }) test_that("as_response works with fct(0,1) and .condition is selected", { test_iris <- create_iris_df() expected_species <- test_iris$Species_bin_fct_inv transformed_species <- as_response( x = test_iris$Species_bin_fct, .condition = "0" ) expect_equal(transformed_species, expected_species) })