tests/testthat/test-shortest.R

In beerda/hexmatrix: What the Package Does (One Line, Title Case)

test_that("shortest (generic test)", {
  d <- array(1, dim=c(4, 3, 6))
  d[2, 2, 5] <- 2
  d[3, 1, 5] <- 2
  d[4, 2, 6] <- 2

  expectedPrices <- matrix(c(1,1,2,4, 1, 0, 1, 2, 2, 1, 2, 2),
                           nrow=4, ncol=3, byrow=FALSE)
  expectedPaths <- matrix(c(6, 6, 6, 8, 6, 0, 6, 7, 10, 6, 10, 7),
                          nrow=4, ncol=3, byrow=FALSE)

  for (i in seq_len(12)) {
    p <- path(i, expectedPaths)[[1]]
    res <- shortest(source=6, target=i, dist=d)
    expect_equal(res$prices, rev(expectedPrices[p]))
    expect_equal(res$path, rev(p))
  }
})


test_that("shortest (simple path)", {
  d <- matrix(1, nrow=4, ncol=3)
  d[2, 2] <- 10
  d[3, 1] <- 10

  res <- shortest(7, 2, d)
  expect_equal(res$path, c(7, 10, 5, 1, 2))
  expect_equal(res$prices, c(0, 1, 2, 3, 4))

  d[2, 3] <- NA
  res <- shortest(7, 2, d)
  expect_equal(res$path, c(7, 6, 2))
  expect_equal(res$prices, c(0, 1, 11))
})


test_that("shortest (nonexistent path)", {
  d <- matrix(NA, nrow=4, ncol=3)
  d[1, ] <- 1

  res <- shortest(1, 11, d)
  expect_equal(res, NULL)
})

test_that("shortest (multilayer path)", {
  d <- array(100:144, dim=c(4, 3, 2, 6))
  t <- array(100:124, dim=c(4, 3, 2))

  d[2, 2, 1, 2] <- 1
  t[1, 2, 1] <- 1
  d[1, 2, 2, 4] <- 1
  d[2, 3, 2, 4] <- 1
  t[3, 3, 2] <- 1
  d[3, 3, 1, 5] <- 1

  res <- shortest(6, 12, d, t)
  expect_equal(res$path, c(6, 5, 17, 22, 23, 11, 12))
  expect_equal(res$prices, c(0, 1, 2, 3, 4, 5, 6))
})