tests/testthat/test-flat-pick-matrix-column.R

In sakrejda/dt: Core modeling tools

test_that("can pick out a flat matrix column.", {
  n_tests = 20
  for (i in 1:n_tests) {
    N = sample.int(n = 1000, size = 1)
    K = sample.int(n = 50, size = 1) + 5
    m1 = generate_random_matrix_with_zeros(N, K, 99, 0.6)
    j = sample.int(n = K, size = 1)
    c1 = m1[,j]
    fm1 = row_flatten_matrix(m1)
    fc1 = flat_pick_matrix_column(fm1, j)
    rc1 = flat_to_matrix(fc1)
    expect_equivalent(c1, rc1)
  }
})