code: def test_with_multiple_input_shapes(
      self, input_dims, expected_output_dims
  ):
      test_data = self.run_one_to_many_encode_decode(
          self.default_encoding_stage(),
          lambda: tf.random.stateless_normal(input_dims, (7, 77)),
      )
      self.common_asserts_for_test_data(test_data)
      # Make sure output shape is as expected.
      self.assertEqual(
          expected_output_dims,
          test_data.encoded_x[
              stages_impl.HadamardEncodingStage.ENCODED_VALUES_KEY
          ].shape,
      )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.stages import stages_impl
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 6 for calling the HadamardEncodingStageTest obj's underlying member method code with example usage: obj.test_with_multiple_input_shapes(input_dims, expected_output_dims) without return types