code: def testLaxFftErrors(self):
      with self.assertRaisesRegex(
          ValueError,
          r"FFT input shape \(14, 15\) must have at least as many input "
          r"dimensions as fft_lengths \(4, 5, 6\)",
      ):
          lax.fft(
              np.ones((14, 15)), fft_type="fft", fft_lengths=(4, 5, 6)
          )
      with self.assertRaisesRegex(
          ValueError,
          r"FFT input shape \(14, 15\) minor dimensions must be equal to "
          r"fft_lengths \(17,\)",
      ):
          lax.fft(np.ones((14, 15)), fft_type="fft", fft_lengths=(17,))
      with self.assertRaisesRegex(
          ValueError,
          r"RFFT input shape \(2, 14, 15\) minor dimensions must be equal to "
          r"fft_lengths \(14, 12\)",
      ):
          lax.fft(
              np.ones((2, 14, 15)),
              fft_type="rfft",
              fft_lengths=(14, 12),
          )
      with self.assertRaisesRegex(
          ValueError,
          r"IRFFT input shape \(2, 14, 15\) minor dimensions must be equal to "
          r"all except the last fft_length, got fft_lengths=\(13, 15\)",
      ):
          lax.fft(
              np.ones((2, 14, 15)),
              fft_type="irfft",
              fft_lengths=(13, 15),
          )
      with self.assertRaisesRegex(
          ValueError,
          "RFFT input must be float32 or float64, got bfloat16",
      ):
          lax.fft(
              np.ones((14, 15), jnp.bfloat16),
              fft_type="rfft",
              fft_lengths=(5, 6),
          )
  
imports: import itertools
  import numpy
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import lax
  from jax import numpy
  from jax._src import config
  from jax._src import dtypes
  from jax._src import test_util
  from jax._src.numpy.util import promote_dtypes_complex
label_desc: Use image node_id 2 for calling the FftTest obj's underlying member method code with example usage: obj.testLaxFftErrors() without return types