code: def test_equinox_omega_wrapper():
      domain = geometry.Domain((1,), (1.0,))
      a = OnGrid(jnp.asarray([1.0]), domain)
  
      b = a**ω
      c = ω(a)
  
      assert type(b) == type(c)
  
label_desc: Use image node_id 1 for calling a global function with example usage: test_equinox_omega_wrapper() without return types