code: def test_jit_with_float(discretization):
      N = (1,)
      domain = Domain(N, dx=[1.0] * len(N))
      x = discretization.from_grid(jnp.asarray([1.0]), domain)
      y = discretization.from_grid(jnp.asarray([2.0]), domain)
  
      @jit
      def add(x, y):
          return x + y * 10
  
      _ = add(x, y)
      _ = add(x, 6.0)
      _ = add(-5.0, x)
  
imports: import pytest
  from jax import jit
  from jax import numpy
  from jax import random
  from jaxdf.discretization import FiniteDifferences, FourierSeries, OnGrid
  from jaxdf.geometry import Domain
  from jaxdf.operators import inverse
label_desc: Use image node_id 4 for calling a global function with example usage: test_jit_with_float(discretization) and returns: unknown