code: def test_expand_apply_raises(self):
      with self.assertRaisesRegex(
          ValueError, "only supports axis=0 or axis=-1"
      ):
          basic.expand_apply(lambda: 1, axis=1)()
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import basic
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  from jax import random
  import jax.numpy
  import numpy
label_desc: Use image node_id 16 for calling the BasicTest obj's underlying member method code with example usage: obj.test_expand_apply_raises() without return types