code: def test_adult_autoencoder(
      input_dim, hidden_dim, latent_dim, output_dims
  ):
      """Tests heterogeneous autoencoder composition."""
      x = np.random.randn(1, input_dim)
  
      enc = ADULTEncoder(hidden_dim=hidden_dim, latent_dim=latent_dim)
      dec = ADULTDecoder(hidden_dim=hidden_dim, output_dims=output_dims)
      ae = HeAE(encoder=enc, decoder=dec)
  
      x_hat1 = ae(x)
      x_hat2 = dec(enc(x))
      for x1, x2 in zip(x_hat1, x_hat2):
          np.testing.assert_allclose(x1, x2)
  
file_path: alibi/alibi/models/tensorflow/tests/test_cfrl_models.py
imports: import numpy
  import pytest
  from alibi.models.tensorflow.autoencoder import AE, HeAE
  from alibi.models.tensorflow.cfrl_models import ADULTDecoder, ADULTEncoder, MNISTClassifier, MNISTDecoder, MNISTEncoder
label_desc: Use image node_id 5 for calling a global function with example usage: test_adult_autoencoder(input_dim, hidden_dim, latent_dim, output_dims) without return types