class_docstr: Tests for Fidelity function between two QNodes.
docstr: Test the gradient of the fidelity between Rx and PauliZ circuits.
code: def test_fidelity_qnodes_rx_pauliz_jax_grad_jit(
      self, param, interface
  ):
      """Test the gradient of the fidelity between Rx and PauliZ circuits."""
      import jax
  
      dev = qml.device("default.qubit", wires=2)
  
      @qml.qnode(dev, interface=interface)
      def circuit0(x):
          qml.RX(x, wires=0)
          return qml.state()
  
      @qml.qnode(dev, interface="jax")
      def circuit1():
          qml.PauliZ(wires=0)
          return qml.state()
  
      fid_grad = jax.jit(
          jax.grad(
              qml.qinfo.fidelity(
                  circuit0, circuit1, wires0=[0], wires1=[0]
              )
          )
      )((jax.numpy.array(param)))
      expected_fid = expected_grad_fidelity_rx_pauliz(param)
      assert qml.math.allclose(
          fid_grad, expected_fid, rtol=1e-04, atol=1e-03
      )
  
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 23 for calling the TestFidelityQnode obj's underlying member method code with example usage: obj.test_fidelity_qnodes_rx_pauliz_jax_grad_jit(param, interface) and returns: qml, qml