code: def test_invalid_input_shape(self):
      with_bias = True
  
      with self.assertRaisesRegex(
          ValueError,
          r"Input to ConvND needs to have rank in \[4, 5\], but input has shape",
      ):
          data = jnp.ones([1, 5, 5, 5, 1, 9, 9])
          net = conv.Conv3D(
              output_channels=1,
              kernel_shape=3,
              stride=1,
              padding="VALID",
              with_bias=with_bias,
              **create_constant_initializers(1.0, 1.0, with_bias),
          )
          net(data)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import conv
  from haiku._src import initializers
  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 3 for calling the Conv3DTest obj's underlying member method code with example usage: obj.test_invalid_input_shape() without return types