tests/testthat/test-reshape.R

In rray: Simple Arrays

context("test-reshape")

test_that("can reshape up in dimensions", {

  expect_equal(
    rray_reshape(1, c(1, 1)),
    new_matrix(1)
  )

  expect_equal(
    rray_reshape(1:5, c(5, 1)),
    new_matrix(1:5)
  )

  expect_equal(
    rray_reshape(matrix(1), c(1, 1, 1)),
    new_array(1, c(1, 1, 1))
  )

})

test_that("can reshape down in dimensions", {

  expect_equal(
    rray_reshape(matrix(1), 1),
    new_array(1)
  )

  expect_equal(
    rray_reshape(matrix(1:5), 5),
    new_array(1:5)
  )

  expect_equal(
    rray_reshape(array(1:5, c(5, 1, 1)), 5),
    new_array(1:5)
  )
})

test_that("dimension names are maintained where appropriate", {

  x <- new_matrix(1:5, c(5, 1), list(meta = letters[1:5], "c1"))

  expect_equal(
    rray_dim_names(rray_reshape(x, 5)),
    list(meta = letters[1:5])
  )

  expect_equal(
    rray_dim_names(rray_reshape(x, c(5, 1, 1))),
    c(rray_dim_names(x), rray_empty_dim_names(1))
  )

})

test_that("meta names are retained if the axis size doesn't change (#201)", {
  x <- new_matrix(1:5, c(5, 1), list(rows = letters[1:5], cols = "c1"))

  expect_equal(
    rray_dim_names(rray_reshape(x, c(5, 1))),
    rray_dim_names(x)
  )

  expect_equal(
    rray_dim_names(rray_reshape(x, c(1, 1, 5))),
    list(NULL, cols = "c1", NULL)
  )

  expect_equal(
    rray_dim_names(rray_reshape(x, c(5, 1, 1))),
    c(rray_dim_names(x), list(NULL))
  )
})

test_that("can reshape 0 row input", {

  x <- new_matrix(numeric(), c(0, 0))

  expect_equal(
    rray_reshape(x, c(2, 0)),
    new_matrix(numeric(), c(2, 0))
  )

  expect_equal(
    rray_reshape(x, c(2, 0, 1)),
    new_array(numeric(), c(2, 0, 1))
  )

})

test_that("can reshape 0 column input", {

  x <- new_matrix(numeric(), c(2, 0))

  expect_equal(
    rray_reshape(x, c(0, 2)),
    new_matrix(numeric(), c(0, 2))
  )

  expect_equal(
    rray_reshape(x, c(0, 2, 2)),
    new_array(numeric(), c(0, 2, 2))
  )

  expect_equal(
    rray_reshape(x, c(0, 2, 0)),
    new_array(numeric(), c(0, 2, 0))
  )
})

test_that("can reshape with `NULL` input", {
  expect_equal(rray_reshape(NULL, c(1, 2)), NULL)
})

test_that("errors are thrown when reshaping to the wrong size", {
  expect_error(rray_reshape(numeric(), 1), "The size you are reshaping from")
  expect_error(rray_reshape(1, c(1, 2)), "The size you are reshaping from")
})