tests/testthat/test-types.R

In tensorflow: R Interface to 'TensorFlow'

test_that("TensorShapes are not converted to lists", {

  skip_if_no_tensorflow()

  x <- tf$constant(10, shape = shape(5,10))
  expect_true(x$shape[2] == shape(10))
  expect_identical(x$shape[[2]], 10L)

  y <- tf$TensorShape(shape(5L, 10L))
  expect_s3_class(y, "tensorflow.python.framework.tensor_shape.TensorShape")
  expect_true(y[2] == shape(10))
  expect_identical(y[[2]], 10L)
})


test_that("tf.random works", {
  # Installing tensorflow-metal 1.2.0 makes this error.
  x <- tf$random$stateless_uniform(
    shape = tuple(10L),
    seed = tuple(2L, 3L),
    minval = 0L,
    maxval = 10L,
    dtype = tf$dtypes$int32)
  expect_s3_class(x, "tensorflow.tensor")
  x <- as.array(x)
  if (!is_windows())
    expect_type(x, "integer")
  expect_identical(dim(x), 10L)
})