tests/testthat/test.tidytensor.R

In oneilsh/tidytensor: TidyTensor

context("as.tidytensor")

test_that("as.tidytensor() works properly", {
  x <- c(1, 2, 3)
  tx <- as.tidytensor(x)
  expect_equal(class(tx)[1], "tidytensor")
  expect_equal(dim(tx), 3)

  x <- matrix(1:6, nrow = 2, ncol = 3)
  tx <- as.tidytensor(x)
  expect_equal(class(tx)[1], "tidytensor")
  expect_equal(dim(x), c(2, 3))

  x <- array(1:(3 * 4 * 5), dim = c(3, 4, 5))
  tx <- as.tidytensor(x)
  expect_equal(class(tx)[1], "tidytensor")
  expect_equal(dim(x), c(3, 4, 5))

  x <- data.frame(col1 = c(1, 2), col2 = c(3, 4))
  expect_error(as.tidytensor(x))

  x <- c(1, 2, 3)
  names(x) <- c("A", "B", "C")
  tx <- as.tidytensor(x)
  expect_equal(dimnames(tx)[[1]], c("A", "B", "C"))
})