add transformer test

OHDSI · Sep 7, 2023 · 66cfbb4 · 66cfbb4
1 parent 5540e7d
commit 66cfbb4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/testthat/test-Transformer.R b/tests/testthat/test-Transformer.R
@@ -119,6 +119,10 @@ test_that("dimHidden ratio works as expected", {
                                   randomSample = randomSample)
   dimHidden <- unlist(lapply(modelSettings$param, function(x) x$dimHidden))
   tokens <-   unlist(lapply(modelSettings$param, function(x) x$dimToken))
-  expect_true(all(dimHidden == dimHiddenRatio * tokens))
+  testthat::expect_true(all(dimHidden == dimHiddenRatio * tokens))
+  testthat::expect_error(setTransformer(dimHidden = NULL,
+                                        dimHiddenRatio = NULL))
+  testthat::expect_error(setTransformer(dimHidden = 256,
+                                        dimHiddenRatio = 4/3))
 
 })