test_that("Is a neural network module.", {
  expect_class(nn_model(in_dim = 1, noise_dim = 100,
                        out_dim = 1), "nn_module")
})