code: def get_fields(N):
      domain = Domain(N, dx=[1.0] * len(N))
  
      def f(p, x):
          return jnp.expand_dims(jnp.sum(p * (x**2)), -1)
  
      # Defining fields
      a = Continuous(5.0, domain, f)
      b = Continuous(6.0, domain, f)
      return a, b
  
imports: import numpy
  import pytest
  from jax import jit
  from jax import numpy
  from jax import random
  from jaxdf.discretization import Continuous
  from jaxdf.geometry import Domain
label_desc: Use image node_id 4 for calling a global function with example usage: get_fields(N) and returns: a,  b, jnp