library(testthat)
library(aricode)

test_that("Testing supposedly handled input types", {
  c1 <- rep(1:3, each = 3)
  c2 <- rep(4:6, each = 3)

  cat("\n-numeric type-")
  c1.numeric <- as.numeric(c1)
  c2.numeric <- as.numeric(c2)
  ref.object <-
    list(
      spMat = NULL,
      levels = list(c1 = unique(c1.numeric), c2 = unique(c2.numeric)),
      nij = rep(3, 3), ni. = rep(3, 3), n.j = rep(3, 3),
      pair_c1 = 0:2, pair_c2 = 0:2
    )
  expect_that(sort_pairs(c1.numeric, c2.numeric), equals(ref.object))

  cat("\n-integer type-")
  c1.integer <- as.integer(c1)
  c2.integer <- as.integer(c2)
  ref.object <- list(spMat = NULL, levels = list(c1 = unique(c1.integer), c2 = unique(c2.integer)), nij = rep(3, 3), ni. = rep(3, 3), n.j = rep(3, 3), pair_c1 = 0:2, pair_c2 = 0:2)
  expect_that(sort_pairs(c1.integer, c2.integer), equals(ref.object))

  cat("\n-character type-")
  c1.char <- as.character(c1)
  c2.char <- as.character(c2)
  ref.object <- list(spMat = NULL, levels = list(c1 = as.integer(unique(c1.char)), c2 = as.integer(unique(c2.char))), nij = rep(3, 3), ni. = rep(3, 3), n.j = rep(3, 3), pair_c1 = 0:2, pair_c2 = 0:2)
  expect_that(sort_pairs(c1.char, c2.char), equals(ref.object))

  cat("\n-factor type-")
  c1.factor <- as.factor(c1.char)
  c2.factor <- as.factor(c2.char)
  ref.object <- list(spMat = NULL, levels = list(c1 = levels(c1.factor), c2 = levels(c2.factor)), nij = rep(3, 3), ni. = rep(3, 3), n.j = rep(3, 3), pair_c1 = 0:2, pair_c2 = 0:2)
  expect_that(sort_pairs(c1.factor, c2.factor), equals(ref.object))

  cat("\n-different types-")
  ref.object <-
    list(
      spMat = NULL,
      levels = list(
        c1 = as.integer(unique(c1.char)),
        c2 = unique(as.integer(c2.factor))
      ),
      nij = rep(3, 3), ni. = rep(3, 3),
      n.j = rep(3, 3), pair_c1 = 0:2, pair_c2 = 0:2
    )
  expect_that(sort_pairs(c1.char, c2.factor), equals(ref.object))
})

test_that("Testing error for not handled input types", {
  c1 <- rep(1:3, each = 3) * 1.5
  c2 <- rep(4:6, each = 3) * 2.7

  cat("\n-NA-")
  c2NA <- c2
  c2NA[1] <- NA
  expect_error(sort_pairs(c1, c2NA))

  cat("\n-different sizes-")
  expect_error(sort_pairs(c1, c2[-1]))

  cat("\n-list type-")
  expect_error(sort_pairs(as.list(c1), as.list(c2)))

  cat("\n-matrix type-")
  expect_error(sort_pairs(as.matrix(c1), as.matrix(c2)))

  cat("\n-data.frame type-")
  expect_error(sort_pairs(as.data.frame(c1), as.data.frame(c2)))
})