tests/testthat/test_freeze.R

In redist: Simulation Methods for Legislative Redistricting

test_that("redist.freeze works", {
    data(fl25)
    data(fl25_adj)

    out <- redist.freeze(adj = fl25_adj, plan = plans_10[, 1], freeze_row =  (2 == plans_10[, 1]))
    expected <- c(1L, 2L, 2L, 2L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9L, 10L, 11L, 12L,
        2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L)

    expect_equal(out, expected)


    out <- redist.freeze(adj = fl25_adj, plan = plans_10[, 1],
        freeze_row =  (2 == plans_10[, 1]) | (3 == plans_10[, 1]))
    expected <- c(1L, 2L, 2L, 2L, 2L, 3L, 4L, 5L, 6L, 5L, 5L, 5L, 5L, 5L, 5L,
        2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L)

    expect_equal(out, expected)

    out <- redist.freeze(adj = fl25_adj,
        freeze_row =  (2 == plans_10[, 1]) | (3 == plans_10[, 1]))


})

test_that("freeze works", {
    data(fl25)

    fl25$plan <- plans_10[, 1]
    st_crs(fl25) <- 4269

    flmap <- redist_map(fl25, existing_plan = plan) %>% suppressMessages()
    out <- flmap %>% freeze(plan == 2, .data = .)

    expected <- c(1L, 2L, 2L, 2L, 2L, 3L, 4L, 5L, 6L, 7L, 8L, 9L, 10L, 11L, 12L,
        2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L, 2L)

    expect_equal(out, expected)
})