add tests for if random sample is higher than possible combinations

OHDSI · Oct 10, 2022 · ee09f83 · ee09f83
1 parent 599737e
commit ee09f83
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 0 deletions.
diff --git a/tests/testthat/test-MLP.R b/tests/testthat/test-MLP.R
@@ -108,3 +108,20 @@ test_that("MLP nn-module works ", {
   # model works without numeric variables
   expect_equal(output$shape, 10)
 })
+
+
+test_that("Errors are produced by settings function", {
+  randomSample <- 2
+
+  expect_error(setMultiLayerPerceptron(
+    numLayers = 1,
+    sizeHidden = 128,
+    dropout = 0.0,
+    sizeEmbedding = 128,
+    weightDecay = 1e-6,
+    learningRate = 0.01,
+    seed = 42,
+    hyperParamSearch = 'random',
+    randomSample = randomSample))
+
+})
diff --git a/tests/testthat/test-ResNet.R b/tests/testthat/test-ResNet.R
@@ -124,3 +124,20 @@ test_that("Default Resnet works", {
   expect_equal(params$sizeEmbedding, 256)
 
 }) 
+
+test_that("Errors are produced by settings function", {
+  randomSample <- 2
+
+  expect_error(setResNet(
+    numLayers = 1,
+    sizeHidden = 128,
+    hiddenFactor = 1,
+    residualDropout = 0.0,
+    hiddenDropout = 0.0,
+    sizeEmbedding = 128,
+    weightDecay = 1e-6,
+    learningRate = 0.01,
+    seed = 42,
+    hyperParamSearch = 'random',
+    randomSample = randomSample))
+})
diff --git a/tests/testthat/test-Transformer.R b/tests/testthat/test-Transformer.R
@@ -74,3 +74,9 @@ test_that("Default Transformer works", {
 
   expect_equal(settings$name, 'defaultTransformer')
 }) 
+
+test_that("Errors are produced by settings function", {
+  randomSample <- 2
+
+  expect_error(setTransformer(randomSample = randomSample))
+})