class_docstr: Tests for LAX-backed scipy.interpolate implementations
class_docstr_tok: ["Tests","for","LAX-backed","scipy.interpolate","implementations"]
code: def testRegularGridInterpolator(self, spaces, method):
      rng = jtu.rand_default(self.rng())
      scipy_fun = lambda init_args, call_args: sp_interp.RegularGridInterpolator(
          *init_args[:2], method, False, *init_args[2:]
      )(
          *call_args
      )
      lax_fun = lambda init_args, call_args: jsp_interp.RegularGridInterpolator(
          *init_args[:2], method, False, *init_args[2:]
      )(
          *call_args
      )
  
      def args_maker():
          points = tuple(map(lambda x: np.linspace(*x), spaces))
          values = rng(
              reduce(operator.add, tuple(map(np.shape, points))), float
          )
          fill_value = np.nan
  
          init_args = (points, values, fill_value)
          n_validation_points = 50
          valid_points = tuple(
              map(
                  lambda x: np.linspace(
                      x[0] - 0.2 * (x[1] - x[0]),
                      x[1] + 0.2 * (x[1] - x[0]),
                      n_validation_points,
                  ),
                  spaces,
              )
          )
          valid_points = np.squeeze(np.stack(valid_points, axis=1))
          call_args = (valid_points,)
          return init_args, call_args
  
      self._CheckAgainstNumpy(
          scipy_fun, lax_fun, args_maker, check_dtypes=False, tol=1e-4
      )
      self._CompileAndCheck(
          lax_fun, args_maker, rtol={np.float64: 1e-14}
      )
  
file_path: jax/tests/scipy_interpolate_test.py
imports: from absl.testing import absltest
  import operator
  from functools import reduce
  import numpy
  from jax._src import test_util
  import scipy.interpolate
  import jax.scipy.interpolate
  from jax import config
label_desc: Use image node_id 1 for calling the LaxBackedScipyInterpolateTests obj's underlying member method code with example usage: obj.testRegularGridInterpolator(spaces, method) and returns: init_args,  call_args