add tests

Author: sebffischer

tests/testthat/test_materialize.R

@@ -189,6 +189,22 @@ test_that("0-length", {
   expect_equal(list(), materialize(lazy_tensor(), rbind = FALSE))
 })
 
+test_that("materialize.data.frame with empty data.frame", {
+  lt = as_lazy_tensor(torch_randn(5, 3))
+  df = data.frame(x = I(lt[integer(0)]))
+
+  res_rbind = materialize(df, rbind = TRUE)
+  expect_list(res_rbind)
+  expect_equal(names(res_rbind), "x")
+  expect_class(res_rbind$x, "torch_tensor")
+  expect_equal(res_rbind$x$shape, 0L)
+
+  res_list = materialize(df, rbind = FALSE)
+  expect_list(res_list)
+  expect_equal(names(res_list), "x")
+  expect_list(res_list$x, len = 0)
+})
+
 test_that("materialize with shape (NA, NA) and .getbatch implementation", {
   # this can e.g. happen when we do padding in the dataset
   ds = dataset(

tests/testthat/test_utils.R

@@ -1,3 +1,12 @@
+test_that("auto_device handles NULL and 'auto'", {
+  # NULL should be returned as-is
+  expect_null(auto_device(NULL))
+  # explicit device should be returned as-is
+  expect_equal(auto_device("cpu"), "cpu")
+  # "auto" should return either "cuda" or "cpu"
+  expect_true(auto_device("auto") %in% c("cuda", "cpu"))
+})
+
 test_that("make_check_vector works", {
   check_vector1 = make_check_vector(1)
   expect_true(check_vector1(1))