def test_feedforward_symmetric_basic(self, _):
"""
Test that feedforward_symmetric calls feedforward_model correctly
"""
n_features, enc_dim, dec_dim, enc_func, dec_func = feedforward_symmetric(
5, [4, 3, 2, 1], ["relu", "relu", "tanh", "tanh"]
)
self.assertEqual(n_features, 5)
self.assertEqual(enc_dim, [4, 3, 2, 1])
self.assertEqual(dec_dim, [1, 2, 3, 4])
self.assertEqual(enc_func, ["relu", "relu", "tanh", "tanh"])
self.assertEqual(dec_func, ["tanh", "tanh", "relu", "relu"])
def test_feedforward_symmetric_checks_dims(self):
"""
Test that feedforward_symmetric validates parameter requirements
"""
with self.assertRaises(ValueError):
feedforward_symmetric(4, [], [])
def test_feedforward_symmetric_basic(self, _):
"""
Test that feedforward_symmetric calls feedforward_model correctly
"""
n_features, n_features_out, encoding_dim, decoding_dim, encoding_func, decoding_func, optimizer, optimizer_kwargs, compile_kwargs = feedforward_symmetric(
5, 5, (4, 3, 2, 1), ("relu", "relu", "tanh", "tanh"), {})
self.assertEqual(n_features, 5)
self.assertEqual(compile_kwargs, {})
self.assertEqual(encoding_dim, (4, 3, 2, 1))
self.assertEqual(decoding_dim, (1, 2, 3, 4))
self.assertEqual(encoding_func, ("relu", "relu", "tanh", "tanh"))
self.assertEqual(decoding_func, ("tanh", "tanh", "relu", "relu"))
