code: def test_build_model(self, model_config_with_input_dim):
      model = BaseAE(model_config_with_input_dim)
      assert all(
          [
              model.input_dim == model_config_with_input_dim.input_dim,
              model.latent_dim
              == model_config_with_input_dim.latent_dim,
          ]
      )
  
file_path: benchmark_VAE/tests/test_baseAE.py
imports: import os
  import shutil
  import pytest
  import torch
  from pythae.customexception import BadInheritanceError
  from pythae.models import BaseAE, BaseAEConfig
  from tests.data.custom_architectures import Decoder_AE_Conv, Encoder_AE_Conv, NetBadInheritance
label_desc: Use image node_id 1 for calling the Test_Model_Building obj's underlying member method code with example usage: obj.test_build_model(model_config_with_input_dim) without return types