class_docstr: Tests for the qml.map_wires method used with callables.
class_docstr_tok: ["Tests","for","the","qml.map_wires","method","used","with","callables","."]
docstr: Test that we can jit qnodes that have a mapped quantum function.
docstr_tok: ["Test","that","we","can","jit","qnodes","that","have","a","mapped","quantum","function","."]
code: def test_jitting_simplified_qfunc(self):
      """Test that we can jit qnodes that have a mapped quantum function."""
      import jax
  
      @jax.jit
      @partial(qml.map_wires, wire_map=wire_map)
      @qml.qnode(qml.device("default.qubit", wires=5))
      def circuit(x):
          qml.adjoint(qml.RX(x, wires=0))
          _ = qml.PauliX(0) ** 2
          return qml.expval(qml.PauliY(0))
  
      x = jax.numpy.array(4 * jax.numpy.pi + 0.1)
      res = circuit(x)
      assert qml.math.allclose(res, jax.numpy.sin(x))
  
      grad = jax.grad(circuit)(x)
      assert qml.math.allclose(grad, jax.numpy.cos(x))
  
imports: from functools import partial
  import pytest
  import pennylane
  from pennylane.ops import Prod
  from pennylane.tape import QuantumScript
  from pennylane.wires import Wires
label_desc: Use image node_id 2 for calling the TestMapWiresCallables obj's underlying member method code with example usage: obj.test_jitting_simplified_qfunc() and returns: qml