class_docstr: Tests for Fidelity function between two QNodes.
docstr: Test the fidelity between Rx and PauliZ circuits with Jax jit.
code: def test_fidelity_qnodes_rx_pauliz_jax_jit(self, param, wire):
      """Test the fidelity between Rx and PauliZ circuits with Jax jit."""
      import jax
  
      # TODO: add default mixed
      dev = qml.device("default.qubit", wires=wire)
  
      @qml.qnode(dev, interface="jax-jit")
      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 = jax.jit(
          qml.qinfo.fidelity(circuit0, circuit1, wires0=[0], wires1=[0])
      )((jax.numpy.array(param)))
      expected_fid = expected_fidelity_rx_pauliz(param)
      assert qml.math.allclose(
          fid, expected_fid, rtol=1e-03, atol=1e-04
      )
  
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 21 for calling the TestFidelityQnode obj's underlying member method code with example usage: obj.test_fidelity_qnodes_rx_pauliz_jax_jit(param, wire) and returns: qml, qml