tests/testthat/test-extract_sparse_parts.R

In rstan: R Interface to Stan

test_that("extract_sparse_parts works", {
  A <- rbind(
    c(19L, 27L,  0L,  0L),
    c( 0L,  0L,  0L,  0L),
    c( 0L,  0L,  0L, 52L),
    c(81L,  0L, 95L, 33L)
  )
  w <- c(19, 27, 52, 81, 95, 33)
  v <- c(1, 2, 4, 1, 3, 4)
  u <- c(1, 3, 3, 4, 7)
  parts <- rstan::extract_sparse_parts(A)
  expect_equal(parts$w, w)
  expect_equal(parts$v, v)
  expect_equal(parts$u, u)

  parts <- rstan::extract_sparse_parts(A * 1.0)
  expect_equal(parts$w, w)
  expect_equal(parts$v, v)
  expect_equal(parts$u, u)

  parts <- rstan::extract_sparse_parts(Matrix::Matrix(A))
  expect_equal(parts$w, w)
  expect_equal(parts$v, v)
  expect_equal(parts$u, u)
})