class_docstr: Unit tests for jax execution
class_docstr_tok: ["Unit","tests","for","jax","execution"]
docstr: Test setting jacobian options
docstr_tok: ["Test","setting","jacobian","options"]
code: def test_jacobian_options(self, mocker):
      """Test setting jacobian options"""
      spy = mocker.spy(qml.gradients, "param_shift")
  
      a = jax.numpy.array([0.1, 0.2])
  
      dev = qml.device("default.qubit.legacy", wires=1)
  
      def cost(a, device):
          with qml.queuing.AnnotatedQueue() as q:
              qml.RY(a[0], wires=0)
              qml.RX(a[1], wires=0)
              qml.expval(qml.PauliZ(0))
  
          tape = qml.tape.QuantumScript.from_queue(q)
          return execute(
              [tape],
              device,
              gradient_fn=param_shift,
              gradient_kwargs={"shifts": [(np.pi / 4,)] * 2},
          )[0]
  
      jax.grad(cost)(a, device=dev)
  
      for args in spy.call_args_list:
          assert args[1]["shifts"] == [(np.pi / 4,)] * 2
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane.gradients import param_shift
  from pennylane import execute
label_desc: Use image node_id 1 for calling the TestJaxExecuteUnitTests obj's underlying member method code with example usage: obj.test_jacobian_options(mocker) and returns: execute