# test_lookup.R dat = as.data.frame(state.x77) dat$statename = rownames(dat) #lookup(state.name, dat$statename, dat$Frost) #vlookup(state.name, dat, "statename", "Frost") test_that("lookup", { expect_equal( lookup(state.name, dat$statename, dat$Frost), c(20, 152, 15, 65, 20, 166, 139, 103, 11, 60, 0, 126, 127, 122, 140, 114, 95, 12, 161, 101, 103, 125, 160, 50, 108, 155, 139, 188, 174, 115, 120, 82, 80, 186, 124, 82, 44, 126, 127, 65, 172, 70, 35, 137, 168, 85, 32, 100, 149, 173)) expect_equal( vlookup(state.name, dat, "statename", "Frost"), c(20, 152, 15, 65, 20, 166, 139, 103, 11, 60, 0, 126, 127, 122, 140, 114, 95, 12, 161, 101, 103, 125, 160, 50, 108, 155, 139, 188, 174, 115, 120, 82, 80, 186, 124, 82, 44, 126, 127, 65, 172, 70, 35, 137, 168, 85, 32, 100, 149, 173)) }) test_that("missing key", { A <- factor(c("A","E","F")) B <- factor(c("E","F","G")) v <- c(4,2,0) dat <- data.frame(trt = B, val=v) expect_equal( lookup(A,B,v), c(NA, 4, 2) ) expect_equal( vlookup(A,dat, "trt", "val"), c(NA, 4, 2) ) })