code: def test_output_is_normalized(self, shape, axis):
      input_tensor = tf.random.normal(shape=shape)
  
      normalized_tensor = keras_layers.PixelNormalization(axis=axis)(
          input_tensor
      )
      normalized_lengths = tf.norm(normalized_tensor, axis=axis)
  
      self.assertAllClose(
          normalized_lengths,
          tf.fill(
              dims=normalized_lengths.shape,
              value=math.sqrt(shape[axis]),
          ),
      )
  
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 PixelNormalizationTest obj's underlying member method code with example usage: obj.test_output_is_normalized(shape, axis) without return types