code: def test_fd_shift_operator():
      domain = geometry.Domain((5, 5), (0.5, 0.5))
      params = jnp.zeros((5, 5, 1))
      params = params.at[2, 2].set(1.0)
      x = FiniteDifferences(params, domain, accuracy=2)
  
      y = operators.shift_operator(x, dx=[0.125, 0.0]).on_grid[..., 0]
      y_true = jnp.asarray(
          [
              [
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  0.25,
                  0.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  0.75,
                  0.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
              ],
              [
                  0.0,
                  0.0,
                  0.0,
                  0.0,
                  0.0,
              ],
          ]
      )
      assert np.allclose(y, y_true, atol=ATOL)
  
file_path: jaxdf/tests/test_functions.py
imports: import jax
  import numpy
  from jax import jit
  from jax import numpy
  from jaxdf import *
label_desc: Use image node_id 8 for calling a global function with example usage: test_fd_shift_operator() without return types