tests/testthat/tests-branch.R

In leeper/mcode: Functions to Merge and Recode Across Multiple Variables

context("Test branch()")

test_that("Output as expected from numeric branch()", {
    a <- c(5L, 2L, 3L, 2L, 4L)
    b1 <- c(1L, 1L, 2L, 1L, 2L)
    b2 <- c(1L, 2L, 1L, 1L, 1L)
    expect_equivalent(branch(a), matrix(c(0,2,0,2,0, 0,0,3,0,0, 0,0,0,0,4, 5,0,0,0,0), nrow = 5))
    expect_equivalent(branch(a, b1), matrix(c(5,2,0,2,0, 0,0,3,0,4), nrow = 5))
    expect_equivalent(branch(a, list(b1, b2)), matrix(c(5,0,0,2,0, 0,0,3,0,4, 0,2,0,0,0, 0,0,0,0,0), nrow = 5))
})

test_that("Output as expected from character branch()", {
    a <- letters[1:5]
    b1 <- c(1L, 1L, 2L, 1L, 2L)
    b2 <- c(1L, 2L, 1L, 1L, 1L)
    expect_equivalent(branch(a), matrix(c("a",0,0,0,0, 0,"b",0,0,0, 0,0,"c",0,0, 0,0,0,"d",0, 0,0,0,0,"e"), nrow = 5))
    expect_equivalent(branch(a, b1), matrix(c("a","b",0,"d",0, 0,0,"c",0,"e"), nrow = 5))
    expect_equivalent(branch(a, list(b1, b2)), matrix(c("a",0,0,"d",0, 0,0,"c",0,"e", 0,"b",0,0,0, 0,0,0,0,0), nrow = 5))
})

test_that("Errors as expected from branch()", {
    expect_error(branch(1:5, .fill = c(1,2)))
    expect_error(branch(1:5, .fill = "foo"))
})

context("Test unbranch()")

test_that("Output as expected from unbranch()", {
    x <- c(NA, 2, 3,NA,NA, 6,NA,NA,NA,10)
    y <- c(NA,NA,NA,14,NA,NA,17,18,19,NA)
    z <- c(NA,NA,NA,NA,25,NA,NA,NA,NA,NA)
    
    expect_equal(unbranch(x), x)
    expect_equal(unbranch(x,y, .ignore = NA, .fill = NA), mergeNA(x,y))
    expect_equal(unbranch(x,y, .ignore = NA, .fill = NA), c(NA, 2, 3, 14, NA, 6, 17, 18, 19, 10))
    expect_equal(unbranch(x,y, .ignore = NA, .fill = 99), c(99, 2, 3, 14, 99, 6, 17, 18, 19, 10))
    expect_equal(unbranch(cbind(x,y), .ignore = NA, .fill = NA), c(NA, 2, 3, 14, NA, 6, 17, 18, 19, 10))
    expect_equal(unbranch(cbind.data.frame(x,y), .ignore = NA, .fill = NA), c(NA, 2, 3, 14, NA, 6, 17, 18, 19, 10))
    
    expect_equal(unbranch(x,y,z, .ignore = NA, .fill = NA), c(NA, 2, 3, 14, 25, 6, 17, 18, 19, 10))
    expect_equal(unbranch(x,y,z, .ignore = NA, .fill = NA), mergeNA(x,y,z))
})

test_that("Errors as expected from branch()", {
    expect_error(unbranch(1:5, .fill = c(99,100)))
})

test_that("Non-mutually exclusive ignore values errors unbranch()", {
    x <- c(NA,1,2,NA,NA,2,NA,NA,NA,1)
    w <- c(NA,3,3,NA,1,NA,NA,NA,NA,NA)
    expect_error(unbranch(x,w, .ignore = NA))
    expect_equal(unbranch(x,w, .ignore = c(3,NA)), c(0, 1, 2, 0, 1, 2, 0, 0, 0, 1))
})

test_that("Test branch(unbranch())", {
    a <- c(5L, 2L, 3L, 2L, 4L)
    expect_equivalent(unbranch(branch(a, .fill = NA), .ignore = NA, .fill = NA), a)
    expect_equivalent(unbranch(branch(a, .fill = 0), .ignore = 0, .fill = 0), a)
})

test_that("Test unbranch(branch())", {
    x <- c(NA, 2, 3,NA,NA, 6,NA,NA,NA,10)
    y <- c(NA,NA,NA,14,NA,NA,17,18,19,NA)
    #branch(unbranch(x,y, .ignore = NA), list(is.na(x), is.na(y)), .fill = NA)
})

test_that("Factors coerced correctly in unbranch()", {
    # need to write this
})

test_that("Unbranch a branched matrix", {
    # need to write this
})