Skip to content

Commit f33be71

Browse files
committed
expand custom-layer test
1 parent 9202548 commit f33be71

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/testthat/test-Layer.R

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,9 @@ test_succeeds("custom layers can accept standard layer args like input_shape", {
247247
layer_simple_dense(20, input_shape = 30) %>%
248248
layer_dense(10)
249249

250+
expect_identical(dim(model$input), c(NA_integer_, 30L))
251+
expect_true(model$built)
252+
250253
res <- model(random_array(1, 30))
251254
expect_tensor(res, shape = c(1L, 10L))
252255

0 commit comments

Comments
 (0)