add unit tests

OHDSI · Oct 12, 2023 · 916c32c · 916c32c
1 parent 5c5baad
commit 916c32c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/testthat/test-Transformer.R b/tests/testthat/test-Transformer.R
@@ -86,6 +86,12 @@ test_that("transformer nn-module works", {
   )
   output <- model(input)
   expect_equal(output$shape[0], 10L)
+
+  input$num <- reticulate::py_none()
+  output <- model(input)
+  expect_equal(output$shape[0], 10L)
+
+
 })
 
 test_that("Default Transformer works", {