# test_roommate.R
# test matching using Irving's Algorithm for the stable roommate problem

test_that("Stable roommate?", {
  set.seed(1)
  for (i in c(4, 8, 16, 32, 128, 256)) {
    p <- roommate.validate(utils = replicate(i, rnorm(i - 1)))
    results <- roommate(pref = p)
    expect_true(roommate.checkStability(pref = p, matching = results))
  }
})

test_that("Check preference orderings for one sided matching", {

  # generate error
  expect_error(roommate.validate())

  # generate error
  p <- matrix(c(0, 1, 2), nrow = 1, ncol = 3)
  expect_error(roommate.validate(pref = p))

  # generate error
  u <- matrix(c(3, 2, 1), nrow = 1, ncol = 3)
  expect_error(roommate.validate(utils = u))

  # generate warning
  u <- matrix(runif(6), nrow = 2, ncol = 3)
  p <- sortIndexOneSided(u)
  expect_warning(roommate.validate(utils = u, pref = p))

  # incomplete preferences
  p <- matrix(c(
    1, 0, 1,
    3, 2, 0
  ), nrow = 2, byrow = TRUE)
  expect_error(roommate.validate(pref = p))

  # check C++ vs R style indexing
  p <- matrix(c(
    1, 0, 1,
    2, 2, 0
  ), nrow = 2, byrow = TRUE)
  expect_identical(
    roommate.checkPreferences(p),
    roommate.checkPreferences(p + 1)
  )
})

test_that("Check if roommate can handle square matrices for cardinal utilities", {
  set.seed(2)
  utils1 <- matrix(rnorm(16), nrow = 4, ncol = 4)
  results1 <- roommate(utils = utils1)

  utils2 <- matrix(utils1[-c(1, 6, 11, 16)], nrow = 3, ncol = 4)
  results2 <- roommate(utils = utils2)

  expect_identical(results1, results2)
})

test_that("Check against previous results", {
  set.seed(3)

  # tests with even number of roommates
  pref <- matrix(c(2, 3, 4, 1, 3, 4, 1, 2, 4, 1, 2, 3), ncol = 4)
  results <- roommate(pref = pref)
  expect_identical(results, matrix(c(2, 1, 4, 3), ncol = 1))

  pref <- matrix(c(4, 3, 2, 1, 3, 4, 2, 1, 4, 1, 2, 3), ncol = 4)
  results <- roommate(pref = pref)
  expect_identical(results, matrix(c(4, 3, 2, 1), ncol = 1))

  # test with odd number of roommates
  pref <- matrix(c(2, 3, 1, 3, 1, 2), ncol = 3)
  results <- roommate(pref = pref)
  expect_identical(results, matrix(c(2, 1, NA), ncol = 1))
})