add test

mlr-org · Oct 13, 2023 · ce54f27 · ce54f27
1 parent ca0c6c9
commit ce54f27
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/testthat/test_lazy_tensor.R b/tests/testthat/test_lazy_tensor.R
@@ -17,6 +17,8 @@ test_that("lazy_tensor works", {
   expect_class(lt1_empty, "lazy_tensor")
 
   lt1_empty[1] = lt1[1]
+
+  expect_class(materialize(lazy_tensor()), "torch_tensor")
 })
 
 test_that("transform_lazy_tensor works", {