code: def config_contains_correct_axis(self, axis):
      layer = keras_layers.PixelNormalization(axis=axis)
  
      config = layer.get_config()
  
      self.assertIn(config, "axis")
      self.assertEqual(config["axis"], 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 2 for calling the PixelNormalizationTest obj's underlying member method code with example usage: obj.config_contains_correct_axis(axis) without return types