# test cross-sectional reconciliation
if (require(testthat)) {
  A <- matrix(c(1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1, 1), 3, byrow = TRUE)
  set.seed(123)
  res1 <- matrix(rnorm(100 * sum(dim(A))), 100, sum(dim(A)))
  base1 <- t(rnorm(sum(dim(A)), 1))
  res2 <- matrix(rnorm(100 * sum(dim(A))), 100, sum(dim(A)))
  base2 <- t(rnorm(sum(dim(A)), 1))
  res3 <- matrix(rnorm(100 * sum(dim(A))), 100, sum(dim(A)))
  base3 <- t(rnorm(sum(dim(A)), 1))
  C <- cbind(diag(NROW(A)), -A)
  comb <- "shr"
  base <- list(base1, base2, base3)
  res <- list(res1, res2, res3)
  test_that("Optimal cross-sectional coherent combination", {
    r1 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "strc"
    )
    r2 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj"
    )
    r3 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "strc_osqp"
    )
    r4 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj_osqp"
    )
    r5 <- csocc(
      base = base,
      cons_mat = C,
      comb = comb,
      res = res,
      approach = "strc"
    )
    r6 <- csocc(
      base = base,
      cons_mat = C,
      comb = comb,
      res = res,
      approach = "proj"
    )

    expect_equal(r1, r2, ignore_attr = TRUE)
    expect_equal(r1, r3, ignore_attr = TRUE)
    expect_equal(r1, r4, ignore_attr = TRUE)
    expect_equal(r1, r5, ignore_attr = TRUE)
    expect_equal(r1, r6, ignore_attr = TRUE)
    expect_equal(max(abs(C %*% t(r1))), 0)
  })

  test_that("Covariance check", {
    for (i in c(
      "ols",
      "str",
      "wls",
      "shr",
      "shrbe",
      "shrbv",
      "sam",
      "sambe",
      "sambv"
    )) {
      expect_no_error({
        csocc(base = base, agg_mat = A, comb = i, res = res, approach = "proj")
      })
    }
  })

  base[[1]][1, NCOL(base1)] <- -10
  test_that("Optimal nonegative cross-sectional reconciliation", {
    r1 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "strc_osqp"
    )
    r2 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "proj_osqp"
    )
    r3 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "sntz"
    )

    expect_equal(r1, r2, ignore_attr = TRUE)
    expect_equal(max(abs(C %*% t(r1))), 0)
    expect_equal(max(abs(C %*% t(r3))), 0)
  })

  base[[1]][1, 1] <- NA
  base_err <- base
  base_err[[2]][1, 1] <- NA
  test_that("Optimal cross-sectional coherent combination with NA", {
    r1 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "strc"
    )
    r2 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj"
    )
    r3 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "strc_osqp"
    )
    r4 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj_osqp"
    )
    r5 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "strc_osqp"
    )
    r6 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "proj_osqp"
    )
    r7 <- csocc(
      base = base,
      agg_mat = A,
      comb = comb,
      res = res,
      approach = "proj",
      nn = "sntz"
    )

    expect_equal(r1, r2, ignore_attr = TRUE)
    expect_equal(r1, r3, ignore_attr = TRUE)
    expect_equal(r1, r4, ignore_attr = TRUE)
    expect_equal(r5, r6, ignore_attr = TRUE)
    expect_equal(max(abs(C %*% t(r1))), 0)
    expect_equal(max(abs(C %*% t(r5))), 0)
    expect_equal(max(abs(C %*% t(r7))), 0)
  })

  test_that("Covariance check with NA", {
    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "ols",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "str",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "wls",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "shr",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "sam",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "shrbe",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "shrbv",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "sambe",
        res = res,
        approach = "proj"
      )
    })

    expect_no_error({
      csocc(
        base = base,
        agg_mat = A,
        comb = "sambe",
        res = res,
        approach = "proj"
      )
    })
  })
}