code: def test_output_shape_correct(
      self, constant_shape=(4, 4, 6), batch_size=3
  ):
      input_tensor = tf.zeros((batch_size,))
  
      output_tensor = keras_layers.LearnedConstant(
          shape=constant_shape
      )(input_tensor)
  
      self.assertSequenceEqual(
          output_tensor.shape, (batch_size,) + constant_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 1 for calling the LearnedConstantTest obj's underlying member method code with example usage: obj.test_output_shape_correct(constant_shape, batch_size) without return types