code: def test_output_shape_correct_with_input_noise(
      self, shape=(2, 5, 7, 3)
  ):
      input_feature_map = tf.ones(shape)
      input_noise = tf.ones(shape[:-1] + (1,))
  
      output_tensor = keras_layers.Noise()(
          (input_feature_map, input_noise)
      )
  
      self.assertSequenceEqual(output_tensor.shape, shape)
  
imports: import math
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_graphics.projects.gan import keras_layers
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 2 for calling the NoiseTest obj's underlying member method code with example usage: obj.test_output_shape_correct_with_input_noise(shape) without return types