tests/testthat/test-do_fill_missing_ids.R

In accessibility: Transport Accessibility Measures

tester <- function(access_df,
                   possible_combinations,
                   groups,
                   access_col = "access",
                   fill_value = 0) {
  do_fill_missing_ids(
    access_df,
    possible_combinations,
    groups,
    access_col,
    fill_value
  )
}

test_that("fill correctly with one id column", {
  access_df <- data.table::data.table(
    from_id = c("1", "2", "5"),
    access = c(10, 10, 10)
  )
  possible_combinations <- data.table::data.table(from_id = as.character(1:5))
  groups <- "from_id"

  result <- tester(access_df, possible_combinations, groups)
  data.table::setkey(result, NULL)
  expect_equal(
    result,
    data.table::data.table(
      from_id = as.character(1:5),
      access = c(10, 10, 0, 0, 10)
    )
  )
})

test_that("fill correctly with two id columns", {
  access_df <- data.table::data.table(
    from_id = c("1", "2", "5", "1"),
    mode = c("mode1", "mode1", "mode1", "mode2"),
    access = c(10, 10, 10, 20)
  )
  possible_combinations <- data.table::CJ(
    from_id = as.character(1:5),
    mode = c("mode1", "mode2")
  )
  groups <- c("from_id", "mode")

  result <- tester(access_df, possible_combinations, groups)
  data.table::setkey(result, NULL)
  expect_equal(
    result,
    data.table::data.table(
      from_id = rep(as.character(1:5), each = 2),
      mode = rep(c("mode1", "mode2"), times = 5),
      access = c(10, 20, 10, 0, 0, 0, 0, 0, 10, 0)
    )
  )
})

test_that("fills correct access_col", {
  access_df <- data.table::data.table(
    from_id = c("1", "2", "5"),
    oi = c(10, 10, 10)
  )
  possible_combinations <- data.table::data.table(from_id = as.character(1:5))
  groups <- "from_id"

  result <- tester(access_df, possible_combinations, groups, access_col = "oi")
  data.table::setkey(result, NULL)
  expect_equal(
    result,
    data.table::data.table(
      from_id = as.character(1:5),
      oi = c(10, 10, 0, 0, 10)
    )
  )
})

test_that("fills with correct fill_value", {
  access_df <- data.table::data.table(
    from_id = c("1", "2", "5"),
    access = c(10, 10, 10)
  )
  possible_combinations <- data.table::data.table(from_id = as.character(1:5))
  groups <- "from_id"

  result <- tester(access_df, possible_combinations, groups, fill_value = NA)
  data.table::setkey(result, NULL)
  expect_equal(
    result,
    data.table::data.table(
      from_id = as.character(1:5),
      access = c(10, 10, NA, NA, 10)
    )
  )
})