code: def testInvalidAxis(self, axis):
      with self.assertRaisesRegex(
          ValueError,
          "`axis` should be an int, slice or iterable of ints.",
      ):
          group_norm.GroupNorm(
              groups=5,
              axis=axis,
              create_scale=False,
              create_offset=False,
          )
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import group_norm
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 11 for calling the GroupNormTest obj's underlying member method code with example usage: obj.testInvalidAxis(axis) without return types